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+10;
const LL INF=0x3f3f3f3f;
const LL MOD=20090717;

struct AC
{
    int ch[MAXN][26];
    vector<int> val[MAXN];
    int sz;
    AC() { init(); }
    void init() { sz=1; memset(ch[0],0,sizeof(ch[0])); val[0].clear();}
    int idx(char c)
    {
        return c-'a';
    }
    void insert(char *s,int v)
    {
        int u=0,n=strlen(s);
        for(int i=0;i<n;i++)
        {
            int c=idx(s[i]);
            if(!ch[u][c])
            {
                memset(ch[sz],0,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[0]=0;
        for(int c=0;c<26;c++)
        {
            int u=ch[0][c];
            if(u) {f[u]=0;q.push(u);last[u]=0;}
        }
        while(!q.empty())
        {
            int r=q.front();q.pop();
            for(int c=0;c<26;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[2][300][3000];
bool isOK(int x,int k)
{
    int sum=0;
    for(int i=1;i<=10;i++)
        if(x&(1<<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=1;i<=m;i++)
        {
            char s[100];
            scanf("%s",s);
            T.insert(s,i);
        }
        T.getFail();
        memset(f,0,sizeof(f));
        f[0][0][0]=1;
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<T.sz;j++)
                for(int k=0;k<(1<<(m+1));k++)
                    f[1-i%2][j][k]=0;
            for(int j=0;j<T.sz;j++)
                for(int k=0;k<(1<<(m+1));k++)
                    if(f[i%2][j][k])
                        for(int jj=0;jj<26;jj++)
                        {
                            int v= T.ch[j][jj];
                            int kk=k;
                            if(T.val[v].size())
                            {
                                for(unsigned int ii=0;ii<T.val[v].size();ii++)
                                {
                                    if((kk&(1<<T.val[v][ii]))==0)
                                        kk+=(1<<T.val[v][ii]);
                                }
                            }
                            while(T.last[v]!=0)
                            {
                                v=T.last[v];
                                if(T.val[v].size())
                                {
                                    for(unsigned int ii=0;ii<T.val[v].size();ii++)
                                    {
                                        if((kk&(1<<T.val[v][ii]))==0)
                                            kk+=(1<<T.val[v][ii]);
                                    }
                                }
                            }
                            v=T.ch[j][jj];
                            f[(i+1)%2][v][kk]=(f[i%2][j][k]+f[(i+1)%2][v][kk])%MOD;
                        }
        }
        int ans=0;
        for(int i=0;i<T.sz;i++)
            for(int k=0;k<(1<<(m+1));k+=2)
                if(isOK(k,K))
                    ans=(ans+f[n%2][i][k])%MOD;
        printf("%d
",ans);
    }
    return 0;
}