hdu 5384 Danganronpa (ac自动机)

题目：http://acm.hdu.edu.cn/showproblem.php?pid=5384

题意：很直接的多模式匹配，不过这里的文本串有多个。

分析：建ac机的时候，把到达每个节点的贡献算出来，这样匹配的时候就不用在ac机上回溯了。时间复杂度O（|A|+|B|）。

代码：

#include <iostream>
#include <cstring>
#include <string>
#include <cstdio>
#include <queue>
using namespace std;
const int maxn = 1e5+6;
const int kd = 26;
struct trie
{
	int son[maxn][kd],fail[maxn],end[maxn];
	int cnt,root;
	int newnode()
	{
		fill(son[cnt],son[cnt]+kd,-1);
		end[cnt]=0;
		return cnt++;
	}
	void Init()
	{
		cnt=0;
		root=newnode();
	}
	void Insert(char str[])
	{
		int index,now=root,i;
		for(i=0;str[i];i++)
		{
			index=str[i]-'a';
			if(son[now][index]==-1)
				son[now][index]=newnode();
			now=son[now][index];
		}
		++end[now];
	}
	void findfail()
	{
		queue <int > q;
		fail[root]=root;
		int i,temp;
		for(i=0;i<kd;i++)
			if(son[root][i]==-1)
				son[root][i]=root;
			else
			{
				fail[son[root][i]]=root;
				q.push(son[root][i]);
			}
		while(!q.empty())
		{
			temp=q.front();
			q.pop();
			end[temp]+=end[fail[temp]];
			for(i=0;i<kd;i++)
				if(son[temp][i]==-1)
					son[temp][i]=son[fail[temp]][i];
				else
				{
					fail[son[temp][i]]=son[fail[temp]][i];
					q.push(son[temp][i]);
				}
		}
	}
	int Match(string &s)
	{
		int index,now=root,i,sz=s.size(),ret=0;
		for(i=0;i<sz;i++)
		{
			index=s[i]-'a';
			now=son[now][index];
			ret+=end[now];
		}
		return ret;
	}
}ac;
string A[maxn];
char B[maxn];
int main()
{
	int ncase,n,m,i;
	scanf("%d",&ncase);
	while(ncase--)
	{
		scanf("%d%d",&n,&m);
		for(i=0;i<n;i++)
			cin>>A[i];
		ac.Init();
		for(i=0;i<m;i++)
		{
			scanf("%s",B);
			ac.Insert(B);
		}
		ac.findfail();
		for(i=0;i<n;i++)
			printf("%d\n",ac.Match(A[i]));
	}
	return 0;
}