【模板】AC自动机

模板题

 

#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;
struct mzls
{
	int next[26],fail,count;
	void init()
	{
		memset(next,-1,sizeof(next));
		count=0;
		fail=0;
	}
}s[500005];
char s1[55],s2[1000005];
int n,len,id;
inline void ins()
{
	int nt=0;
	for(int i=0;i<len;i++)
	{
		int j=s1[i]-'a';
		if(s[nt].next[j]==-1)
		{
			s[++id].init();
			s[nt].next[j]=id;
		}
		nt=s[nt].next[j];
	}
	s[nt].count++;
}
inline void make_fail()
{
	queue<int>que;
	for(int i=0;i<26;i++)
		if(s[0].next[i]!=-1)
			que.push(s[0].next[i]);
	int nt;
	while(!que.empty())
	{
		int ml=que.front();
		que.pop();
		for(int i=0;i<26;i++)
			if(s[ml].next[i]!=-1)
			{
				int to=s[ml].next[i];
				que.push(to);
				nt=s[ml].fail;
				while(nt>0&&s[nt].next[i]==-1)
					nt=s[nt].fail;
				if(s[nt].next[i]!=-1)
					nt=s[nt].next[i];
				s[to].fail=nt;
			}
	}
}
inline int cal()
{
	int ans=0,nt=0;
	for(int i=0;i<len;i++)
	{
		int j=s2[i]-'a';
		while(nt>0&&s[nt].next[j]==-1)
			nt=s[nt].fail;
		if(s[nt].next[j]!=-1)
		{
			nt=s[nt].next[j];
			int ml=nt;
			while(ml&&s[ml].count!=-1)
			{
				ans+=s[ml].count;
				s[ml].count=-1;
				ml=s[ml].fail;
			}
		}
	}
	return ans;
}
int main()
{
	int T;
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d",&n);
		s[id=0].init();
		while(n--)
		{
			scanf("%s",s1);
			len=strlen(s1);
			ins();
		}
		make_fail();
		scanf("%s",s2);
		len=strlen(s2);
		printf("%d\n",cal());
	}
    return 0;
}