模板 AC自动机

板子题目：

n个串以及1个总串，求出n个串内在这个总串内（整串）出现的个数。

即HDU2222

AC自动机的模板题目。

AC自动机主要就是fail指针的巧妙性。。

什么是AC自动机我就不说了，网上资料也是很多的……

在这里就给出我一直在用的代码吧。

首先当然是把那n个串放入trie树了……然后匹配。。

中间AC()也就是求fail的过程值得记一记。

感觉这份代码还是挺好的。

#include<bits/stdc++.h>
using namespace std;
int n,sz,len;
int trie[1000005][26],val[1000005],fail[1000005],flag[1000005];
char ty1[55],ty2[1000005];
int Q[50005];
int find(int x,int t){
if (trie[x][t]) return trie[x][t];
if (!x) return 0;
return find(fail[x],t);
}        //类似于KMP的find函数
void insert(){
int now=0;
int len=strlen(ty1);
for (int i=0;i<len;i++){
int t=ty1[i]-'a';
if (trie[now][t]) now=trie[now][t];
else now=trie[now][t]=++sz;
}
val[now]++;        //在now上的单词个数val[now]
}
void AC(){     //广搜求出fail指针
int head=0,tail=1;
Q[0]=0;
while (head<tail){
int now=Q[head++];
for (int i=0;i<26;i++){
if (!trie[now][i]) continue;
int t=find(fail[now],i);
if (!now) t=0;
fail[trie[now][i]]=t;
Q[tail++]=trie[now][i];
}
}
}
void solve(){
int t=0,ans=0;
for (int i=0;i<len;i++){
int tt=ty2[i]-'a';
flag[t]=1;
t=find(t,tt);
if (flag[t]) continue;
for (int j=t;j;j=fail[j])
ans+=val[j],val[j]=0;
}
printf("%d\n",ans);
}
int main(){
int fil;
scanf("%d",&fil);    //多组数据
while (fil--){
memset(val,0,sizeof(val));
memset(trie,0,sizeof(trie));
memset(fail,0,sizeof(fail));
memset(flag,0,sizeof(flag));
scanf("%d",&n);
sz=0;
for (int i=1;i<=n;i++){
scanf("%s",ty1);
insert();        //放入trie树
}
AC();
scanf("%s",ty2);
len=strlen(ty2);
solve();
}
return 0;
}