zoukankan      html  css  js  c++  java
  • AC自助机

    http://blog.henix.info/blog/trie-aho-corasick.html

    这个博客说的很清楚了。自己实现了个,留个模版

    View Code
      1 #include<iostream>
      2 #include<string>
      3 #include<algorithm>
      4 using namespace std;
      5 
      6 struct node
      7 {
      8     struct node *fail;
      9     struct node *next[26];
     10     int cnt;
     11     node()
     12     {
     13         cnt=0;
     14         fail=NULL;
     15         for(int i=0;i<26;i++)
     16             next[i]=NULL;
     17     }
     18 };
     19 node *root;
     20 char str[1000001];
     21 int n;
     22 
     23 void insert(char word[])
     24 {
     25     int i=0,len=strlen(word);
     26     node *tmp=root;
     27     if(tmp==NULL)
     28     {
     29         tmp=new node();
     30         root=tmp;
     31     }
     32     while(word[i])
     33     {
     34         int b=word[i]-'a';
     35         if(tmp->next[b]==NULL)
     36         {
     37             tmp->next[b]=new node();
     38         }
     39         if(i==len-1)
     40             tmp->next[b]->cnt++;
     41         tmp=tmp->next[b];
     42         i++;
     43     }
     44 }
     45 
     46 node *q[500010];
     47 int head,tail;
     48 
     49 void addFail()
     50 {
     51     head=tail=0;
     52     q[tail++]=root;
     53     while(head<tail)
     54     {
     55         node *x=q[head++];
     56         int i;
     57         for(i=0;i<26;i++)
     58         {
     59             if(x->next[i]!=NULL)
     60             {
     61                 q[tail++]=x->next[i];
     62                 node *t=x->fail;
     63                 while((t!=NULL) && t->next[i]==NULL)
     64                     t=t->fail;
     65                 if(t==NULL)
     66                     x->next[i]->fail=root;
     67                 else
     68                     x->next[i]->fail=t->next[i];
     69             }
     70         }
     71     }
     72 }
     73 
     74 int match(char word[])
     75 {
     76     int i=0,ans=0;
     77     node *tmp=root;
     78     while(word[i])
     79     {
     80         int b=word[i]-'a';
     81         while(tmp!=NULL && tmp->next[b]==NULL)
     82             tmp=tmp->fail;
     83         if(tmp!=NULL)
     84         {
     85             tmp=tmp->next[b];
     86             node *t=tmp;
     87             while(t!=NULL)
     88             {
     89                 if(t->cnt>0)
     90                 {
     91                     ans+=t->cnt;
     92                     t->cnt=0;
     93                 }
     94                 t=t->fail;
     95             }
     96         }
     97         else
     98             tmp=root;
     99         i++;
    100     }
    101     return ans;
    102 }
    103 
    104 int main()
    105 {
    106     int i,cas;
    107     freopen("D:\\in.txt","r",stdin);
    108     scanf("%d",&cas);
    109     while(cas--)
    110     {
    111         scanf("%d",&n);
    112         root=NULL;
    113         for(i=0;i<n;i++)
    114         {
    115             scanf("%*c%s",str);
    116             insert(str);
    117         }
    118         scanf("%*c%s",str);
    119         addFail();
    120         int ans=match(str);
    121         printf("%d\n",ans);
    122     }
    123     return 0;
    124 }
    View Code
      1 #include<iostream>
      2 #include<string>
      3 #include<algorithm>
      4 using namespace std;
      5 
      6 struct node
      7 {
      8     int cnt;
      9     struct node *fail;
     10     struct node *next[26];
     11 };
     12 
     13 node root[500010];
     14 int n,num;
     15 char str[1000001];
     16 
     17 void insert(char word[])
     18 {
     19     int i=0;
     20     node * tmp=root;
     21     while(word[i])
     22     {
     23         int b=word[i]-'a';
     24         if(tmp->next[b]==NULL)
     25         {
     26             tmp->next[b]=root+num;
     27             memset(root+num,0,sizeof(struct node));
     28             num++;
     29         }
     30         tmp=tmp->next[b];
     31         i++;
     32     }
     33     tmp->cnt++;
     34 }
     35 
     36 node *q[500010];
     37 int head,tail;
     38 
     39 void add_Fail()
     40 {
     41     head=tail=0;
     42     q[tail++]=root;
     43     while(head<tail)
     44     {
     45         node *x=q[head++];
     46         for(int i=0;i<26;i++)
     47         {
     48             if(x->next[i]!=NULL)
     49             {
     50                 q[tail++]=x->next[i];
     51                 node *t=x->fail;
     52                 while((t!=NULL) && t->next[i]==NULL)
     53                     t=t->fail;
     54                 if(t==NULL)
     55                     x->next[i]->fail=root;
     56                 else
     57                     x->next[i]->fail=t->next[i];
     58             }
     59         }
     60     }
     61 }
     62 
     63 int ans;
     64 
     65 void match(char word[])
     66 {
     67     int i=0;
     68     node *tmp=root;
     69     while(word[i])
     70     {
     71         int b=word[i]-'a';
     72         while(tmp!=NULL && tmp->next[b]==NULL)
     73             tmp=tmp->fail;
     74         if(tmp!=NULL)
     75         {
     76             tmp=tmp->next[b];
     77             node *t=tmp;
     78             while(t!=NULL)
     79             {
     80                 if(t->cnt>0)
     81                 {
     82                     ans+=t->cnt;
     83                     t->cnt=0;
     84                 }
     85                 t=t->fail;
     86             }
     87         }
     88         else
     89             tmp=root;
     90         i++;
     91     }
     92 }
     93 
     94 int main()
     95 {
     96     int cas,i;
     97     //freopen("D:\\in.txt","r",stdin);
     98     scanf("%d",&cas);
     99     while(cas--)
    100     {
    101         memset(root,0,sizeof(struct node));
    102         num=1;
    103         scanf("%d",&n);
    104         for(i=0;i<n;i++)
    105         {
    106             scanf("%*c%s",str);
    107             insert(str);
    108         }
    109         add_Fail();
    110         ans=0;
    111         scanf("%*c%s",str);
    112         match(str);
    113         printf("%d\n",ans);
    114     }
    115     return 0;
    116 }
  • 相关阅读:
    Python合集之文件操作(二)
    Python合集之文件操作(一)
    Python合集之异常(二)
    Python合集之异常(一)
    Python合集之模块(五)
    Visual Studio 配置额外工具 Windows Terminal 等
    CMakeList.txt
    alpha智能图像(全栈的进阶之路)
    位运算实现多状态控制
    缓存函数 memorize
  • 原文地址:https://www.cnblogs.com/ka200812/p/2680905.html
Copyright © 2011-2022 走看看