結果

問題 No.672 最長AB列
コンテスト
ユーザー moti
提出日時 2020-04-29 13:54:11
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,838 bytes
コンパイル時間 1,721 ms
コンパイル使用メモリ 169,436 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-11-28 07:59:43
合計ジャッジ時間 2,982 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 WA * 1
other AC * 8 WA * 8
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;

#define REP(i,a,b) for(int i=a;i<(int)b;i++)
#define rep(i,n) REP(i,0,n)
#define all(c) (c).begin(), (c).end()
#define zero(a) memset(a, 0, sizeof a)
#define minus(a) memset(a, -1, sizeof a)
#define watch(a) { std::cout << #a << " = " << a << "\n"; }
template<class T1, class T2> inline bool minimize(T1 &a, T2 b) { return b < a && (a = b, 1); }
template<class T1, class T2> inline bool maximize(T1 &a, T2 b) { return a < b && (a = b, 1); }
template<class T, class V> istream& operator>> (istream& ist, pair<T, V>& p) { return ist >> p.first >> p.second; }
template<class T> ostream& operator<< (ostream& ost, pair<T, T>& p) { return ost << p.first << ", " << p.second; }
template<class T> istream& operator>> (istream& ist, vector<T>& vs) { for(auto& e: vs) ist >> e; return ist; }

typedef long long ll;
int const inf = INT_MAX / 2;

int main() {
  string S; cin >> S;
  int const N = S.size();
  vector<int> A(N + 1), B(N + 1);
  int max_len = 0;
  rep(i, N) {
    A[i + 1] += A[i] + (S[i] == 'A');
    B[i + 1] += B[i] + (S[i] == 'B');
    auto diff = A[i + 1] - B[i + 1];
    if (diff >= 0) {
      auto iter = lower_bound(A.begin(), A.begin() + i + 1, diff);
      if (iter != A.end() && *iter == diff) {
        max_len = max(max_len, (int)(A.begin() + i + 1 - iter));
      }
    }
    else {
      diff *= -1;
      auto iter = lower_bound(B.begin(), B.begin() + i + 1, diff);
      if (iter != B.end() && *iter == diff) {
//        cout << "p: " << (iter - B.begin()) << ", diff: " << diff
//             << ", len: " << (int)(B.begin() + i + 1 - iter) << "\n";
        max_len = max(max_len, (int)(B.begin() + i + 1 - iter));
      }
    }
  }
//  rep(i, N + 1) cout << A[i] << " "; cout << "\n";
//  rep(i, N + 1) cout << B[i] << " "; cout << "\n";
  cout << max_len << "\n";
}
0