結果
問題 | No.430 文字列検索 |
ユーザー | kazuma |
提出日時 | 2017-07-25 23:12:35 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 35 ms / 2,000 ms |
コード長 | 1,893 bytes |
コンパイル時間 | 2,159 ms |
コンパイル使用メモリ | 212,476 KB |
実行使用メモリ | 26,880 KB |
最終ジャッジ日時 | 2024-11-10 00:17:25 |
合計ジャッジ時間 | 3,007 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
5,248 KB |
testcase_01 | AC | 35 ms
26,880 KB |
testcase_02 | AC | 11 ms
9,344 KB |
testcase_03 | AC | 10 ms
9,344 KB |
testcase_04 | AC | 1 ms
5,248 KB |
testcase_05 | AC | 2 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 | 3 ms
5,248 KB |
testcase_10 | AC | 2 ms
5,248 KB |
testcase_11 | AC | 27 ms
18,816 KB |
testcase_12 | AC | 30 ms
21,248 KB |
testcase_13 | AC | 31 ms
21,248 KB |
testcase_14 | AC | 22 ms
16,768 KB |
testcase_15 | AC | 17 ms
12,800 KB |
testcase_16 | AC | 16 ms
12,800 KB |
testcase_17 | AC | 15 ms
12,800 KB |
ソースコード
#include <bits/stdc++.h> using namespace std; using ll = long long; class Aho_Corasick { struct node { node *no; vector<node*> next; vector<int> matched; node() : no(nullptr), next(128, nullptr) {} ~node() { for (auto ite : next) if (ite != nullptr) delete ite; } }; vector<int> unite(const vector<int>& a, const vector<int>& b) { vector<int> res; set_union(a.begin(), a.end(), b.begin(), b.end(), back_inserter(res)); return res; } int K; node *root; public: Aho_Corasick(const vector<string>& Ts) : K(Ts.size()), root(new node) { node *now; root->no = root; for (int i = 0; i < K; i++) { auto &T = Ts[i]; now = root; for (auto c : T) { if (now->next[c] == nullptr) { now->next[c] = new node; } now = now->next[c]; } now->matched.push_back(i); } queue<node*> q; for (int i = 0; i < 128; i++) { if (root->next[i] == nullptr) { root->next[i] = root; } else { root->next[i]->no = root; q.push(root->next[i]); } } while (!q.empty()) { now = q.front(); q.pop(); for (int i = 0; i < 128; i++) { if (now->next[i] != nullptr) { node *nx = now->no; while (nx->next[i] == nullptr) { nx = nx->no; } now->next[i]->no = nx->next[i]; now->next[i]->matched = unite(now->next[i]->matched, nx->next[i]->matched); q.push(now->next[i]); } } } } vector<int> count(const string& S) { vector<int> res(K); node *now = root; for (auto c : S) { while (now->next[c] == nullptr) { now = now->no; } now = now->next[c]; for (auto k : now->matched) { res[k]++; } } return res; } }; int main() { string S; int M; cin >> S >> M; vector<string> C(M); for (int i = 0; i < M; i++) { cin >> C[i]; } Aho_Corasick aho(C); auto cnt = aho.count(S); ll res = 0; for (auto t : cnt) { res += t; } cout << res << endl; return 0; }