ac自動機模板

ac自動機是一種解決多模匹配的問題

KMP+Trie樹

直接優化建成Trie圖

hdu2222

#include<cstdio>
#include<iostream>
#include<cstring>
#include<queue>
using namespace std;
int n;
char s[55];
int edge=0;
int son[1000005][26];
int pd[1000005];
int fail[1000005];
queue<int> q;
char dd[1000005];
int main()
{
	int m;
	scanf("%d",&m);
	while(m--)
	{
		memset(fail,0,sizeof(fail));
		memset(pd,0,sizeof(pd));
		memset(son,0,sizeof(son));
		edge=0;
		while(!q.empty())q.pop();
		scanf("%d",&n);
		for(int i=1;i<=n;i++)
		{
			scanf("%s",s);
			int now=0;
			int len=strlen(s);
			for(int j=0;j<len;j++)
			{
				if(!son[now][s[j]-'a'])
				{
					edge++;
					son[now][s[j]-'a']=edge;
				}
				now=son[now][s[j]-'a'];	
			}
			pd[now]++;
		}
		fail[0]=0;
		for(int i=0;i<26;i++)
		{
			if(son[0][i])q.push(son[0][i]);
		}
		while(!q.empty())
		{
			int x=q.front();
			q.pop();
			for(int i=0;i<26;i++)
			{
				if(son[x][i])
				{
					fail[son[x][i]]=son[fail[x]][i];
					q.push(son[x][i]);
				}
				else son[x][i]=son[fail[x]][i];
			}
		}
		scanf("%s",dd);
		int ne=0;
		int ans=0;
		int len=strlen(dd);
		for(int i=0;i<len;i++)
		{
			ne=son[ne][dd[i]-'a'];
			for(int j=ne;j&&pd[j]!=-1;j=fail[j])
			{
				ans+=pd[j];
				pd[j]=-1;
			}
		}
		printf("%d\n",ans);
	}
	return 0;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章