POJ 1625 Censored!（自动机DP+大数相加）

题意：给出包含n个可见字符的字符集，以下所提字符串均由该字符集中的字符构成。给出p个长度不超过10的字符串，求长为m且不包含上述p个字符串的字符串有多少个。

数据范围：1<=n,m<=50，0<=p<=10

状态设计：dp[i][j]，i 步之内未经过危险结点且第 i 步到达结点 j 的路径数目。

状态转移：dp[i][j]=∑dp[i-1][k]，在结点 k 加输入 s[i] 能跳到结点 j

初始化：dp[0][0]=1，对于其余的 i ：dp[0][i]=0

注意：由于最后结果很大，而题中又没提到取模，所以要用到大数相加。

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <queue>
using namespace std;
#define NODE 110
int next[NODE][50];
int fail[NODE];
bool flag[NODE];
int n,L,m,node;

char ch[51];

int dp[51][NODE][100];

int cmp(const void *a,const void *b)
{
    return *(char*)a-*(char*)b;
}
void init()
{
    node=1;
    memset(next[0],0,sizeof(next[0]));
}
void add(int cur,int k)
{
    memset(next[node],0,sizeof(next[node]));
    flag[node]=0;
    next[cur][k]=node++;
}
int hash(char c)
{
    int min=0,max=n,mid;
    while(min+1!=max)
    {
        mid=min+max>>1;
        if(ch[mid]>c)    max=mid;
        else    min=mid;
    }
    return min;
}
void insert(char *s)
{
    int i,cur,k;
    for(i=cur=0;s[i];i++)
    {
        k=hash(s[i]);
        if(!next[cur][k])   add(cur,k);
        cur=next[cur][k];
    }
    flag[cur]=1;
}
void build_ac()
{
    queue<int>q;
    int cur,nxt,tmp,k;

    fail[0]=0;
    q.push(0);

    while(!q.empty())
    {
        cur=q.front(),q.pop();
        for(k=0;k<n;k++)
        {
            nxt=next[cur][k];
            if(nxt)
            {
                if(!cur)    fail[nxt]=0;
                else
                {
                    for(tmp=fail[cur];tmp&&!next[tmp][k];tmp=fail[tmp]);
                    fail[nxt]=next[tmp][k];
                }
                if(flag[fail[nxt]]) flag[nxt]=1;
                q.push(nxt);
            }
            else    next[cur][k]=next[fail[cur]][k];
        }
    }
}
void ADD(int *a,int *b)
{
    int i,c=0;
    for(i=0;i<100;i++)
    {
        a[i]+=b[i]+c;
        c=a[i]/10;
        a[i]%=10;
    }
}
void solve()
{
    memset(dp,0,sizeof(dp));
    dp[0][0][0]=1;

    for(int step=1;step<=L;step++)
    {
        for(int pre=0;pre<node;pre++)
        {
            if(flag[pre])   continue;
            for(int k=0;k<n;k++)
            {
                int cur=next[pre][k];
                if(flag[cur])   continue;
                ADD(dp[step][cur],dp[step-1][pre]);
            }
        }
    }

    int ans[100],i;
    memset(ans,0,sizeof(ans));
    for(i=0;i<node;i++) if(!flag[i])    ADD(ans,dp[L][i]);

    for(i=99;i>=0 && ans[i]==0;i--);
    if(i<0) puts("0");
    else
    {
        for(;i>=0;i--)  printf("%d",ans[i]);
        puts("");
    }
}
int main()
{
    char s[51];
    while(~scanf("%d%d%d",&n,&L,&m))
    {
        getchar();
        gets(ch);
        qsort(ch,strlen(ch),sizeof(char),cmp);

        init();
        for(int i=0;i<m;i++)
        {
            gets(s);
            insert(s);
        }
        build_ac();
        solve();
    }
    return 0;
}