hdu 2222 ac自动机 模板

记录下网上学来的代码

#include <bits/stdc++.h>
#define maxn 1000008
#define tot 500005
using namespace std;
struct Aho{
struct state{
int next[26];
int fail,cnt;
}stateTable[tot];
queue<int>q;
int size;
void init(){
while(q.size())q.pop();
for(int i=0;i<tot;++i){
memset(stateTable[i].next,0,sizeof(stateTable[i].next));
stateTable[i].fail=stateTable[i].cnt=0;
}
size=1;
}
void insert(char *s){
int n=strlen(s);
int now=0;
for(int i=0;i<n;++i){
char c=s[i];
if(!stateTable[now].next[c-'a'])
stateTable[now].next[c-'a']=size++;
now=stateTable[now].next[c-'a'];
}
stateTable[now].cnt++;
}
void build(){
stateTable[0].fail=-1;
q.push(0);
while(q.size()){
int u=q.front();
q.pop();
for(int i=0;i<26;++i)
if(stateTable[u].next[i]){
if(u==0)stateTable[stateTable[u].next[i]].fail=0;
else {
int v=stateTable[u].fail;
while(v!=-1){
if(stateTable[v].next[i]){
stateTable[stateTable[u].next[i]].fail=stateTable[v].next[i];
break;
}
v=stateTable[v].fail;
}
if(v==-1)stateTable[stateTable[u].next[i]].fail=0;
}
q.push(stateTable[u].next[i]);
}
}
}
int Get(int u){
int res=0;
while(u){
res+=stateTable[u].cnt;
stateTable[u].cnt=0;
u=stateTable[u].fail;
}
return res;
}
int match(char *s){
int n=strlen(s);
int res=0,now=0;
for(int i=0;i<n;++i){
char c=s[i];
if(stateTable[now].next[c-'a'])
now=stateTable[now].next[c-'a'];
else {
int p=stateTable[now].fail;
while(p!=-1&&stateTable[p].next[c-'a']==0)p=stateTable[p].fail;
if(p==-1)now=0;
else now=stateTable[p].next[c-'a'];
}
if(stateTable[now].cnt)
res=res+Get(now);
}
return res;
}
}aho;
char s[maxn];
int main(){
int loop,n;
scanf("%d",&loop);
while(loop--){
scanf("%d",&n);
aho.init();
for(int i=1;i<=n;++i){
scanf("%s",s);
aho.insert(s);
}
aho.build();
scanf("%s",s);
printf("%d\n",aho.match(s));
}
return 0;
}