POJ1625 Censored!（AC自动机+DP）

题目问长度m不包含一些不文明单词的字符串有多少个。

依然是水水的AC自动机+DP。。做完后发现居然和POJ2778是一道题，回过头来看都水水的。。。

• dp[i][j]表示长度i（在自动机转移i步）且后缀状态为自动机第j个结点的合法字符串数
• dp[0][0]=1
• 转移转移。。。

注意要用高精度，因为答案最多50⁵⁰。

还有就是要用unsigned char，题目的输入居然有拓展的ASCII码，编码128-255。

#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;
int tn,ch[555][256],fail[555],flag[555];
void insert(unsigned char *s){
    int x=0;
    for(int i=0; s[i]; ++i){
        int y=s[i];
        if(ch[x][y]==0) ch[x][y]=++tn;
        x=ch[x][y];
    }
    flag[x]=1;
}
void init(){
    memset(fail,0,sizeof(fail));
    queue<int> que;
    for(int i=0; i<256; ++i){
        if(ch[0][i]) que.push(ch[0][i]);
    }
    while(!que.empty()){
        int x=que.front(); que.pop();
        for(int i=0; i<256; ++i){
            if(ch[x][i]) que.push(ch[x][i]),fail[ch[x][i]]=ch[fail[x]][i],flag[ch[x][i]]|=flag[ch[fail[x]][i]];
            else ch[x][i]=ch[fail[x]][i];
        }
    }
}
#define ULL unsigned long long
#define M 1000000000000000000L
struct BigInt{
    ULL bit[5];
    BigInt(int a=0){
        bit[0]=bit[1]=bit[2]=bit[3]=0;
        bit[4]=a;
    }
};
BigInt operator+(const BigInt &b1,const BigInt &b2){
    BigInt b;
    ULL carry=0;
    for(int i=4; i>=0; --i){
        ULL tmp=b1.bit[i]+b2.bit[i]+carry;
        b.bit[i]=tmp%M;
        carry=tmp/M;
    }
    return b;
};
void output(const BigInt &b){
    int i=0;
    while(i<5 && b.bit[i]==0) ++i;
    if(i==5){
        puts("0");
        return;
    }
    printf("%llu",b.bit[i]);
    for(int j=i+1; j<5; ++j){
        printf("%018llu",b.bit[j]);
    }
    putchar('
');
}
int main(){
    BigInt one(1);
    unsigned char str[55];
    int n,m,p;
    while(~scanf("%d%d%d",&n,&m,&p)){
        scanf("%s",str);
        bool vis[256]={0};
        for(int i=0; str[i]; ++i) vis[str[i]]=1;
        tn=0;
        memset(ch,0,sizeof(ch));
        memset(flag,0,sizeof(flag));
        for(int i=0; i<p; ++i){
            scanf("%s",str);
            insert(str);
        }
        init();
        BigInt **d=new BigInt*[m+1];
        for(int i=0; i<=m; ++i) d[i]=new BigInt[tn+1];
        d[0][0]=one;
        for(int i=0; i<m; ++i){
            for(int j=0; j<=tn; ++j){
                if(flag[j]) continue;
                for(int k=33; k<256; ++k){
                    if(!vis[k] || flag[ch[j][k]]) continue;
                    d[i+1][ch[j][k]]=d[i+1][ch[j][k]]+d[i][j];
                }
            }
        }
        BigInt res=0;
        for(int i=0; i<=tn; ++i) res=res+d[m][i];
        output(res);
        for(int i=0; i<=m; ++i) delete[] d[i];
        delete[] d;
    }
    return 0;
}