hud 2825 AC自动机dp

dp[i][j][k] 表示长度为i，以第j个节点为结尾，当前所用单词状态为k时的结果数目，其中单词只有10个，可以状态压缩。

然后利用AC自动机转移。

#include<stdio.h>
#include<string.h>
#include<iostream>
using namespace std;
const int M=20090717;
const int nMax=105;
const int SON=26;
int dp[30][nMax][1<<10],cnt,n,m,K;
struct Tree
{
    int id,flag;
    Tree *fail,*next[SON];
}t[nMax],*Q[nMax],*root;
Tree* newnode()
{
    Tree *p=&t[cnt];
    p->id=cnt++;p->flag=0;
    memset(p->next,0,sizeof(p->next));
    return p;
}
void insert(char s[],int k)
{
    Tree *p=root;
    for(int i=0;s[i];i++)
    {
        if(p->next[s[i]-'a']==NULL)
            p->next[s[i]-'a']=newnode();
        p=p->next[s[i]-'a'];
    }
    p->flag|=(1<<k);
}
void build()
{
    int f=0,r=-1;
    Q[++r]=root;
    while(f<=r)
    {
        Tree *p=Q[f++];
        for(int i=0;i<SON;i++)
        {
            if(p->next[i]!=NULL)
            {
                if(p==root) p->next[i]->fail=root;
                else
                {
                    p->next[i]->fail=p->fail->next[i];
                    p->next[i]->flag|=p->fail->next[i]->flag;
                }
                Q[++r]=p->next[i];
            }
            else
            {
                if(p==root) p->next[i]=root;
                else p->next[i]=p->fail->next[i];
            }
        }

    }
}
void dp_init()
{
    memset(dp,0,sizeof(dp));dp[0][0][0]=1;
    for(int i=1;i<=n;i++)
        for(int j=0;j<cnt;j++)
            for(int k=0;k<(1<<m);k++)
            if(dp[i-1][j][k]&&(t[j].flag&k)==t[j].flag) //k中必须包含j节点的状态，否则无效
                for(int s=0;s<SON;s++)
                {
                    int id=t[j].next[s]->id,st=k|t[j].next[s]->flag;
                    dp[i][id][st]+=dp[i-1][j][k];
                    if(dp[i][id][st]>=M) dp[i][id][st]%=M;
                }

}
int count(int st)
{
    int res=0;
    for(int i=0;i<10;i++)
    if(st&(1<<i)) res++;
    return res;
}
int main()
{
//    freopen("test.txt","r",stdin);
    while(scanf("%d%d%d",&n,&m,&K)&&n+m+K)
    {
        char s[20];
        cnt=0; root=newnode();
        for(int i=0;i<m;i++)
        {
            scanf("%s",s);insert(s,i);
        }
        build();dp_init();
        int res=0;
        for(int j=0;j<cnt;j++)
            for(int k=0;k<(1<<m);k++)
            if(count(k)>=K)
            {
                res+=dp[n][j][k];
                if(res>=M) res%=M;
            }
        printf("%d\n",res);
    }
    return 0;
}