poj 1625 Censored!

题意：

　　给你一个字典，还有一些非法串，利用字典中的字符组成长度为m的不含非法串的字符串，求方案数。

解法：

　　坑了我一整天我擦。。。。

　　因为自动机本身就是一个状态转移图，所以在上面进行DP很好理解，dp[i][j]记录长度为i并且走到DFA中j节点的方案数。。

　　需要注意的是非法节点是不能走的，在建Trie图的时候也要注意，如果当前节点的fail节点是非法的，那么这个点也是非法的，因为这个串包含了某个非法串。。。

#include<cstdio>
#include<cstring>
#include<queue>
#include<map>
#include<algorithm>
using namespace std;
const int base = 10;
const int N = 105;
struct BigInt{
    int v[N],len;
    BigInt(int r = 0){
        memset(v,0,sizeof(v));
        for(len = 0;r > 0;r /= base)v[len++] = r % base;
    }
    BigInt operator + (const BigInt &a){
        BigInt ans;int i ,c = 0;
        for(i = 0;i < len || i < a.len || c > 0;i++){
            if(i < len)c += v[i];
            if(i < a.len)c += a.v[i];
            ans.v[i] = c % base;c /= base;
        }
        ans.len = i;
        return ans;
    }
    void print(){
        printf("%d",len==0?0:v[len - 1]);
        for(int i = len - 2;i >= 0;i--)
            printf("%d",v[i]);
        puts("");
    }
};
BigInt dp[55][N];

struct node{
    node *ch[55],*fail;
    bool flag;
    void clear(){
        for(int i = 0;i < 55;i++)ch[i] = NULL;fail = NULL;
        flag = 0;
    }
};
map<char,int>hash;
node stk[N*6];
struct Trie{
    node *root;int top,sz;
    node* newnode(){
        node *p = &stk[top++];
        p -> clear();
        return p;
    }
    void init(int size){
        top = 0;
        sz = size;
        root = newnode();
    }
    void insert(char *s){
        node *p = root;int len = strlen(s);
        for(int i = 0 ;i < len;i++){
            int id = hash[s[i]];
            if(p -> ch[id] == NULL)
                p -> ch[id] = newnode();
            p = p -> ch[id];
        }
        p -> flag = 1;
    }
    void build(){
        queue<node*> Q;
        root -> fail = root;
        for(int i = 0;i < sz;i++)
            if(root -> ch[i] == NULL)
                root -> ch[i] = root;
            else{
                Q.push(root -> ch[i]);
                root -> ch[i] -> fail = root;
            }
        while(!Q.empty()){
            node *p = Q.front();Q.pop();
            for(int i = 0;i < sz;i++)
                if(p -> ch[i] == NULL)
                    p -> ch[i] = p -> fail -> ch[i];
                else{
                    Q.push(p -> ch[i]);
                    p -> ch[i] -> fail = p -> fail -> ch[i];
                    p -> ch[i] -> flag |= p -> ch[i] -> fail -> flag;
                }
        }
    }
    void solve(int len){
        for(int i = 0;i < 55;i++)
            for(int j = 0;j < top;j++)
                dp[i][j] = BigInt();
        dp[0][root - stk] = BigInt(1);
        for(int i = 0;i < len;i++)
            for(int j = 0;j < top;j++){
                node *cur = &stk[j];
                if(cur -> flag)continue;
                for(int k = 0;k < sz;k++){
                    node *next = cur -> ch[k];
                    if(next -> flag)continue;
                    dp[i+1][next - stk] = dp[i+1][next - stk] + dp[i][j];
                }
            }
        BigInt ans;
        for(int i = 0;i < top;i++)
            ans = ans + dp[len][i];
        ans.print();
    }
};
Trie AC;
char s[N];
int main(){
    int n,m,p;
    while(~scanf("%d%d%d",&n,&m,&p)){
        scanf("%s",s);int len = strlen(s);
        hash.clear();
        for(int i = 0;i < len;i++)
            hash[s[i]] = i;
        AC.init(n);
        for(int i = 0;i < p;i++){
            scanf("%s",s);
            AC.insert(s);
        }
        AC.build();
        AC.solve(m);
    }
    return 0;
}