結果

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

ソースコード

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>,vector<int>>mp;
    vector<long long>ans(N);
    vector<int>tmp(N);
    vector<vector<int>>hs(N,vector<int>(2,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] = (27ll*hs[i][k]+S[i][j]-'a')%mod[k];
            }
            mp[{hs[i][0],hs[i][1]}].push_back(i);
        }
    }
    for(int i = 0; i < N; i++) {
        hs[i][0] = hs[i][1] = 0;
        for(int j = 0; j < S[i].size(); j++) {
            for(int k = 0; k < 2; k++) {
                hs[i][k] = (27ll*hs[i][k]+S[i][j]-'a')%mod[k];
            }
            ans[i] += mp[{hs[i][0],hs[i][1]}].size();
        }
    }
    int Q;
    cin >> Q;
    while(Q--) {
        int f;
        cin >> f;
        if(f == 1) {
            int x;
            char c;
            cin >> x >> c;
            x--;
            for(int j = 0; j < 2; j++) {
                hs[x][j] = (27ll*hs[x][j]+c-'a')%mod[j];
            }
            mp[{hs[x][0],hs[x][1]}].push_back(x);
            for(int i:mp[{hs[x][0],hs[x][1]}]) {
                ans[i]++;
            }
        }
        else {
            int x;
            cin >> x;
            x--;
            cout << ans[x] << '\n';
        }
    }
}
0