Problem p02692 (C++) — buggy submission, verdict: Time Limit Exceeded

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
typedef pair<int, int> ii;

const int N = 100005;
int n, a, b, c;
string s[N];

int main() {
  ios::sync_with_stdio(false);
  cin.tie(0);

  cin >> n >> a >> b >> c;
  for (int i = 0; i < n; ++i) {
    cin >> s[i];
  }
  ll sum = 0;
  sum += a;
  sum += b;
  sum += c;
  bool ok = true;
  string res = "";

  if (sum > 2) {
    // greedy
    for (int i = 0; i < n; ++i) {
      if (s[i] == "AB") {
        if (!a && !b) {
          ok = false;
          break;
        }
        if (a > b) {
          a--;
          b++;
          res += 'B';
        } else {
          a++;
          b--;
          res += 'A';
        }
      } else if (s[i] == "AC") {
        if (!a && !c) {
          ok = false;
          break;
        }
        if (a > c) {
          a--;
          c++;
          res += 'C';
        } else {
          a++;
          c--;
          res += 'A';
        }
      } else {
        if (!b && !c) {
          ok = false;
          break;
        }
        if (b > c) {
          b--;
          c++;
          res += 'C';
        } else {
          b++;
          c--;
          res += 'B';
        }
      }
    }
  } else if (sum == 0) {
    ok = false;
  } else if (sum == 1) {
    for (int i = 0; i < n; ++i) {
      if (s[i] == "AB") {
        if (!a && !b) {
          ok = false;
          break;
        }
        if (a == 1) {
          a = 0;
          b = 1;
          res += 'B';
        } else {
          a = 1;
          b = 0;
          res += 'A';
        }
      } else if (s[i] == "AC") {
        if (!a && !c) {
          ok = false;
          break;
        }
        if (a == 1) {
          a = 0;
          c = 1;
          res += 'C';
        } else {
          a = 1;
          c = 0;
          res += 'A';
        }
      } else { // BC
        if (!b && !c) {
          ok = false;
          break;
        }
        if (b == 1) {
          b = 0;
          c = 1;
          res += 'C';
        } else {
          b = 1;
          c = 0;
          res += 'B';
        }
      }
    }
  } else { // sum == 2;
    for (int i = 0; i < n; ++i) {
      if (s[i] == "AB") {
        if (!a && !b) {
          ok = false;
          break;
        }
        if (a == b) {
          if (i < n - 1) {
            if (s[i + 1][0] != 'A') {
              a--;
              b++;
              res += 'B';
            } else {
              a++;
              b--;
              res += 'A';
            }
          } else {
            res += 'A';
          }
        } else if (a > b) {
          a--;
          b++;
          res += 'B';
        } else {
          a++;
          b--;
          res += 'A';
        }
      } else if (s[i] == "AC") {
        if (!a && !c) {
          ok = false;
          break;
        }
        if (a == c) {
          if (i < n - 1) {
            if (s[i + 1][0] != 'A') {
              a--;
              c++;
              res += 'C';
            } else {
              a++;
              c--;
              res += 'A';
            }
          } else {
            res += 'A';
          }
        } else if (a > c) {
          a--;
          c++;
          res += 'C';
        } else {
          a++;
          c--;
          res += 'A';
        }
      } else { // "BC"
        if (!b && !c) {
          ok = false;
          break;
        }
        if (b == c) {
          if (i < n - 1) {
            if (s[i + 1][0] != 'B' && s[i + 1][1] != 'B') {
              b--;
              c++;
              res += 'C';
            } else {
              b++;
              c--;
              res += 'B';
            }
          } else {
            res += 'B';
          }
        } else if (b > c) {
          b--;
          c++;
          res += 'C';
        } else {
          b++;
          c--;
          res += 'B';
        }
      }
    }
  }

  if (ok) {
    cout << "Yes" << endl;
    for (auto cc : res)
      cout << cc << endl;
  } else {
    cout << "No" << endl;
  }

  return 0;
}


Error: TLE

Fixed submission:

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
typedef pair<int, int> ii;

const int N = 100005;
int n, A, B, C;
string s[N];
string res;
unordered_map<string, bool> ma;

bool dfs(int index, int a, int b, int c, string cur) {
  if (index == n) {
    res = cur;
    return true;
  }
  string k = to_string(index) + ',' + to_string(a) + ',' + to_string(b) + ',' +
             to_string(c);
  if (ma.count(k))
    return ma[k];

  if (s[index] == "AB") {
    if (a > 0) {
      if (dfs(index + 1, a - 1, b + 1, c, cur + 'B')) {
        ma[k] = true;
        return true;
      }
    }
    if (b > 0) {
      return ma[k] = dfs(index + 1, a + 1, b - 1, c, cur + 'A');
    }
    return ma[k] = false;
  } else if (s[index] == "AC") {
    if (a > 0)
      if (dfs(index + 1, a - 1, b, c + 1, cur + 'C')) {
        ma[k] = true;
        return true;
      }
    if (c > 0)
      return ma[k] = dfs(index + 1, a + 1, b, c - 1, cur + 'A');
    return ma[k] = false;
  } else {
    if (b > 0)
      if (dfs(index + 1, a, b - 1, c + 1, cur + 'C')) {
        ma[k] = true;
        return true;
      }
    if (c > 0)
      return ma[k] = dfs(index + 1, a, b + 1, c - 1, cur + 'B');
    return ma[k] = false;
  }
}

int main() {
  ios::sync_with_stdio(false);
  cin.tie(0);

  cin >> n >> A >> B >> C;
  for (int i = 0; i < n; ++i) {
    cin >> s[i];
  }
  if (dfs(0, A, B, C, "")) {
    cout << "Yes" << endl;
    for (auto c : res)
      cout << c << endl;
  } else {
    cout << "No" << endl;
  }

  return 0;
}