ac自动机代码模板

#include<bits/stdc++.h>
using namespace std;
const int maxn=1000006;
const int maxt=500005;

struct Aho
{
struct StateTable
{
int next[26];
int cnt,fail;
}stateTable[maxt];

queue<int> que;
int Size;

void init()
{
while(!que.empty())
que.pop();

for(int i=0;i<maxt;i++)
{
memset(stateTable[i].next,0,sizeof(stateTable[i].next));
stateTable[i].cnt=stateTable[i].fail=0;
}

Size=1;
}

void Insert(char *S)//插入待查找串s1,s2...sn
{
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;
que.push(0);

while(!que.empty())
{
int u=que.front();
que.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;
}
que.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 now=0,res=0;
int n=strlen(S);

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+=Get(now);
}
}
return res;
}

}aho;

int T,N;
char s[maxn];

int main()
{
ios::sync_with_stdio(false);
cin>>T;
while(T--)
{
cin>>N;
aho.init();
for(int i=0;i<N;i++)
{
cin>>s;
aho.Insert(s);
}
aho.build();
cin>>s;
cout<<aho.match(s)<<endl;
}
return 0;
}