AC自动机 数组实现

AC自动机的实现原理是KMP + 字典树。 学AC自动机之前要先去学KMP 和 字典树。

第一步先构建一个字典树。

 void Insert(){
     int rt = , len = strlen(str);
     for(int i = ; i < len; i++){
         int id = str[i] - 'a';
         if(trie[rt][id] == ){
             cnt[tot] = ;
             fair[tot] = ;
             trie[rt][id] = tot++;
         }
         rt = trie[rt][id];
     }
     cnt[rt]++;
 }

插入字典树

第二步 通过BFS来构造fair指针。

 void Build_tree(){
     queue<int> q;
     q.push();
     int p;
     while(!q.empty()){
         int tmp = q.front();
         q.pop();
         for(int i = ; i < ; i++){
             if(trie[tmp][i] != ){
                 if(tmp == )
                     fair[trie[tmp][i]] = ;
                 else{
                     p = fair[tmp];
                     while(p){
                         if(trie[p][i]){
                             fair[trie[tmp][i]] = trie[p][i];
                             break;
                         }
                         else p = fair[p];
                     }
                     if(!p)  fair[trie[tmp][i]] = ;
                 }
                 q.push(trie[tmp][i]);
             }
         }
     }
 }

构造fair指针

第三步 进行匹配。

统计有多少个单词的出现过。

 int Query(){
     int rt = , ret = , len = strlen(str);
     for(int i = ; i < len; i++){
         int id = str[i] - 'a';
         while(!trie[rt][id] && rt != ) rt = fair[rt];
         rt = trie[rt][id];
         if(rt == ) rt = ;
         int tmp = rt;
         while(tmp != ){
             if(cnt[tmp] >= ){
                 ret += cnt[tmp];
                 cnt[tmp] = -;
             }
             else break;
             tmp = fair[tmp];
         }
     }
     return ret;
 }

统计单词个数。

AC自动机 模板题 HDU-2222 Keywords Search

 #include<bits/stdc++.h>
 using namespace std;
 #define LL long long
 #define ULL unsigned LL
 #define fi first
 #define se second
 #define lson l,m,rt<<1
 #define rson m+1,r,rt<<1|1
 #define max3(a,b,c) max(a,max(b,c))
 #define min3(a,b,c) min(a,min(b,c))
 const int INF = 0x3f3f3f3f;
 const LL mod = 1e9+;
 typedef pair<int,int> pll;
 const int N = 1e5+, M = 1e3;
 char str[N*];
 int trie[N*][];
 int fair[N*];
 int cnt[N*];
 int tot = ;
 void Insert(){
     int rt = , len = strlen(str);
     for(int i = ; i < len; i++){
         int id = str[i] - 'a';
         if(trie[rt][id] == ){
             cnt[tot] = ;
             fair[tot] = ;
             trie[rt][id] = tot++;
         }
         rt = trie[rt][id];
     }
     cnt[rt]++;
 }
 void Build_tree(){
     queue<int> q;
     q.push();
     int p;
     while(!q.empty()){
         int tmp = q.front();
         q.pop();
         for(int i = ; i < ; i++){
             if(trie[tmp][i] != ){
                 if(tmp == )
                     fair[trie[tmp][i]] = ;
                 else{
                     p = fair[tmp];
                     while(p){
                         if(trie[p][i]){
                             fair[trie[tmp][i]] = trie[p][i];
                             break;
                         }
                         else p = fair[p];
                     }
                     if(!p)  fair[trie[tmp][i]] = ;
                 }
                 q.push(trie[tmp][i]);
             }
         }
     }
 }
 int Query(){
     int rt = , ret = , len = strlen(str);
     for(int i = ; i < len; i++){
         int id = str[i] - 'a';
         while(!trie[rt][id] && rt != ) rt = fair[rt];
         rt = trie[rt][id];
         if(rt == ) rt = ;
         int tmp = rt;
         while(tmp != ){
             if(cnt[tmp] >= ){
                 ret += cnt[tmp];
                 cnt[tmp] = -;
             }
             else break;
             tmp = fair[tmp];
         }
     }
     return ret;
 }
 void init(){
     for(int i = ; i < tot; i++){
         for(int j = ; j < ; j++)
             trie[i][j] = ;
     }
     tot = ;
 }
 int main(){
     int T;
     scanf("%d", &T);
     while(T--){
         init();
         int n;
         scanf("%d", &n);
         while(n--){
             scanf("%s", str);
             Insert();
         }
         Build_tree();
         scanf("%s", str);
         printf("%d\n", Query());
     }
     return ;
 }

HDU-2222