結果
問題 | No.2020 Sum of Common Prefix Length |
ユーザー |
|
提出日時 | 2022-04-19 22:34:08 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
RE
|
実行時間 | - |
コード長 | 1,214 bytes |
コンパイル時間 | 2,183 ms |
コンパイル使用メモリ | 206,920 KB |
最終ジャッジ日時 | 2025-01-28 19:19:32 |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 1 WA * 4 RE * 33 |
ソースコード
#include<bits/stdc++.h>using namespace std;int main(){int N;cin >> N;vector<string> S(N);for(string &s:S) cin >> s;vector<char> node({0});vector<vector<int>> nex(1,vector<int>(26,-1)),cnt(1);for(int i=0;i<N;i++){int now = 0;cnt[0].emplace_back(i);for(char c:S[i]){int z = c-'a';if(nex[now][z] == -1){nex[now][z] = node.size();node.emplace_back(c);nex.emplace_back(vector<int>(26,-1));cnt.emplace_back(vector<int>());}now = nex[now][z];cnt[now].emplace_back(i);}}vector<int> ans(N),pos(N);for(int i=0;i<N;i++){int now = 0;for(char c:S[i]){int z = c-'a';now = nex[now][z];ans[i] += cnt[now].size();}pos[i] = now;}int Q;cin >> Q;for(int i=0;i<Q;i++){int t,x;cin >> t >> x;x--;if(t == 1){char c;cin >> c;int z = c-'a';if(nex[pos[x]][z] == -1){nex[pos[x]][z] = node.size();node.emplace_back(c);cnt.emplace_back(vector<int>());}pos[x] = nex[pos[x]][z];ans[x] += cnt[pos[x]].size();for(int itr:cnt[pos[x]]) ans[itr]++;}else{cout << ans[x] << "\n";}}}