HDU 2825 Wireless Password

题目链接：HDU-2825

题意：给出m个单词，要构造出满足包含其中大于等于k个单词的字符串，字符只包括小写字母，问长度为n的这样的串有多少个。

思路：令dp[i][j][k]表示当前已经构造了i个字符，在ac自动机上跑到结点j，且单词状态为k情况下的方案数。从dp[i][j][k]向dp[i+1]递推。注意单词可能重复出现。

代码：

 #include<cstdio>
 #include<cstdlib>
 #include<vector>
 #include<set>
 #include<cstring>
 #include<algorithm>
 #include<cmath>
 #include<iostream>
 #include<queue>
 using namespace std;
 typedef long long LL;
 const LL MAXN=2e3+;
 const LL INF=0x3f3f3f3f;
 const LL MOD=;

 struct AC
 {
     int ch[MAXN][];
     vector<int> val[MAXN];
     int sz;
     AC() { init(); }
     void init() { sz=; memset(ch[],,sizeof(ch[])); val[].clear();}
     int idx(char c)
     {
         return c-'a';
     }
     void insert(char *s,int v)
     {
         int u=,n=strlen(s);
         for(int i=;i<n;i++)
         {
             int c=idx(s[i]);
             if(!ch[u][c])
             {
                 memset(ch[sz],,sizeof(ch[sz]));
                 val[sz].clear();
                 ch[u][c]=sz++;
             }
             u=ch[u][c];
         }
         val[u].push_back(v);
     }
     int f[MAXN];
     int last[MAXN];
     void getFail()
     {
         queue<int> q;
         f[]=;
         for(int c=;c<;c++)
         {
             int u=ch[][c];
             if(u) {f[u]=;q.push(u);last[u]=;}
         }
         while(!q.empty())
         {
             int r=q.front();q.pop();
             for(int c=;c<;c++)
             {
                 int u=ch[r][c];
                 if(!u)
                 {
                     ch[r][c]=ch[f[r]][c];
                     continue;
                 }
                 q.push(u);
                 int v=f[r];
                 while(v && !ch[v][c]) v=f[v];
                 f[u]=ch[v][c];
                 last[u]=(val[f[u]].size())?f[u]:last[f[u]];
             }
         }
     }
 };
 AC T;
 int f[][][];
 bool isOK(int x,int k)
 {
     int sum=;
     for(int i=;i<=;i++)
         if(x&(<<i))
             sum++;
     return sum>=k;
 }
 int main()
 {
 #ifdef LOCAL
     freopen("in.txt","r",stdin);
     // freopen("out.txt","w",stdout);
 #endif
     int n,m,K;
     while(scanf("%d%d%d",&n,&m,&K)!=EOF && (n || m || K))
     {
         T.init();
         for(int i=;i<=m;i++)
         {
             char s[];
             scanf("%s",s);
             T.insert(s,i);
         }
         T.getFail();
         memset(f,,sizeof(f));
         f[][][]=;
         for(int i=;i<n;i++)
         {
             for(int j=;j<T.sz;j++)
                 for(int k=;k<(<<(m+));k++)
                     f[-i%][j][k]=;
             for(int j=;j<T.sz;j++)
                 for(int k=;k<(<<(m+));k++)
                     if(f[i%][j][k])
                         for(int jj=;jj<;jj++)
                         {
                             int v= T.ch[j][jj];
                             int kk=k;
                             if(T.val[v].size())
                             {
                                 for(unsigned int ii=;ii<T.val[v].size();ii++)
                                 {
                                     if((kk&(<<T.val[v][ii]))==)
                                         kk+=(<<T.val[v][ii]);
                                 }
                             }
                             while(T.last[v]!=)
                             {
                                 v=T.last[v];
                                 if(T.val[v].size())
                                 {
                                     for(unsigned int ii=;ii<T.val[v].size();ii++)
                                     {
                                         if((kk&(<<T.val[v][ii]))==)
                                             kk+=(<<T.val[v][ii]);
                                     }
                                 }
                             }
                             v=T.ch[j][jj];
                             f[(i+)%][v][kk]=(f[i%][j][k]+f[(i+)%][v][kk])%MOD;
                         }
         }
         int ans=;
         for(int i=;i<T.sz;i++)
             for(int k=;k<(<<(m+));k+=)
                 if(isOK(k,K))
                     ans=(ans+f[n%][i][k])%MOD;
         printf("%d\n",ans);
     }
     return ;
 }