結果

問題 No.3685 ワロングアンサーやんけ！
コンテスト
ユーザー のらら
提出日時 2026-09-05 19:22:16
言語 C++23
(gcc 15.3.0 + boost 1.92.0)
コンパイル:
g++-15 -O2 -lm -std=c++23 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
WA  
実行時間 -
コード長 2,383 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 3,179 ms
コンパイル使用メモリ 265,796 KB
実行使用メモリ 9,780 KB
最終ジャッジ日時 2026-09-05 19:22:24
合計ジャッジ時間 5,947 ms
ジャッジサーバーID
（参考情報） 		judge2_0 / judge3_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 11 WA * 21
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <iostream>
#include <algorithm>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
using ll = long long;
//#define endl "\n";
int main(){
    ll Q;
    cin >> Q;
    for(int q = 1; q <= Q; q++){
        string R, S;
        cin >> R;
        cin >> S;
        ll K;
        cin >> K;
        ll w = 0, a = 0;
        fenwick_tree<ll> fwA(R.size()), fwW(R.size());
        for(int i = 0; i < R.size(); i++){
            if(R[i] == 'W') fwW.add(i, 1);
            if(R[i] == 'A') fwA.add(i, 1);
        }
        vector<char> ans(R.size());
        for(int i = 0; i < R.size(); i++){
            if(R[i] == '?'){
                if(S == "Warong"){
                    bool a = false, w = false;
                    //A
                    fwA.add(i, 1);
                    if(fwW.sum(0, K) == 0 && fwA.sum(0, R.size()) + fwW.sum(0, R.size()) < R.size()) a = true;
                    fwA.add(i, -1);
                    //W
                    fwW.add(i, 1);
                    if(fwW.sum(0, K) == 0 && fwA.sum(0, R.size()) + fwW.sum(0, R.size()) < R.size()) w = true;
                    fwW.add(i, -1);
                    if(a && !w){
                        ans[i] = 'A';
                        fwA.add(i, 1);
                    }else if(!a && w){
                        ans[i] = 'W';
                        fwW.add(i, 1);
                    }else{
                        ans[i] = '?';
                    }
                }else{
                    bool a = false, w = false;
                    //A
                    fwA.add(i, 1);
                    if(fwW.sum(0, K) > 0 || fwW.sum(0, R.size()) == 0) a = true;
                    fwA.add(i, -1);
                    //W
                    fwW.add(i, 1);
                    if(fwW.sum(0, K) > 0 || (fwA.sum(0, K) + fwW.sum(0, K) < K)) w = true;
                    fwW.add(i, -1);
                    if(a && !w){
                        ans[i] = 'A';
                        fwA.add(i, 1);
                    }else if(!a && w){
                        ans[i] = 'W';
                        fwW.add(i, 1);
                    }else{
                        ans[i] = '?';
                    }
                }
            }else{
                ans[i] = R[i];
            }
        }
        for(int i = 0; i < R.size(); i++) cout << ans[i];
        cout << endl;
    }
    return 0;
}
0