結果

問題 No.2231 Surprising Flash!
ユーザー 👑 AngrySadEightAngrySadEight
提出日時 2023-02-14 21:49:46
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 4,473 bytes
コンパイル時間 3,662 ms
コンパイル使用メモリ 158,168 KB
実行使用メモリ 100,440 KB
最終ジャッジ日時 2024-09-13 07:19:50
合計ジャッジ時間 25,955 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 4 ms
6,940 KB
testcase_02 AC 27 ms
6,944 KB
testcase_03 AC 377 ms
6,944 KB
testcase_04 AC 17 ms
6,940 KB
testcase_05 AC 37 ms
6,944 KB
testcase_06 AC 49 ms
6,944 KB
testcase_07 AC 3 ms
6,940 KB
testcase_08 AC 10 ms
6,940 KB
testcase_09 AC 2 ms
6,940 KB
testcase_10 AC 11 ms
6,940 KB
testcase_11 AC 698 ms
77,128 KB
testcase_12 AC 698 ms
100,388 KB
testcase_13 AC 709 ms
100,440 KB
testcase_14 AC 641 ms
66,664 KB
testcase_15 AC 697 ms
77,232 KB
testcase_16 AC 667 ms
6,944 KB
testcase_17 AC 691 ms
10,448 KB
testcase_18 AC 691 ms
66,444 KB
testcase_19 AC 677 ms
66,716 KB
testcase_20 AC 688 ms
66,504 KB
testcase_21 AC 687 ms
66,440 KB
testcase_22 AC 687 ms
66,500 KB
testcase_23 AC 683 ms
66,668 KB
testcase_24 AC 684 ms
66,560 KB
testcase_25 AC 686 ms
66,508 KB
testcase_26 AC 684 ms
66,568 KB
testcase_27 AC 682 ms
66,596 KB
testcase_28 AC 683 ms
66,628 KB
testcase_29 AC 720 ms
77,192 KB
testcase_30 AC 725 ms
77,188 KB
testcase_31 AC 722 ms
77,064 KB
testcase_32 AC 723 ms
77,132 KB
testcase_33 AC 723 ms
77,164 KB
testcase_34 AC 643 ms
77,208 KB
testcase_35 AC 652 ms
77,072 KB
testcase_36 AC 650 ms
77,072 KB
testcase_37 AC 651 ms
77,180 KB
testcase_38 AC 652 ms
77,180 KB
testcase_39 AC 272 ms
9,996 KB
testcase_40 AC 186 ms
10,140 KB
testcase_41 AC 2 ms
6,944 KB
testcase_42 WA -
testcase_43 WA -
testcase_44 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <string>
#include <vector>
#include <atcoder/string>
#include <atcoder/convolution>
#include <atcoder/segtree>
using namespace std;
using namespace atcoder;
typedef long long ll;

template<typename T> void debug(vector<T> &v){
    for (ll i = 0; i < v.size(); i++){
        cerr << v[i] << " ";
    }
    cerr << endl;
}

ll ctol(char c){
    ll ret;
    if (c == '?') ret = 0;
    else{
        ret = (ll)(c - 'a' + 1);
    }
    return ret;
}

