結果

問題 No.2020 Sum of Common Prefix Length
ユーザー ぷらぷら
提出日時 2022-07-22 22:50:43
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,140 bytes
コンパイル時間 3,154 ms
コンパイル使用メモリ 215,380 KB
最終ジャッジ日時 2025-01-30 12:53:18
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 1 WA * 5 RE * 32
権限があれば一括ダウンロードができます

ソースコード

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