結果
問題 | No.430 文字列検索 |
ユーザー | sigma425 |
提出日時 | 2019-10-01 13:28:13 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 17 ms / 2,000 ms |
コード長 | 2,562 bytes |
コンパイル時間 | 2,119 ms |
コンパイル使用メモリ | 219,796 KB |
実行使用メモリ | 7,060 KB |
最終ジャッジ日時 | 2024-11-10 00:35:15 |
合計ジャッジ時間 | 2,798 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
5,248 KB |
testcase_01 | AC | 17 ms
7,060 KB |
testcase_02 | AC | 5 ms
5,248 KB |
testcase_03 | AC | 5 ms
5,248 KB |
testcase_04 | AC | 1 ms
5,248 KB |
testcase_05 | AC | 1 ms
5,248 KB |
testcase_06 | AC | 2 ms
5,248 KB |
testcase_07 | AC | 2 ms
5,248 KB |
testcase_08 | AC | 3 ms
5,248 KB |
testcase_09 | AC | 2 ms
5,248 KB |
testcase_10 | AC | 2 ms
5,248 KB |
testcase_11 | AC | 9 ms
5,888 KB |
testcase_12 | AC | 10 ms
6,400 KB |
testcase_13 | AC | 10 ms
6,400 KB |
testcase_14 | AC | 8 ms
5,680 KB |
testcase_15 | AC | 6 ms
5,248 KB |
testcase_16 | AC | 6 ms
5,248 KB |
testcase_17 | AC | 6 ms
5,248 KB |
ソースコード
#include <bits/stdc++.h> #define rep(i,n) for(int i=0;i<(int)(n);i++) #define rep1(i,n) for(int i=1;i<=(int)(n);i++) #define all(c) c.begin(),c.end() #define pb push_back #define fs first #define sc second #define chmin(x,y) x=min(x,y) #define chmax(x,y) x=max(x,y) using namespace std; template<class S,class T> ostream& operator<<(ostream& o,const pair<S,T> &p){ return o<<"("<<p.fs<<","<<p.sc<<")"; } template<class T> ostream& operator<<(ostream& o,const vector<T> &vc){ o<<"{"; for(const T& v:vc) o<<v<<","; o<<"}"; return o; } using ll = long long; template<class T> using V = vector<T>; template<class T> using VV = vector<vector<T>>; constexpr ll TEN(int n) { return (n == 0) ? 1 : 10 * TEN(n-1); } #ifdef LOCAL #define show(x) cerr << "LINE" << __LINE__ << " : " << #x << " = " << (x) << endl #else #define show(x) true #endif struct Aho{ using NP = Aho*; map<char,NP> nx; NP fail = nullptr, skip = nullptr; V<int> matched; //string ids for this node void add(const string& s, int id, int i = 0){ if(i == (int)s.size()){ matched.pb(id); return; } if(nx[s[i]] == nullptr){ nx[s[i]] = new Aho(); } nx[s[i]] -> add(s,id,i+1); } template<class OP> NP count(OP op, int r){ if(fail == nullptr) return this; //root now for(int id: matched){ op(id,r); } if(skip){ skip -> count(op,r); }else{ skip = fail -> count(op,r); } return matched.empty() ? skip : this; } template<class OP> void match(const string& s, OP op, int i = 0){ if(i == (int)s.size()) return; if(nx.count(s[i]) && nx[s[i]] != nullptr){ nx[s[i]] -> count(op,i+1); nx[s[i]] -> match(s,op,i+1); }else{ if(fail == nullptr){ //root now match(s,op,i+1); }else{ fail -> match(s,op,i); } } } static NP build(V<string> v){ NP r = new Aho(); rep(i,v.size()){ r -> add(v[i],i); } queue<NP> que; que.push(r); assert(r -> fail == nullptr); while(!que.empty()){ NP v = que.front(); que.pop(); for(auto e: v->nx){ auto c = e.fs; NP u = e.sc; NP fail = v -> fail; while(fail != nullptr && !fail->nx.count(c)){ fail = fail -> fail; } u -> fail = (fail == nullptr) ? r : fail -> nx[c]; que.push(u); } } return r; } }; int main(){ cin.tie(0); ios::sync_with_stdio(false); //DON'T USE scanf/printf/puts !! cout << fixed << setprecision(20); string s; int N; cin >> s >> N; V<string> patterns(N); rep(i,N) cin >> patterns[i]; auto AC = Aho::build(patterns); int cnt = 0; AC -> match(s,[&](int id,int r){ cnt++; }); cout << cnt << endl; }