HDU 2222 AC自动机 裸题

题意：

问母串中出现多少个模式串

注意ac自动机的节点总数

 

 

#include <stdio.h>  
#include <string.h>  
#include <queue>  
using namespace std;  
inline int Max(int a,int b){return a>b?a:b;}  
inline int Min(int a,int b){return a>b?b:a;}  
int ANS;  
  
  
#define N 1000010  
#define maxnode 250001  
#define sigma_size 26  
  
struct Trie{  
    int ch[maxnode][sigma_size];  
    int val[maxnode];  
    int last[maxnode];  
    int f[maxnode];  
    int sz;  
    void init(){  
        sz=1;  
        memset(ch,0,sizeof(ch));  
        memset(val, 0, sizeof(val));  
        memset(f,0,sizeof(f));  
        memset(last,0,sizeof(last));  
    }  
    int idx(char c){  
        return c-'a';  
    }  
    void print(int j){  
        if(j){  
            printf("%d: %d\n", j, val[j]);  
            print(last[j]);  
        }  
    }  
    void Creat(char *s){  
        int u = 0, len = strlen(s);  
        for(int i = 0;i < len;i++){  
            int c = idx(s[i]);  
            if(!ch[u][c])  
                ch[u][c] = sz++;  
  
            u = ch[u][c];  
        }  
        val[u] ++;  
    }  
    void getFail(){  
        queue<int> q;  
        for(int i = 0; i<sigma_size; i++)  
            if(ch[0][i]) q.push(ch[0][i]);  
  
        while(!q.empty()){  
            int r = q.front(); q.pop();  
            for(int c = 0; c<sigma_size; c++){  
                int u = ch[r][c];  
                if(!u)continue;  
                q.push(u);  
                int v = f[r];  
                while(v && ch[v][c] == 0) v = f[v]; //沿失配边走上去 如果失配后有节点 且 其子节点c存在则结束循环  
                f[u] = ch[v][c];  
                last[u] = val[f[u]] ? f[u] : last[f[u]];  
            }  
        }  
    }  
    void find(char *T){  
        int len = strlen(T), j = 0;  
        for(int i = 0; i < len; i++){  
            int c = idx(T[i]);  
            while(j && ch[j][c]==0) j = f[j];  
            j = ch[j][c];  
  
            int temp = j;  
            while(temp && val[temp]){  
                ANS += val[temp];  
                val[temp] = 0;  
                temp = f[temp];  
            }  
        }  
    }  
};  
Trie ac;  
char S1[N];  
int Slen;  
  
void InputString(){  
    scanf("%s",S1);  
    Slen = strlen(S1);  
    S1[Slen]='\0';  
}  
int main(){  
  
    int T,i,j,n;scanf("%d",&T);  
  
    while(T--){  
        ac.init();  
        scanf("%d",&n);  
        while(n--){  
            scanf("%s",S1);  
            ac.Creat(S1);  
        }  
        ac.getFail();  
        InputString();  
        ANS = 0;  
        ac.find(S1);  
  
        printf("%d\n",ANS);  
  
    }  
    return 0;  
}  
/*  
1  
5  
she  
he  
say  
shr  
her  
yasherhs  
  
ans:  
3  

*/