vector<ll> wild_card_matching(string &S, string &T){
    ll len_S = S.size();
    ll len_T = T.size();
    vector<vector<ll>> pow_S(4, vector<ll>(len_S));
    vector<vector<ll>> pow_T(4, vector<ll>(len_T));
    for (ll i = 0; i < len_S; i++){
        pow_S[1][i] = ctol(S[i]);
        for (ll j = 1; j < 3; j++){
            pow_S[j + 1][i] = pow_S[j][i] * pow_S[1][i];
        }
    }
    for (ll i = 0; i < len_T; i++){
        pow_T[1][i] = ctol(T[i]);
        for (ll j = 1; j < 3; j++){
            pow_T[j + 1][i] = pow_T[j][i] * pow_T[1][i];
        }
    }
    for (ll i = 0; i < 4; i++){
        reverse(pow_T[i].begin(), pow_T[i].end());
    }
    vector<ll> vec_conv1 = convolution<998244353>(pow_S[3], pow_T[1]);
    vector<ll> vec_conv2 = convolution<998244353>(pow_S[2], pow_T[2]);
    vector<ll> vec_conv3 = convolution<998244353>(pow_S[1], pow_T[3]);
    vector<ll> vec_conv4 = convolution<469762049>(pow_S[3], pow_T[1]);
    vector<ll> vec_conv5 = convolution<469762049>(pow_S[2], pow_T[2]);
    vector<ll> vec_conv6 = convolution<469762049>(pow_S[1], pow_T[3]);

    vector<ll> ret(0);
    for (ll i = 0; i < len_S - len_T + 1; i++){
        if ((vec_conv1[len_T - 1 + i] - 2 *  vec_conv2[len_T - 1 + i] + vec_conv3[len_T - 1 + i] + 998244353 * 2) % 998244353 == 0){
            if ((vec_conv4[len_T - 1 + i] - 2 * vec_conv5[len_T - 1 + i] + vec_conv6[len_T - 1 + i] + 469762049 * 2) % 469762049 == 0) ret.push_back(i);
        }
    }
    return ret;
}

ll op(ll a, ll b){
    return min(a, b);
}

ll e(){
    return 1e9;
}

int main(){
    ll T;
    cin >> T;
    while(T--){
        ll N, M;
        cin >> N >> M;
        string S1;
        cin >> S1;
        string S2;
        cin >> S2;

        vector<ll> wcm = wild_card_matching(S1, S2);
        if (wcm.size() == 0){
            cout << -1 << endl;
            continue;
        }

        //debug(wcm);

        string S3 = "";
        for (ll i = 0; i < N; i++){
            if (S1[i] == '?') S3.push_back('a');
            else S3.push_back(S1[i]);
        }

        //S3 + S2 の LCP を求める
        vector<int> sa = suffix_array(S3 + S2);
        vector<int> lcp = lcp_array(S3 + S2, sa);
        vector<int> rev_sa(N + M);
        for (int i = 0; i < N + M; i++){
            rev_sa[sa[i]] = i;
        }
        segtree<ll, op, e> seg(N + M);
        for (ll i = 0; i < N + M; i++){
            seg.set(i, lcp[i]);
        }
        ll idx = 0;
        for (ll i = 1; i < wcm.size(); i++){
            ll left, right;
            ll diffs = wcm[i] - wcm[idx];
            if (diffs >= M){
                idx = i;
                continue;
            }
            left = min(rev_sa[wcm[idx]], rev_sa[N]);
            right = max(rev_sa[wcm[idx]], rev_sa[N]);
            ll lcp1 = seg.prod(left, right);
            if (lcp1 < diffs){
                if (S3[wcm[i] + lcp1] < S2[lcp1]){
                    idx = i;
                    continue;
                }
                else{
                    continue;
                }
            }
            left = min(rev_sa[N], rev_sa[N + diffs]);
            right = max(rev_sa[N], rev_sa[N + diffs]);
            ll lcp2 = seg.prod(left, right);
            if (lcp2 < M - diffs){
                if (S2[lcp2] < S2[diffs + lcp2]){
                    idx = i;
                    continue;
                }
                else{
                    continue;
                }
            }
            left = min(rev_sa[wcm[idx] + M], rev_sa[N + M - diffs]);
            right = max(rev_sa[wcm[idx] + M], rev_sa[N + M - diffs]);
            ll lcp3 = seg.prod(left, right);
            if (lcp3 < diffs){
                if (S2[M - diffs + lcp3] < S3[wcm[idx] + M + lcp3]){
                    idx = i;
                    continue;
                }
                else{
                    continue;
                }
            }
        }
        cout << S3.substr(0, wcm[idx]) << S2 << S3.substr(wcm[idx] + M, N - wcm[idx]) << endl;
    }
}
0