hdu 2825 Wireless Password (ac自动机+状态压缩dp)

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

题意：给定m个模式串，问有多少个长为n的文本串至少包含K种模式串。

分析：首先将模式串建ac自动机，然后在求fail指针的时候把到达某个节点需要加的贡献求出来（由于最多只有10种模式串，可以将模式串种数压缩到一个int型）。然后记忆化搜索，dp[len][cur][num],len代表当前字符串长度，cur代表对应在ac机上的位置，num代表已经选的模式串的种数，dp[len][cur][num]表示方案数。ps:dp数组清零用memset超时，测试数据组数比较多吧。

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
using namespace std;
typedef long long LL;
const LL mod = 20090717;
const int maxn = 108;
const int kd = 26;
int N;
struct trie
{
	int son[maxn][kd],fail[maxn],end[maxn];
	int cnt,root;
	int L,K;
	LL dp[27][maxn][1200];
	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 id)
	{
		int index,i,now=root;
		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]|=(1<<id);
	}
	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]);
				}
		}
	}
	LL DFS(int len,int cur,int num)
	{
		if(len==L)
		{
			int i,c;
			for(i=0,c=0;i<10;i++)
				if(num&(1<<i))
					c++;
			return c>=K;
		}
		if(dp[len][cur][num]!=-1)
			return dp[len][cur][num];
		LL ret=0;
		for(int i=0;i<kd;i++)
			ret+=DFS(len+1,son[cur][i],num|end[son[cur][i]])%mod;
		return dp[len][cur][num]=ret%mod;
	}
	void solve()
	{
		for(int i=0;i<=L;i++)
			for(int j=0;j<=cnt;j++)
				for(int k=0;k<=(1<<N);k++)
					dp[i][j][k]=-1;
		LL ans=DFS(0,root,0);
		printf("%I64d\n",ans);
	}
}ac;
char s[maxn];
int main()
{
	while(scanf("%d%d%d",&ac.L,&N,&ac.K),(ac.L || N || ac.K))
	{
		ac.Init();
		for(int i=0;i<N;i++)
		{
			scanf("%s",s);
			ac.Insert(s,i);
		}
		ac.findfail();
		ac.solve();
	}
	return 0;
}