hdu2222 AC自动机

来源：互联网发布：中国软件协会培训中心编辑：程序博客网时间：2024/05/16 00:56
#include <iostream>#include <cstring>#include <cstdio>#include <algorithm>using namespace std;const int kind = 26; struct node {     node *fail;       //失败指针     node *next[kind]; //Tire每个节点的26个子节点（最多26个字母）     int counter;        //是否为该单词的最后一个节点     node(){       //构造函数初始化         fail=NULL;         counter=0;         memset(next,NULL,sizeof(next));     } }*q[500001];          //队列，方便用于bfs构造失败指针 char keyword[51];     //输入的单词 char str[1000001];    //模式串 int head,tail;        //队列的头尾指针 void Insert(char *str, node *root){     node *p=root;     int i=0,index;     while(str[i]) {         index=str[i]-'a';         if(p->next[index]==NULL) p->next[index]=new node();         p=p->next[index];         i++;     }     p->counter++; } void build_ac_automation(node *root){     int i;     root->fail = NULL;     q[head++] = root;     while(head != tail){         node *temp = q[tail++];         node *p = NULL;         for(i=0; i<26; i++){             if(temp->next[i]!=NULL){                 if(temp==root) temp->next[i]->fail = root;                 else {                     p = temp->fail;                     while(p!=NULL) {                         if(p->next[i]!=NULL){                             temp->next[i]->fail=p->next[i];                             break;                         }                        p=p->fail;                     }                     if(p==NULL) temp->next[i]->fail=root;                }// end else.                q[head++]=temp->next[i];            }//end if.         }// end for     }// end while } int query(node *root) {     int i=0, cnt=0, index, len=strlen(str);     node *p=root;     while(str[i]) {         index = str[i]-'a';         while(p->next[index]==NULL && p!=root) p=p->fail;         p = p->next[index];         p = (p==NULL)?root:p;         node *temp=p;        while(temp != root && temp->counter != -1){             cnt+=temp->counter;             temp->counter=-1;             temp=temp->fail;         }         i++;     }     return cnt; } int main(){     int n, t;     scanf("%d",&t);     while(t--) {         head = tail = 0;         node *root=new node();         scanf("%d",&n);         getchar();         while(n--){             gets(keyword);             Insert(keyword,root);         }         build_ac_automation(root);         scanf("%s",str);         printf("%d\n",query(root));     }     return 0; }// 转自：http://www.cppblog.com/mythit/archive/2009/04/21/80633.html
0 0