【AC自动机】HDU2222Keywords Search

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

这个题目刚开始还不会，但是轮到了我给我同学讲课，之前就有看过，一直都没懂。就唯一的知道要会AC自动机，就得先会KMP和字典树。于是我就在杭电上挂了这题，默默的刷着KMP和字典树。突然要我讲课，没办法，就赶紧赶紧的临时在认真的看，分析，最后赶鸭子上架，懵懵懂懂的就上去讲了。讲着讲着发现这个模板我就这样理解了。。。哈哈，真的让我明白，你能做题，不一定能讲清楚一道题，当你能讲清楚一道题，说明你弄懂了这道题。嘿嘿。。。

我是看到这两个博客，学过来的，代码也是综合了一下。

链接：http://www.cppblog.com/mythit/archive/2009/04/21/80633.html

          http://blog.csdn.net/hnust_xiehonghao/article/details/9130539

#include<iostream>#include<string>#include<cstring>#include<cstdio>#include<queue>using namespace std;const int N=26;char str[1000005];char s[55];struct node{    node *fail;    node *next[N];    int cnt;    node(){        fail=NULL;        cnt=0;        memset(next,0,sizeof(next));    }};node *p,*temp,*head;void Insert(char s[])           //  建立字典树；{    p=head;    int i=0,id;    while(s[i]){        id=s[i++]-'a';        if(p->next[id]==NULL) p->next[id]=new node();        p=p->next[id];    }    p->cnt++;}void get_fail()             //  BFS构造fail指针；{    node *son;              //  保存子节点；    p=head;    queue<struct node *>q;          //  用队列BFS；    q.push(p);    while(!q.empty()){        temp=q.front();        q.pop();        for(int i=0;i<26;i++){            son=temp->next[i];            if(son!=NULL){                      //  子节点不为空，说明存在该字母；                if(temp==head) son->fail=head;  //  如果子节点的父节点为head,那么fail指针指向head;                else{                    p=temp->fail;               //  p等于当前son的父节点的fail指针；                    while(p){                        if(p->next[i]){         //  如果存在该字母，说明p->next[i]链前面出现的字母也与之匹配；                            son->fail=p->next[i];                            break;                        }                        p=p->fail;                    }                    if(!p) son->fail=head;                }                q.push(son);            }        }    }}int Query()     //  遍历；{    int i=0,ans=0;    p=head;    while(str[i]){        int id=str[i++]-'a';        while( p->next[id]==NULL && p!=head) p=p->fail;     //  当p的子节点不存在，且p不是head的节点时，(因为这种情况当前位置p后面是不会有匹配的字母了)p=p->fail;        p=p->next[id];        if(!p) p=head;          //  如果p为NULL，那么说明该字母开头的单词匹配失败了；        temp=p;        while(temp!=head && temp->cnt!=-1){     //  当temp存在时，且没有被计算过时，进入while循环；            ans+=temp->cnt;            temp->cnt=-1;            temp=temp->fail;        //  访问下一个匹配位置；        }    }    return ans;}int main(){    int t,n;    scanf("%d",&t);    while(t--){        head=new node();        scanf("%d",&n);        getchar();        while(n--){            gets(s);            Insert(s);        }        get_fail();        scanf("%s",str);        printf("%d\n",Query());    }    return 0;}