HDU2825 AC自动机+状压DP

题意：给你m个字符串，求长度为n的包含至少k个给定串的字符串个数，答案对20090717取模

思路：因为m的值很小，所以可以进行状压DP

dp[i][j][k]   i为字符串当前长度，j为当前走到的自动机的节点编号，k为状压后的当前状态

于是只需在自动机上面跑满长度n，跑的过程中统计个数即可。

代码如下：

#include<bits/stdc++.h>
using namespace std;
const int maxn=105;
const int mod=20090717;
struct AC
{
    int next[maxn][26],fail[maxn],ed[maxn];
    int root,L;
    int newnode()
    {
        for(int i=0;i<26;i++)
            next[L][i]=-1;
        ed[L++]=0;
        return L-1;
    }
    void init()
    {
        L=0;
        root=newnode();
    }
    void insert(char s[],int id)
    {
        int now=root;
        for(int i=0;s[i];i++)
        {
            if(next[now][s[i]-'a']==-1)
                next[now][s[i]-'a']=newnode();
            now=next[now][s[i]-'a'];
        }
        ed[now]=1<<id;
    }
    void bfs()
    {
        queue<int> q;
        fail[root]=root;
        for(int i=0;i<26;i++)
        {
            if(next[root][i]==-1)
                next[root][i]=root;
            else
            {
                fail[next[root][i]]=root;
                q.push(next[root][i]);
            }
        }
        while(!q.empty())
        {
            int now=q.front();q.pop();
            ed[now]|=ed[fail[now]];
            for(int i=0;i<26;i++)
            {
                if(next[now][i]==-1)
                    next[now][i]=next[fail[now]][i];
                else
                {
                    fail[next[now][i]]=next[fail[now]][i];
                    q.push(next[now][i]);
                }
            }
        }
    }
}ac;
char s[12];
int dp[30][105][1025];
int cnt[1025];
int main()
{
    ios_base::sync_with_stdio(0);
    int n,m,num;
    for(int i=0;i<1024;i++)
        cnt[i]=__builtin_popcount(i);
    while(cin>>n>>m>>num)
    {
        if(!n&&!m&&!num)
            break;
        ac.init();
        for(int i=0;i<m;i++)
        {
            cin>>s;
            ac.insert(s,i);
        }
        ac.bfs();
        memset(dp,0,sizeof dp);
        dp[0][0][0]=1;
        int all=1<<m;
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<ac.L;j++)
            {
                for(int k=0;k<all;k++)
                {
                    if(!dp[i][j][k]) continue;
                    for(int t=0;t<26;t++)
                    {
                        int to=ac.next[j][t];
                        dp[i+1][to][ac.ed[to]|k]+=dp[i][j][k];
                        dp[i+1][to][ac.ed[to]|k]%=mod;
                    }
                }
            }
        }
        int ans=0;
        for(int i=0;i<ac.L;i++)
        {
            for(int j=0;j<all;j++)
            {
                if(cnt[j]>=num)
                {
                    ans+=dp[n][i][j];
                    ans%=mod;
                }
            }
        }
        cout<<ans<<endl;
    }
    return 0;
}