結果

問題 No.3370 AB → BA
コンテスト
ユーザー t98slider
提出日時 2025-11-17 22:42:38
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,239 bytes
コンパイル時間 4,638 ms
コンパイル使用メモリ 264,224 KB
実行使用メモリ 53,576 KB
最終ジャッジ日時 2025-11-17 22:42:49
合計ジャッジ時間 10,219 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 6 WA * 10 RE * 4
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using ll = long long;
using mint = atcoder::modint998244353;

int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    string s;
    cin >> s;
    while(!s.empty() && s.back() == 'B') s.pop_back();
    int n = s.size();
    reverse(s.begin(), s.end());
    vector<vector<mint>> tb(n);
    vector<pair<int,int>> rng(n);
    for(int i = 0; i < n; i++){
        if(s[i] == 'A'){
            tb[i] = {1, 1};
            rng[i] = make_pair(0, 1);
        }else{
            tb[i] = {1, 1};
            rng[i] = make_pair(-1, 0);
        }
    }
    auto dfs = [&](auto dfs, int l, int r) -> void {
        if(r - l <= 1) return;
        int mid = (l + r) / 2;
        dfs(dfs, l, mid);
        dfs(dfs, mid, r);
        tb[l] = atcoder::convolution(tb[l], tb[mid]);
        rng[l] = make_pair(rng[l].first + rng[mid].first, rng[l].second + rng[mid].second);
        if(l == 0){
            reverse(tb[l].begin(), tb[l].end());
            while(rng[l].first < 0){
                rng[l].first++;
                tb[l].pop_back();
            }
            reverse(tb[l].begin(), tb[l].end());
        }
    };
    dfs(dfs, 0, n);
    cout << tb[0][0].val() << '\n';
}
0