問題

B – ABC

プログラミング初級者から上級者まで楽しめる、競技プログラミングコンテストサイト「AtCoder」。オンラインで毎週開催プログラミングコンテストを開催しています。競技…

atcoder.jp

解法

1. AとBCに分けて考える。
2. Aより右にあるBCの個数だけ操作できる。
3. しかし、途中でBやCが途中で出てきたらカウントをリセットする。
4. うしろから考えた方が実装しやすい。

コード例

#include <bits/stdc++.h>
using namespace std;
#include <atcoder/all>
using namespace atcoder;
using ll = long long;
using P = pair<int, int>;
#define rep(i, n) for(int i = 0; i < (n); ++i)
// using mint = modint998244353;
// using mint = modint1000000007;
// const int mod = 1000000007;
// const ll INF = 1LL << 60;
// const int INF = 1001001001;

// AGC034-B
// https://atcoder.jp/contests/agc034/tasks/agc034_b

/**
 * ABCABCABC
 * A BC A BC A BC
 */

int main() {
  string S;
  cin >> S;

  reverse(S.begin(), S.end());

  ll ans = 0;
  ll cntA = 0;
  ll cntB = 0;
  int n = S.size();
  int i = 0;
  while(i < n) {
    if(S[i] == 'A') {
      cntA++;
      ans += cntB;
      i++;
    } else if(i < n - 1) {
      if(S[i] == 'C' && S[i + 1] == 'B') {
        cntB++;
        i += 2;
      } else {
        cntA = 0;
        cntB = 0;
        i++;
      }
    } else {
      i++;
    }
  }

  cout << ans << endl;
  return 0;
}