hdu 2222 Keywords Search (AC自动机)

hdu 2222 Keywords Search

比较基础的AC自动机，适合初学练手

模板来自lrj白书

需要注意：1、数据中有相同的模板串 2、每个模板串只匹配一次

#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
#define MAXN 500005
#define MAXS 1000005
#define MAXSEG 26
struct acauto
{
	queue<int> q;
	int ch[MAXN][MAXSEG],next[MAXN],last[MAXN];
	int val[MAXN];
	int sz;
	void ini() {sz=1;memset(ch[0],0,sizeof(ch[0]));}
	int idx(char c) {return c-'a';}
	void insert(char *s,int v)
	{
		int pre=0,len=strlen(s);
		for(int i=0;i<len;i++)
		{
			int nex=idx(s[i]);
			if(!ch[pre][nex])
			{
				memset(ch[sz],0,sizeof(ch[sz]));
				val[sz]=0;
				ch[pre][nex]=sz++;
			}
			pre=ch[pre][nex];
		}
		val[pre]+=v;//有重复出现的模板串
	}
	int ans;
	void print(int j)
	{
		if(j)
		{
			ans+=val[j];
			val[j]=0;//每个只记录一次,用完之后清零
			print(last[j]);
		}
	}
	void find(char *t)
	{
		ans=0;
		int len=strlen(t);
		int j=0;
		for(int i=0;i<len;i++)
		{
			int nex=idx(t[i]);
			while(j&&!ch[j][nex]) j=next[j];
			j=ch[j][nex];
			if(val[j]) print(j);
			else if(last[j]) print(last[j]);
		}
	}
	void getnext()
	{
		while(!q.empty()) q.pop();
		next[0]=0;
		for(int nex=0;nex<MAXSEG;nex++)
		{
			int u=ch[0][nex];
			if(u) {next[u]=0;q.push(u);last[u]=0;}
		}
		while(!q.empty())
		{
			int pre=q.front();
			q.pop();
			for(int nex=0;nex<MAXSEG;nex++)
			{
				int u=ch[pre][nex];
				if(!u) continue;
				q.push(u);
				int v=next[pre];
				while(v&&!ch[v][nex]) v=next[v];
				next[u]=ch[v][nex];
				last[u]=val[next[u]]?next[u]:last[next[u]];
			}
		}
	}
};
char s[MAXS],t[70];
acauto ac;
int main()
{
	int cas,n;
	scanf("%d",&cas);
	while(cas--)
	{
		ac.ini();
		scanf("%d",&n);
		while(n--)
		{
			scanf("%s",t);
			ac.insert(t,1);
		}
		ac.getnext();
		scanf("%s",s);
		ac.find(s);
		printf("%d\n",ac.ans);
	}
	return 0;
}