HDOJ 2825 Wireless Password (AC自动机+DP)

题意：给m个字符串，求长为n且至少包含k个上述字符串的字符串有多少个。

数据范围：(1<=n<=25)，(0<=m<=10)

分析：用dp[i][cur][s]表示走i步后，到达结点cur，包含的字符串压缩为状态s（第x位为1表示包含第x个字符串）的字符串有多少个。

TLE：当dp[i-1][pre][s]=0时，直接continue，不用进入下一层循环，经此优化后就AC了。

#include <stdio.h>
#include <string.h>
#include <queue>
using namespace std;
#define NODE 101
#define MOD 20090717

int n,m,cnt;
int next[NODE][26],fail[NODE],flag[NODE],node;
int dp[26][NODE][1<<10];

int newnode()
{
    fail[node]=flag[node]=0;
    memset(next[node],0,sizeof(next[0]));
    return node++;
}
void insert(char *s,int id)
{
    int i,k,cur;
    for(i=cur=0;s[i];i++)
    {
        k=s[i]-'a';
        if(!next[cur][k])   next[cur][k]=newnode();
        cur=next[cur][k];
    }
    flag[cur] |=(1<<id);
}
void makenext()
{
    int u,v;
    queue<int>q;

    q.push(0);
    while(!q.empty())
    {
        u=q.front(),q.pop();
        for(int k=0;k<26;k++)
        {
            v=next[u][k];
            if(v)   q.push(v);
            else    next[u][k]=next[fail[u]][k];
            if(u&&v)
            {
                fail[v]=next[fail[u]][k];
                flag[v] |=flag[fail[v]];
            }
        }
    }
}
int cal(int s)
{
    int ret=0;
    while(s)
    {
        ret+=s&1;
        s>>=1;
    }
    return ret;
}
void DP()
{
    for(int i=0;i<=n;i++)
    {
        for(int j=0;j<node;j++)
            memset(dp[i][j],0,sizeof(dp[0][0][0])*(1<<m));
    }
    dp[0][0][0]=1;

    for(int i=1;i<=n;i++)
    {
        for(int pre=0;pre<node;pre++)
        {
            for(int s=0;s<(1<<m);s++)
            {
                if(dp[i-1][pre][s]==0)  continue;
                for(int k=0;k<26;k++)
                {
                    int cur=next[pre][k];
                    int ns=s | flag[cur];
                    dp[i][cur][ns]+=dp[i-1][pre][s];
                    dp[i][cur][ns]%=MOD;
                }
            }
        }
    }
    int ans=0;
    for(int cur=0;cur<node;cur++)
    {
        for(int s=0;s<(1<<m);s++)
        {
            if(cal(s)>=cnt)
            {
                ans+=dp[n][cur][s];
                ans%=MOD;
            }
        }
    }
    printf("%d\n",ans);
}
int main()
{
    char s[11];
    while(scanf("%d%d%d",&n,&m,&cnt))
    {
        if(!n&&!m&&!cnt)    break;
        node=0,newnode();
        for(int i=0;i<m;i++)
        {
            scanf("%s",s);
            insert(s,i);
        }
        makenext();
        DP();
    }
    return 0;
}