【HDU2825】Wireless Password【AC自动机，状态压缩DP】

题意

题目给出m(m<=10)个单词，每个单词的长度不超过10且仅由小写字母组成，给出一个正整数n(n<=25)和正整数k，问有多少方法可以组成长度为n的文本且最少包含k个给出的单词。

分析

和上一个AC自动机很相似，上一篇博客是不包含任何一个单词长度为n的方案数，这个题是包含至少k个单词的方案数，而且n，m，k都非常的小。

按照前面的经验很容易想到，我们还是得先建一个AC自动机，然后把它的单词结点标记出来。与前面不同的是我们在状态转移的时候需要考虑到当前走过的结点已经包含多少单词了。所以我们想到用dp[i][j][k]来表示当前在i结点，已经走了j步，且走过了k个单词结点。但是我们发现，这样表示状态的话没有办法转移，因为当我们遇到一个单词结点的时候，我们并不知道这个单词节点前面时候已经走过被计数了。所以我们想到，要使用状态压缩dp来解决这个题目。

f[i][j][S]当前在i结点，还需要走j步，包含的结点由S通过二进制来表示。在建AC自动机的时候，用match[i]=1<<j,来表示i结点是单词j的单词结点。

f[i][j][S]=sum(f[v][j-1][S|match[v]]).其中v是结点i的儿子结点。

当j==0时，如果S中包含的单词数目>=k，则f[i][0][S]=1，否则为0

然后我们就很容易用记忆搜索解决这个问题。

 #include <cstdio>
 #include <cstring>
 #include <algorithm>
 #include <iostream>
 #include <queue>

 using namespace std;
 const int maxnode=;
 const int MOD=;
 const int sigma_size=;
 int ch[maxnode][sigma_size],match[maxnode],f[maxnode];
 int dp[maxnode][][(<<)+],vis[maxnode][][(<<)+];
 int sz;
 void init(){
     sz=;
     memset(ch[],,sizeof(ch[]));
     memset(vis,,sizeof(vis));
     match[]=;
 }
 void insert(char *s,int v){
     int n=strlen(s),u=;
     for(int i=;i<n;i++){
         int c=s[i]-'a';
         if(!ch[u][c]){
             ch[u][c]=sz;
             memset(ch[sz],,sizeof(ch[sz]));
             match[sz++]=;
         }
         u=ch[u][c];
     }
     match[u]|=(<<v);
 }

 void getFail(){
     queue<int>q;
     f[]=;
     for(int i=;i<sigma_size;i++){
         int u=ch[][i];
         if(u){
             q.push(u);
             f[u]=;
         }
     }
     while(!q.empty()){
         int r=q.front();q.pop();
         for(int i=;i<sigma_size;i++){
             int u=ch[r][i];
             if(!u){
                 ch[r][i]=ch[f[r]][i];
                 continue;
             }
             q.push(u);
             int v=f[r];
             while(v&&!ch[v][i])v=f[v];
             f[u]=ch[v][i];
             match[u]|=match[f[u]];
         }
     }
 }
 int n,m,k;
 char s[];
 int Count(int S){
     int res=;
     for(int i=;i<m;i++){
         if(S&(<<i))
             res++;
     }
     return res;
 }
 int DP(int u,int L,int S){
     if(vis[u][L][S])
         return dp[u][L][S];
     vis[u][L][S]=;
     int &ans=dp[u][L][S];
     ans=;
     if(L==){
         if(Count(S)>=k){
             return ans=;
         }
         return ans=;
     }
     for(int i=;i<sigma_size;i++){
         int v=ch[u][i];
         ans=(ans%MOD+DP(v,L-,S|match[v])%MOD)%MOD;
     }
     return ans;
 }
 int main(){
     while(scanf("%d%d%d",&n,&m,&k)!=EOF&&(n||m||k)){
         init();
         for(int i=;i<m;i++){
             scanf("%s",s);
             insert(s,i);
         }
         getFail();
         int ans=DP(,n,);
         printf("%d\n",ans%MOD);
     }
 return ;
 }