hdu2825 Wireless Password（AC自动机+状压dp）

题目链接：这里写链接内容
题目描述：给定n、m、k，其中n为密码的长度，m为已知串的数量，k为密码中应该包含的已知串的数量，问密码有多少种可能？

解题思路：AC自动机+状压dp
状态：dp[i][j][k]
其中i为密码的长度，j为当前在AC自动机中所处的状态，k为包含的已知串的信息，k转化为二进制，如果第x位为1则说明包含第x个已知串
状态转移方程：
dp[i+1][next1[j][u]][k|end1[next1[j][u]]]+=dp[i][j][k];

代码：

#include 
#include 
#include 
#define MOD 20090717
using namespace std;
int _hash[1024];
int dp[26][110][1024];
struct Trie
{
    int next1[110][26],fail[110];
    int end1[110];
    int root,L;
    int newnode()
    {
        for(int i=0; i<26; ++i) next1[L][i]=-1;
        end1[L++]=0;
        return L-1;
    }
    void init()
    {
        L=0;
        root=newnode();
    }
    void insertnode(char* str,int id)
    {
        int len=strlen(str);
        int now=root;
        for(int i=0; iint to=str[i]-'a';
            if(next1[now][to]==-1) next1[now][to]=newnode();
            now=next1[now][to];
        }
        end1[now]|=(1<void build()
    {
        fail[root]=root;
        queue<int> q;
        for(int i=0; i<26; ++i)
        {
            if(next1[root][i]==-1)
                next1[root][i]=root;
            else
            {
                fail[next1[root][i]]=root;
                q.push(next1[root][i]);
            }
        }
        while(!q.empty())
        {
            int now=q.front();
            q.pop();
            if(end1[fail[now]]) end1[now]=(end1[now]|end1[fail[now]]);
            for(int i=0; i<26; ++i)
            {
                if(next1[now][i]==-1)
                    next1[now][i]=next1[fail[now]][i];
                else
                {
                    fail[next1[now][i]]=next1[fail[now]][i];
                    q.push(next1[now][i]);
                }
            }
        }
    }
    int solve(int n,int kk)
    {
        memset(dp,0,sizeof(dp));
        dp[0][0][0]=1;
        for(int i=0; ifor(int j=0; jfor(int k=0; k<1024; ++k)
            if(dp[i][j][k]) for(int u=0;u<26;u++){
                dp[i+1][next1[j][u]][k|end1[next1[j][u]]]+=dp[i][j][k];
                dp[i+1][next1[j][u]][k|end1[next1[j][u]]]%=MOD;
            }
        int ans=0;
        for(int i=0; ifor(int j=0;j<1024;++j)
            if(_hash[j]>=kk)
            {
                ans+=dp[n][i][j];
                ans%=MOD;
            }
        return ans;
    }
};
int lowbit(int x)
{
    int t=0;
    while(x)
    {
        x=x&(x-1);
        t++;
    }
    return t;
}
Trie ac;
char st[20];
int n,m,k;
int main()
{
    for(int i=0; i<1024; ++i)
        _hash[i]=lowbit(i);
    while(scanf("%d%d%d",&n,&m,&k)!=EOF&&(n!=0||m!=0||k!=0))
    {
        ac.init();
        for(int i=0; iscanf("%s",st);
            ac.insertnode(st,i);
        };
        ac.build();
        printf("%d\n",ac.solve(n,k));
    }
    return 0;
}