AC自动机模板

摘自：https://www.luogu.com.cn/problem/solution/P3808

#include<cstdio>
#include<cstring>
#include<algorithm> 
#include<queue>
#define maxn 5000000+10
using namespace std;
char str[maxn*2];
struct node{
	int fail;//失配指针；
	int cnt;//单词出现的次数；
	int next[62];// 此节点的下一个（儿子）节点； 
}trie[maxn];//节点结构体； 
int k=0,ans=0;
queue<int> q;//队列：建失配指针使用； 
void build_trie(int id,char *s)//id表示第几个结点，即所有字符串从第0个节点开始向下建；*s即表示这个字符串； 
{
	int len=strlen(s);//该字符串的长度；(相当于字符串最后一个字符的深度)
	int j=0;
	for(int i=0;i<len;i++){
	    j=s[i]-'a';
	    if(trie[id].next[j]==0)/*若此字母未出现在当前位置的下一深度*/
	    {
	        trie[id].next[j]=++k;//当前节点对于j字母节点的位置；即j字母的节点序号； 
	    }
	    id=trie[id].next[j];//id传为下一字母的地址； 
	}
	trie[id].cnt++;//对此单词的数量++； 
}
void build_fail(int id)
{
	while(!q.empty()) q.pop();//为了放心，（be afraid of队列未清空）
	for(int i=0;i<26;i++)//遍历超级节点下的26个字母；
	{
	    int j=trie[id].next[i];
	    if(j!=0){
	        q.push(j);
	        trie[j].fail=id;//第一层的节点失配指针皆指向超级节点； 
	    }
	}
	while(!q.empty())
	{
	    int now=q.front();q.pop();//取出当前位置；（队首元素） 
	    for(int i=0;i<26;i++)
	    {
	        int j=trie[now].next[i];
	        if(j==0)//当前位置下没有这个节点；就调到它失配指针所指向的节点下的此个字母节点； 
	        {
	            trie[now].next[i]=trie[trie[now].fail].next[i];//若为0，不影响，指向超级节点; 
	            continue;//该点遍历完成，直接进入下一节点的遍历； 
	        }
	        trie[j].fail=trie[trie[now].fail].next[i];//如果当前位置下有这个字母节点，则其失配指针指向当前位置的失配指针下的该节点；
	        //若当前位置的失配指针下没有当前遍历的该字母节点，任不影响（为0，指向超级节点）； 
	        q.push(j);//存入数组； 
	    }
	}
}
void solve_trie(int id,char *s)//查询函数； 
{
	int len=strlen(s),j=0;
	for(int i=0;i<len;i++)
	{
	    int j=trie[id].next[s[i]-'a'];//当前位置的下一个节点位置；
	    while(j && trie[j].cnt!=-1)//当此节点存在同时其cnt未被遍历； 
	    {
	        ans+=trie[j].cnt;//将答案加上所搜索的字符串中所包含的该单词数 ； 
	        trie[j].cnt=-1;//标记； 
	        j=trie[j].fail;//直接将位置指向其失配指针的位置（节约时间）； 
	    }
	    id=trie[id].next[s[i]-'a'];//id继承，当前位置的下一个节点； 
	}
}
int main()
{
	int n;
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
	    scanf("%s",str);//第i个短字符串；
	    build_trie(0,str);//0为超级根节点（即所有字符串的共同祖先）；(即从0开始建) 
	}//字典树建立完成；
	build_fail(0);//建失配指针； 
	scanf("%s",str);//输入需查询的字符串；
	solve_trie(0,str); //查询； 
	printf("%d\n",ans);//输出； 
	return 0;
}