結果
問題 | No.2020 Sum of Common Prefix Length |
ユーザー | souta-1326 |
提出日時 | 2022-04-21 22:20:54 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 182 ms / 2,000 ms |
コード長 | 2,144 bytes |
コンパイル時間 | 5,109 ms |
コンパイル使用メモリ | 270,996 KB |
実行使用メモリ | 70,364 KB |
最終ジャッジ日時 | 2023-09-05 15:39:14 |
合計ジャッジ時間 | 10,867 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge14 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
4,376 KB |
testcase_01 | AC | 2 ms
4,380 KB |
testcase_02 | AC | 1 ms
4,380 KB |
testcase_03 | AC | 3 ms
4,376 KB |
testcase_04 | AC | 3 ms
4,376 KB |
testcase_05 | AC | 3 ms
4,376 KB |
testcase_06 | AC | 3 ms
4,380 KB |
testcase_07 | AC | 74 ms
9,952 KB |
testcase_08 | AC | 71 ms
10,016 KB |
testcase_09 | AC | 72 ms
9,968 KB |
testcase_10 | AC | 72 ms
9,956 KB |
testcase_11 | AC | 74 ms
9,964 KB |
testcase_12 | AC | 74 ms
9,956 KB |
testcase_13 | AC | 74 ms
10,008 KB |
testcase_14 | AC | 76 ms
10,156 KB |
testcase_15 | AC | 116 ms
28,348 KB |
testcase_16 | AC | 114 ms
28,552 KB |
testcase_17 | AC | 110 ms
35,176 KB |
testcase_18 | AC | 94 ms
25,996 KB |
testcase_19 | AC | 100 ms
27,940 KB |
testcase_20 | AC | 159 ms
70,364 KB |
testcase_21 | AC | 174 ms
63,560 KB |
testcase_22 | AC | 170 ms
62,972 KB |
testcase_23 | AC | 168 ms
61,948 KB |
testcase_24 | AC | 181 ms
70,236 KB |
testcase_25 | AC | 182 ms
69,412 KB |
testcase_26 | AC | 99 ms
40,808 KB |
testcase_27 | AC | 2 ms
4,376 KB |
testcase_28 | AC | 98 ms
16,708 KB |
testcase_29 | AC | 96 ms
17,292 KB |
testcase_30 | AC | 96 ms
16,288 KB |
testcase_31 | AC | 148 ms
64,900 KB |
testcase_32 | AC | 152 ms
69,152 KB |
testcase_33 | AC | 110 ms
40,580 KB |
testcase_34 | AC | 122 ms
27,824 KB |
testcase_35 | AC | 114 ms
27,820 KB |
testcase_36 | AC | 126 ms
49,796 KB |
testcase_37 | AC | 131 ms
37,524 KB |
ソースコード
#include<bits/stdc++.h> #include<atcoder/all> using namespace std; using namespace atcoder; #define emb emplace_back using ll = long long; class EulerTour{ int N,cnt; vector<int> in,out; fenwick_tree<int> B_v; public: EulerTour(vector<vector<int>> &G):N(G.size()),cnt(0),in(N),out(N),B_v(N*2){ int f = 0; vector<int> itr(N),par(N); par[f] = -1; while(f != -1){ if(itr[f] == 0) in[f] = cnt++; if(itr[f] == G[f].size()){ out[f] = cnt++; f = par[f]; continue; } par[G[f][itr[f]]] = f; f = G[f][itr[f]++]; } } void add(int p,ll num=1){B_v.add(in[p],num);B_v.add(out[p],-num);} ll query(int p){return B_v.sum(0,in[p]+1);} }; int main(){ cin.tie(0);ios::sync_with_stdio(false); //入力 int N;cin >> N; vector<string> S(N); for(string &s:S) cin >> s; int Q;cin >> Q; vector<int> T(Q),X(Q);vector<char> C(Q);//Tはクエリのタイプ for(int i=0;i<Q;i++){ cin >> T[i] >> X[i];X[i]--; if(T[i] == 1) cin >> C[i]; } //クエリ全処理後のTrie木をつくる vector<vector<int>> path(N,vector<int>({0}));//各文字列の文字ごとのnodeのindex vector<char> node({0}); vector<vector<int>> nex(1,vector<int>(26,-1)); vector<string> S2 = S; for(int i=0;i<Q;i++){ if(T[i] == 1) S2[X[i]] += C[i]; } for(int i=0;i<N;i++){ int now_node = 0; for(char c:S2[i]){ int z = c-'a'; if(nex[now_node][z] == -1){ nex[now_node][z] = node.size(); node.emplace_back(c); nex.emplace_back(vector<int>(26,-1)); } now_node = nex[now_node][z]; path[i].emplace_back(now_node); } } //ただの根付き木にする int V = node.size(); vector<vector<int>> G(V); for(int i=0;i<V;i++){ for(int j=0;j<26;j++){ if(nex[i][j] != -1) G[i].emplace_back(nex[i][j]); } } EulerTour Eul(G); for(int i=0;i<N;i++){ for(int j=0;j<S[i].size();j++) Eul.add(path[i][j+1]); } for(int i=0;i<Q;i++){ if(T[i] == 1){ S[X[i]] += C[i];Eul.add(path[X[i]][S[X[i]].size()]); } else cout << Eul.query(path[X[i]][S[X[i]].size()]) << "\n"; } }