文章标题 HDU 2222 : Keywords Search （AC自动机模板）

#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <queue>
using namespace std;
const int maxn=500000+10;
int n;
int ch[maxn][26];//trie中每个点的儿子节点编号
int fail[maxn];
int last[maxn];//后缀链接：沿着i的失配指针走遇到的下一个单词节点的编号
int val[maxn];//点上的标记（比如标记是否是单词节点）
int sz;
int root;
int newnode (){//新节点
memset (ch[sz],0,sizeof (ch[sz]));//将其孩子置为0
val[sz]=0;
return sz++;
}
void init(){//初始化
sz=0;
root=newnode();
}
int insert(char *str){//建树
int len=strlen(str);
int now=root;
for (int i=0;i<len;i++){
int c=str[i]-'a';
if (!ch[now][c]){
memset (ch[sz],0,sizeof (ch[sz]));
val[sz] = 0;
ch[now][c] = sz++;
}
now=ch[now][c];
}
val[now]++;
return now;
}
void getfail(){
queue<int>q;
fail[root]=root;
for (int i=0;i<26;i++){
int u=ch[root][i];
if (u){
fail[u]=0;
last[u]=0;
q.push(u);
}
}
while (!q.empty()){
int now=q.front();q.pop();
for (int i=0;i<26;i++){
int u=ch[now][i];
//u=ch[now][i]就是我们说的当前节点A，now是A的父节点，
//那么fail[now]就相当于我们说的父节点的失败节点C
if (!u){
/*
如果当前节点now没有i这个儿子u，
直接让这个点的第i个儿子u指向now的失败节点的第i个儿子。
因为如果一个串在AC自动机里面走，走到now时要走的下一个节点是i，
now又没有i这个儿子，因此一定会是失配的。
既然失配了，now就要沿着fail指针走，试图去匹配失败节点的第i个儿子。
这里的ch[now][i]=ch[fail[now]][i]就相当于
让他直接去匹配now的失败节点的第i个儿子
*/
ch[now][i]=ch[fail[now]][i];
}
else {
//如果当前节点now有i这个儿子u，
//那么u的失败节点显然要指向C(即fail[now])的第i个儿子
fail[u]=ch[fail[now]][i];
last[u]=val[fail[u]]?fail[u]:last[fail[u]];
q.push(u);
}
}
}
}
int quary(char *str){
int len=strlen(str);
int now=root;
int ans=0;
for (int i=0;i<len;i++){
now=ch[now][str[i]-'a'];
int tmp=now;
if (tmp!=root&&!val[tmp]&&last[tmp]){
tmp=last[tmp];
}
while (tmp!=root&&val[tmp]){
ans+=val[tmp];
val[tmp]=0;
tmp=last[tmp];
}
}
return ans;
}
char text[1000005];
char str[55];
int main()
{
int t;
scanf("%d",&t);
while (t--){
init();
scanf ("%d",&n);
for (int i=0;i<n;i++){
scanf ("%s",str);
insert (str);
}
getfail();
scanf ("%s",text);
printf ("%d\n",quary(text));
}
return 0;
}
/*
100
2
c
abcd
abcde
*/