AC自动机 | 易学教程

例题：洛谷 P3808 AC自动机 1 #include <bits/stdc++.h>
 2 using namespace std;
 3 
 4 struct Tree { //字典树
 5     int fail;//失配指针
 6     int vis[26];//子节点的位置
 7     int end;//标记有几个单词以这个节点结尾
 8 } AC[1000000];//Trie树
 9 
10 int cnt = 0;//Trie的指针
11 int n;
12 string s;
13 
14 void Build(string s) {
15     int l = s.size();
16     int now = 0;//字典树的当前指针
17     for (int i = 0; i < l; i++) {//构造Trie树
18         if (AC[now].vis[s[i] - 'a'] == 0)//Trie树没有这个子节点
19             AC[now].vis[s[i] - 'a'] = ++cnt;//构造出来
20         now = AC[now].vis[s[i] - 'a'];//向下构造
21     }
22     AC[now].end += 1;//标记单词结尾
23 }
24 void Get_fail() {//构造fail指针
25     queue<int> Q;//队列
26     for (int i = 0; i < 26; i++) {//第二层的fail指针提前处理一下
27         if (AC[0].vis[i] != 0) {
28             AC[AC[0].vis[i]].fail = 0;//指向根节点
29             Q.push(AC[0].vis[i]);//压入队列
30         }
31     }
32     while (!Q.empty()) {//BFS求fail指针
33         int u = Q.front();
34         Q.pop();
35         for (int i = 0; i < 26; i++) {//枚举所有子节点
36             if (AC[u].vis[i] != 0) {//存在这个子节点
37                 AC[AC[u].vis[i]].fail = AC[AC[u].fail].vis[i];//子节点的fail指针指向当前节点的的相同子节点
38                 Q.push(AC[u].vis[i]);//压入队列
39             }
40             else //不存在这个子节点
41                 AC[u].vis[i] = AC[AC[u].fail].vis[i];//当前节点的这个子节点指向当前节点fail指针的这个子节点
42         }
43     }
44 }
45 int AC_Query(string s) { //AC自动机匹配
46     int l = s.length();
47     int now = 0, ans = 0;
48     for (int i = 0; i < l; i++) {
49         now = AC[now].vis[s[i] - 'a'];//向下一层
50         for (int t = now; t > 0 && AC[t].end != -1; t = AC[t].fail) {//循环求解
51             ans += AC[t].end;
52             AC[t].end = -1;
53         }
54     }
55     return ans;
56 }
57 int main() {
58     scanf("%d", &n);
59     for (int i = 1; i <= n; i++) {
60         cin>>s;
61         Build(s);
62     }
63     AC[0].fail = 0;//结束标志
64     Get_fail();//求出失配指针
65     cin >> s;//文本串
66     cout << AC_Query(s) << endl;
67     return 0;
68 }
来源：https://www.cnblogs.com/Z-MIKI/p/12257321.html
标签
自动机