HDU - 2825 Wireless Password(AC自動機+狀壓dp)

題目鏈接：點擊查看

題目大意：給出 m 個匹配串，問長度爲 n 的字符串中，至少包含 k 個匹配串(可重疊)的字符串有多少個

題目分析：考慮到n，m，k都特別小，所以可以先用AC自動機將狀態關係轉移出來，然後配合狀壓dp進行狀態轉移，時間複雜度爲n * L * 2^k ，其中 L 爲匹配串的總長度，也就是AC自動機的節點個數

代碼：

#include<iostream>
#include<cstdio>
#include<string>
#include<ctime>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<stack>
#include<climits>
#include<queue>
#include<map>
#include<set>
#include<sstream>
using namespace std;
   
typedef long long LL;
  
typedef unsigned long long ull;
   
const int inf=0x3f3f3f3f;
   
const int N=1e5+100;

const int mod=20090717;

int n,m,k;
 
char s[110];
 
int fail[110],trie[110][26],num[(1<<10)+10],cnt;

int vis[110];

LL dp[30][110][(1<<10)+10];
 
void insert_word(int id)
{
	int len=strlen(s);
	int pos=0;
	for(int i=0;i<len;i++)
	{
		int to=s[i]-'a';
		if(!trie[pos][to])
			trie[pos][to]=++cnt;
		pos=trie[pos][to];
	}
	vis[pos]|=(1<<id);
}
 
void getfail()
{
	queue<int>q;
	for(int i=0;i<26;i++)
	{
		if(trie[0][i])
		{
			fail[trie[0][i]]=0;
			q.push(trie[0][i]);
		}
	}
	while(!q.empty())
	{
		int cur=q.front();
		q.pop();
		vis[cur]|=vis[fail[cur]];//fail指針的狀態轉移下來 
		for(int i=0;i<26;i++)
		{
			if(trie[cur][i])
			{
				fail[trie[cur][i]]=trie[fail[cur]][i];
				q.push(trie[cur][i]);
			}
			else
				trie[cur][i]=trie[fail[cur]][i];
		}
	}
}
 
void init()
{
	cnt=0;
	memset(vis,false,sizeof(vis));
	memset(trie,0,sizeof(trie));
}

int main()
{
//#ifndef ONLINE_JUDGE
//  freopen("input.txt","r",stdin);
//    freopen("output.txt","w",stdout);
//#endif
//  ios::sync_with_stdio(false);
	for(int i=0;i<1<<10;i++)//計算每個二進制中有多少個 1  
	{
		num[i]=0;
		for(int j=0;j<10;j++)
			if(i&(1<<j))
				num[i]++;
	}
	while(scanf("%d%d%d",&n,&m,&k)!=EOF&&n+m+k)
	{
		init(); 
		for(int i=0;i<m;i++)
		{
			scanf("%s",s);
			insert_word(i);
		}
		getfail();
		for(int i=0;i<=n;i++)
			for(int j=0;j<=cnt;j++)
				for(int k=0;k<1<<m;k++)
					dp[i][j][k]=0;
		dp[0][0][0]=1;
		for(int i=0;i<n;i++)//迭代 n 次
			for(int j=0;j<=cnt;j++)
				for(int k=0;k<1<<m;k++)
					if(dp[i][j][k])
						for(int t=0;t<26;t++)
						{
							int nj=trie[j][t];
							int nk=(k|vis[nj]);
							dp[i+1][nj][nk]+=dp[i][j][k];
							dp[i+1][nj][nk]%=mod;
						} 
		LL ans=0;
		for(int i=0;i<1<<m;i++)
			if(num[i]>=k)
				for(int j=0;j<=cnt;j++)
					ans=(ans+dp[n][j][i])%mod;
		printf("%lld\n",ans);
	}
	
	
	
	
	
	
	
	

      
      
      
      
      
      
      
      
      
    return 0;
}