AC 自动机 水题 hdu 3065

AC自动机 :

AC自动机算法分为3步：构造一棵Trie树，构造失败指针和模式匹配过程。

在学习 AC 自动机 之前 ， 一般应先学会 字典树 （也就是 tire树 ） 和 KMP 算法 （fail指针的构造 就是这个思想）。

用模拟指针实现的AC自动机

#include <iostream>#include <cstdio>#include <cstring>#include <queue>using namespace std;const int maxn = 10000100;struct Node {   int num, fail;   int next[26];   void init  (){     for(int i = 0; i<=25 ; i++){        next[i] = 0;     }     fail = -1;     num = 0;   }}node [maxn];int n;char s[1010][55];int tot = 0 ;void insert(char s[]){    int cur = 0;    for(int i=0;s[i];i++){       // cout << s[i]<<" ***";       int t = s[i] - 'A' ;       if(node[cur].next[t]){          cur = node[cur].next[t];     // cout << "cur:" <<cur <<endl;       }else{         node[cur].next[t] = ++tot;     node[tot].init();     cur = tot;       }    }}void get_fail (){  queue<int > Q;  Q.push(0);  while(!Q.empty()){      int now = Q.front();      Q.pop();      for(int i= 0;i<26;i++){          if(node[now].next[i]){//          cout <<"haha"<<endl;           int p = node[now].fail ;           int q = node[now].next[i];              while(p!=-1&& !node[p].next[i]){            p = node[p].fail;          }          if(p == -1 ) node[q].fail  = 0;              else { node[q].fail = node[p].next[i];//            cout <<q<<"fail "<< node[q].fail<<endl;          }          Q.push(q);      }      }  }}void match (char s[]){    int len = strlen(s);    int rt = 0 ;    int p,ans;    for(int i=0 ;i < len ; i++){//        cout << s[i]<<"&&&& ";       int t = s[i]-'A';      if(t>25||t<0){        rt = 0 ;    continue;      }      p = rt ;      while(p!=-1&& !node[p].next[t]){         p = node[p].fail;      }      if(p==-1) rt = 0 ;      else rt = node[p].next[t];      node[rt].num ++ ;   //   cout <<rt << "rt"<<endl;    }}int query(char s[]){    int p= 0 , rt= 0 ;    for(int i= 0 ;s[i];i++){         int t = s[i] - 'A';     if(node[p].next[t]){         p = node[p].next[t];     }else{      return 0 ;     }    }    return node[p].num;}char  mode[2000100];int main(){   while( ~scanf("%d",&n)){  for(int i=1;i<=n;i++){     scanf("%s",s[i]);  }  node[0].init();  tot = 0;  for(int i=1;i<=n ;i++){     insert(s[i]);  }   get_fail(); // cout <<"***"<<endl;  scanf("%s",mode);  match(mode); // cout <<"**(()"<<endl;  for(int i=1;i<= n;i++){//      cout <<s[i]<<endl;      int ans = query(s[i]);  //    cout << ans <<endl;      if(ans){         cout << s[i]<<": "<<ans<<endl;      }   } } return  0;}