hdu2222 AC自动机

最新推荐文章于 2020-09-14 18:22:41 发布

acceptedwwh

最新推荐文章于 2020-09-14 18:22:41 发布

阅读量636

点赞数

分类专栏： ACM_字符串专项

ACM_字符串专项专栏收录该内容

17 篇文章

订阅专栏

本文介绍了一种使用C++实现的AC自动机（Aho-Corasick自动机）代码实例，该自动机主要用于字符串匹配问题，特别适用于关键词检索场景。文中详细展示了如何构建自动机的失败指针，并通过具体算法实现了文本中关键词的高效查找。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#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