結果

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

ソースコード

diff #
raw source code 

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include <map>
#include <set>
#include <unordered_map>
#include <unordered_set>
#include <stack>
#include <queue>
#include <climits>
#include <chrono>
#include <atcoder/all>

using namespace std;

/*
Warong:WがK+1文字目以降にあってK文字目までは全部A
Not:Warongを満たさない（K+1文字目以降にWがないか最初のK文字にWが混ざっている)

WarongならRのK文字目までで?があったらAにする、K+1文字目以降でWがなく、?が一つだけなら?をWにする
NotならK文字目まで全部Aを満たしていたらK+1文字目以降の?はすべてA、
    K+1文字目以降にWが混ざっていたらK文字目までで?が一つでWがなければ?をWにする

*/

int main(){
    long long T;
    cin >> T;
    for (int i = 0; i < T; i++){
        string R,S;
        int K;
        cin >> R >> S >> K;

        if (S[0] == 'W'){
            for (int j = 0; j < K; j++){
                if (R[j] == '?') R[j] = 'A';
            }
            int q_cnt = 0,W_cnt = 0,pos = -1;;

            for (int j = K; j <= int(R.size()); j++){
                if (R[j] == 'W'){
                    W_cnt++;
                }
                if (R[j] == '?'){
                    q_cnt++;
                    if (q_cnt == 1 && pos == -1) pos = j;
                }
            }
            if (q_cnt == 1 && W_cnt == 0) R[pos] = 'W';
            cout << R << "\n";
        }
        else {
            
            int A_cnt = 0, W_cnt = 0, q_cnt = 0,pos = -1;
            for (int j = 0; j < K; j++){
                if (R[j] == 'A') A_cnt++;
                if (R[j] == 'W') W_cnt++;
                if (R[j] == '?') q_cnt++;
                if (pos == -1 && q_cnt == 1) pos = j;
            }
            if (W_cnt != 0){
                cout << R << "\n";
                continue;
                //Wが最初のK文字にあったら終了
            }
            
            for (int j = K; j < int(R.size()); j++){
                if (R[j] == 'W'){
                    W_cnt++;
                }
            }

            if (A_cnt == K){
                for (int j = K; j < int(R.size()); j++){
                    if (R[j] == '?') R[j] = 'A';
                }
                cout << R << "\n";
                continue;
            }
            // ここの時点で最初のK文字はAと?両方で構成されている奴だけ残る
            else {
                if (K == R.size()){
                    cout << R << "\n";
                    continue;
                }
                /*
                ココから調べること
                最初のK文字はAと?で構成されていて、K != R.size()であるから最初のK文字目にWが混合しているかと
                K+1文字目以降がすべてAであるかを調べる
                
                K文字目までに?が1個ならK+1文字目以降にwがあればW確定。AだけしかなかったらA確定。Wが無くて?があったら終了
                ?が2個以上あったら終了
                */
                if (q_cnt >= 2){
                    cout << R << "\n";
                    continue;
                }
                else if (q_cnt == 1){
                    if (W_cnt >= 1){
                        R[pos] = 'W';
                        cout << R << "\n";
                        continue;
                     }
                     else {
                        bool flag = false;
                        for (int j = K; j < int(R.size()); j++){
                            if (R[j] == '?') flag = true;
                        } 
                        if (flag){
                            cout << R << "\n";
                            continue;
                        }
                        else {
                            R[pos] = 'A';
                            cout << R << "\n";
                            continue;
                        }
                     }
                }
            }
        }
    }
    return 0;
}
0