病毒侵袭（ac自动机模板）

最新推荐文章于 2018-08-15 19:13:49 发布

原创最新推荐文章于 2018-08-15 19:13:49 发布 · 323 阅读

0 ·

CC 4.0 BY-SA版权

ac自动机专栏收录该内容

2 篇文章

订阅专栏

本文详细介绍了AC自动机的实现原理及其在字符串匹配问题中的应用。通过具体的代码示例，展示了如何构建AC自动机，包括节点初始化、插入操作、失败指针设置以及搜索过程。此外，还提供了一个完整的C++实现案例，帮助读者更好地理解这一高效算法。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
using namespace std;
const int sigma_size = 128;
const int maxnode = 500005;
int cnt[550],n,m,st;
struct Trie{
    int ch[maxnode][sigma_size];
    int val[maxnode];
    int f[maxnode];
    int root,L;
    int newnode(){
        for(int i = 0;i < sigma_size;i++){
            ch[L][i] = -1;
        }
        val[L++] = -1;
        return L - 1;
    }
    void init()
    {
        L = 0;
        root = newnode();
    }
    int idx(char x){
        return x - ' ';
    }
    void Insert(char *s,int v){
        int len = strlen(s);
        int now = root;
        for(int i = 0;i < len;i++){
            int c = idx(s[i]);
            if(ch[now][c] == -1){
                ch[now][c] = newnode();
            }
            now = ch[now][c];
        }
        val[now] = v;
    }
    void getFail(){
        queue<int> q;
        f[root] = root;
        for(int i = 0;i < sigma_size;i++){
            if(ch[root][i] == -1){
                ch[root][i] = root;
            }
            else {
                f[ch[root][i]] = root;
                q.push(ch[root][i]);
            }
        }
        while(!q.empty()){
            int now = q.front();q.pop();
            for(int i = 0;i < sigma_size;i++){
                if(ch[now][i] == -1){
                    ch[now][i] = ch[f[now]][i];
                }
                else
                {
                    f[ch[now][i]] = ch[f[now]][i];
                    q.push(ch[now][i]);
                }
            }
        }
    }
    void Find(char *s){
        int len = strlen(s);
        int now = root;
        for(int i = 0;i < len;i++){
            int c = idx(s[i]);
            now = ch[now][c];
            int tmp = now;
            while(tmp != root){
                if(val[tmp] != -1){
                cnt[val[tmp]] = 1;
                //val[tmp] = 0;
                }
                tmp = f[tmp];
            }
        }
    }
};


Trie ac;
char buf[10050];
int main()
{
    while(scanf("%d",&n)!=EOF){
        ac.init();
        for(int i = 1;i <= n;i++){
            scanf("%s",buf);
            ac.Insert(buf,i);
        }
        ac.getFail();
        scanf("%d",&m);
        int sum = 0;
        memset(cnt,0,sizeof(cnt));
       // cout<<cnt[0]<<"#"<<endl;
        for(int i = 1;i <= m;i++){
            scanf("%s",buf);
            int flag = 0;
            ac.Find(buf);
          //  cout<<cnt[0]<<"#"<<endl;
            for(int j = 0;j <= n;j++){
                if(cnt[j] == 1){
                    if(!flag) {
                        printf("web %d:",i);
                        flag = 1;
                        sum ++;
                    }
                    cnt[j] = 0;
                    printf(" %d",j);
                }
            }
            if(flag) printf("\n");
            //cout<<cnt[0]<<"#"<<endl;
        }
        printf("total: %d\n",sum);
    }
    return 0;
}