結果

問題 No.2020 Sum of Common Prefix Length
ユーザー ぷらぷら
提出日時 2022-07-22 22:50:43
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 2,140 bytes
コンパイル時間 2,662 ms
コンパイル使用メモリ 223,640 KB
実行使用メモリ 42,748 KB
最終ジャッジ日時 2024-07-04 07:24:28
合計ジャッジ時間 17,200 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 WA -
testcase_02 WA -
testcase_03 RE -
testcase_04 RE -
testcase_05 RE -
testcase_06 RE -
testcase_07 RE -
testcase_08 RE -
testcase_09 RE -
testcase_10 RE -
testcase_11 RE -
testcase_12 RE -
testcase_13 RE -
testcase_14 RE -
testcase_15 WA -
testcase_16 WA -
testcase_17 RE -
testcase_18 RE -
testcase_19 RE -
testcase_20 RE -
testcase_21 RE -
testcase_22 RE -
testcase_23 RE -
testcase_24 RE -
testcase_25 RE -
testcase_26 RE -
testcase_27 AC 2 ms
5,376 KB
testcase_28 RE -
testcase_29 RE -
testcase_30 RE -
testcase_31 RE -
testcase_32 RE -
testcase_33 RE -
testcase_34 RE -
testcase_35 RE -
testcase_36 RE -
testcase_37 RE -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

constexpr int mod[] = {998244353, 1000000007};

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int N;
    cin >> N;
    map<pair<int, int>, int> mp;
    vector<long long> ans(N);
    vector<int> tmp(N);
    vector<vector<vector<int>>> hs(N, vector<vector<int>>(2, vector<int>(1, 0)));
    vector<string> S(N);
    for (int i = 0; i < N; i++) {
        cin >> S[i];
        for (int j = 0; j < S[i].size(); j++) {
            for (int k = 0; k < 2; k++) {
                hs[i][k].push_back((27ll * hs[i][k][j] + S[i][j] - 'a' + 1) % mod[k]);
            }
            mp[{hs[i][0][j], hs[i][1][j]}]++;
        }
    }
    for (int i = 0; i < N; i++) {
        for (int j = 0; j < S[i].size(); j++) {
            ans[i] += mp[{hs[i][0][j], hs[i][1][j]}];
        }
    }
    vector<pair<int,int>> res;
    int Q;
    cin >> Q;
    while (Q--) {
        int f;
        cin >> f;
        if (f == 1) {
            int x;
            char c;
            cin >> x >> c;
            x--;
            res.push_back({x,hs[x][0].size()});
            for (int j = 0; j < 2; j++) {
                hs[x][j].push_back((27ll * hs[x][j][hs[x][j].size()-1] + c - 'a' + 1) % mod[j]);
            }
            mp[{hs[x][0][hs[x][0].size() - 1], hs[x][1][hs[x][1].size() - 1]}]++;
            if (res.size() % 500 == 0) {
                for (int i = 0; i < N; i++) {
                    ans[i] = 0;
                    for (int j = 0; j < S[i].size(); j++) {
                        ans[i] += mp[{hs[i][j][0], hs[i][j][1]}];
                    }
                }
            }
        }
        else {
            int x;
            cin >> x;
            x--;
            int nans = ans[x];
            for(int i = res.size()/500*500; i < res.size(); i++) {
                if(hs[x][0].size() <= res[i].second) {
                    continue;
                }
                nans += (hs[x][0][res[i].second] == hs[res[i].first][0][res[i].second] && hs[x][1][res[i].second] == hs[res[i].first][1][res[i].second]);
            }
            cout << nans << '\n';
        }
    }
}
0