洛谷 P3808 【模板】AC自动机（简单版） 题解

题目：P3808 【模板】AC自动机（简单版）

#include<cstdio>
#include<iostream>
#include<cstring>
#include<string>
#include<queue>
using namespace std;
const int Maxn=500000+2333,Maxm=30,Maxlen=1000000+10,inf=0x3f3f3f3f;
int g[Maxn][Maxm],nxt[Maxn],c[Maxn];
int n,ans,idcnt;
inline int read()
{
	int s=0,w=1;
	char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
	while(ch>='0' && ch<='9')s=(s<<3)+(s<<1)+(ch^48),ch=getchar();
	return s*w;
}
void insert(char *s)
{
	int cur=0,len=strlen(s); 
	for(int i=0;i<len;++i)
	{
		int x=s[i]-'a';
		if(!g[cur][x])g[cur][x]=++idcnt;
		cur=g[cur][x];
	} //Trie 储存字符串
	c[cur]++; // 标记该处是一个字符串的结尾
}
void build()
{
	queue <int> q;
	for(int i=0;i<26;++i)
	{
		if(!g[0][i])continue;
		q.push(g[0][i]);
		//将所有字符串加入队列，以bfs的方式赋值nxt
	}
	while(q.size())
	{
		int x=q.front();
		q.pop();
		for(int i=0;i<26;++i)
		{
			if(g[x][i])
			{
				nxt[g[x][i]]=g[nxt[x]][i];
				q.push(g[x][i]);
			}
			else g[x][i]=g[nxt[x]][i]; // 与上同理
		}
	}
}
void calc(char *s)
{
	int cur=0,len=strlen(s);
	for(int i=0;i<len;++i)
	{
		cur=g[cur][s[i]-'a'];
		for(int x=cur;x && c[x]!=-1;x=nxt[x])
		ans+=c[x],c[x]=-1;
		// 注意这里一定要赋值为 -1 ，若赋值成 0 的话，一遇到0就停止（有可能字符串还没搜完），就会漏掉很多字符串
	}
}
int main()
{
//	freopen("in.txt","r",stdin);
	n=read();
	for(int i=1;i<=n;++i)
	{
		char a[Maxlen];
		scanf("%s",a);
		insert(a);
	}
	build();
	char a[Maxlen];
	scanf("%s",a);
	calc(a);
	printf("%d\n",ans);
	return 0;
}