hdu 2222 keywords search （AC自动机 模板）

题目：http://acm.hdu.edu.cn/showproblem.php?pid=2222

我日，写错好多小细节，还差得远呢。。。

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#define N 500010
char str[1000010],keyword[51];
int head,tail;

typedef struct node{
    struct node*fail;
    struct node*next[26];
    bool exist;
    int count;
}*trie,node;
trie root;
trie q[N];

void creat(trie &p){
    p=(trie)malloc(sizeof(node));
    for(int i=0;i<26;i++)
        p->next[i]=NULL;
    p->exist=false;
    p->count=0;
    p->fail=NULL;             /*原来一切的根源在于root的fail没有初始化，日*/
}

void insert(trie p,char s[]){
    int k=0 ;
    while(s[k]!='\0'){
        if(!p->next[s[k]-'a']){
            trie q;
            creat(q);
            p->next[s[k]-'a']=q;
        }
        p=p->next[s[k]-'a'];
        k++;
    }
    p->count++;
}

void build_ac(){                         /*建立fail指针，BFS*/
    q[tail++]=root;
    while(head!=tail){
        node*p=q[head++];                  /*出队*/
        node*temp=NULL;
        for(int i=0;i<26;i++){
            if(p->next[i]!=NULL){
                if(p==root)
                    p->next[i]->fail=root;   /*root下面的26的fail必指向root*/
                else{
                    temp=p->fail;
                    while(temp!=NULL){
                        if(temp->next[i]!=NULL){
                            p->next[i]->fail=temp->next[i];
                            break;               /*找到了就break*/
                        }
                        temp=temp->fail;/*如果它的fail指针下的26个都没有的话，就指向fail指针的fail再找一次，直到退到root就也能退出来了*/
                    }
                    if(temp==NULL)
                        p->next[i]->fail=root;
                }
                q[tail++]=p->next[i];    /*入队*/
            }
        }
    }
}

int query(trie p){
    int index,len,result;
    result=0;
    len=strlen(str);
    for(int i=0;i<len;i++){                    /*写成了26，找了好久*/
        index=str[i]-'a';
        while(p->next[index]==NULL&&p!=root)   /*预测一下，下面没有了就赶紧跳到fail去。（root跳个屁啊）*/
             p=p->fail;
        p=p->next[index];
        if(p==NULL)
            p=root;
        trie temp=p;
        while(temp!=root&&temp->count!=-1){     /*通过fail指针看一下到目前为止的后缀串有没有单词*/
            result+=temp->count;
            temp->count=-1;
            temp=temp->fail;
        }
    }
    return result;
}
int main(){
    int ncase,n;
    scanf("%d",&ncase);
    while(ncase--){
        head=tail=0;
        creat(root);
        scanf("%d",&n);
        for(int i=0;i<n;i++){
            scanf("%s",keyword);
            insert(root,keyword);
        }
        build_ac();
        scanf("%s",str);
        printf("%d\n",query(root));
    }
    return 0;
}