結果

問題 No.2020 Sum of Common Prefix Length
ユーザー souta-1326
提出日時 2022-04-19 22:40:52
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,295 bytes
コンパイル時間 2,158 ms
コンパイル使用メモリ 205,760 KB
最終ジャッジ日時 2025-01-28 19:20:18
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 30 TLE * 8
権限があれば一括ダウンロードができます

ソースコード

diff # 

#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);
        nex.emplace_back(vector<int>(26,-1));
        cnt.emplace_back(vector<int>());
      }
      pos[x] = nex[pos[x]][z];
      ans[x] += cnt[pos[x]].size();
      cnt[pos[x]].emplace_back(x);
      for(int itr:cnt[pos[x]]) ans[itr]++;
    }
    else{
      cout << ans[x] << "\n";
    }
  }

}
0