[模板] - AC自动机 - 动态 - 感谢小太阳

最新推荐文章于 2024-03-15 14:10:50 发布

原创最新推荐文章于 2024-03-15 14:10:50 发布 · 328 阅读

0 ·

CC 4.0 BY-SA版权

模板相关专栏收录该内容

8 篇文章

订阅专栏

本文详细介绍AC自动机的基本概念、数据结构定义及其核心算法实现，包括构建失败指针、字符串匹配过程等，并提供了一个完整的C++代码示例。通过本教程，读者可以深入理解AC自动机的工作原理并学会如何在实际编程中应用。

#include<bits/stdc++.h>
using namespace std;
const int type_N = 26;///
const int maxn = 100010;///length of text

int idx(char ch){///Ð¡Ð´×ÖÄ¸idx()
    return ch - 'a';
}

struct Trie{
    Trie *nxt[type_N], *fail;
    int finish;

    void init(){
        for(int i = 0; i < type_N; ++ i)
            nxt[i] = NULL;
        fail = NULL;
        finish = 0;
    }
}root;
int n;
char in[30], text[maxn];

void Insert(char *s){
    int index;
    Trie *now = &root, *tmp;
    while(*s){
        index = idx(*s);
        if(!now->nxt[index]){
            tmp = (Trie*) malloc(sizeof(Trie));
            tmp->init();
            now->nxt[index] = tmp;
        }
        now = now->nxt[index];
        ++ s;
    }
    now->finish ++;
}

void Get_fail(){
    queue<Trie*> Q;
    Q.push(&root);
    Trie *now, *anc;
    while(!Q.empty()){
        now = Q.front();
        Q.pop();
        for(int i = 0; i < type_N; ++ i){
            if(now->nxt[i]){
                Q.push(now->nxt[i]);
                if(now == &root){
                    now->nxt[i]->fail = &root;
                    continue;
                }
                anc = now->fail;
                while(anc){
                    if(anc->nxt[i]){
                        now->nxt[i]->fail = anc->nxt[i];
                        break;
                    }
                    anc = anc->fail;
                }
                if(!anc)
                    now->nxt[i]->fail = &root;
            }
        }
    }
}

int AC_Automaton(char *s){///
    Trie *now = &root, *match;
    int index, ret = 0;
    while(*s){
        index = idx(*s);
        while(now != &root && !now->nxt[index])
            now = now->fail;
        now = now->nxt[index];
        match = now;
        if(!now)now = &root;
        while(match){
            ret += match->finish;
            match = match->fail;
        }
        ++ s;
    }
    return ret;
}

void Del(Trie *now = &root){
    for(int i = 0; i < type_N; ++ i){
        if(now->nxt[i]){
            Del(now->nxt[i]);
            free(now->nxt[i]);
        }
    }
}

void init(){
    Del();
    root.init();
}

int main(){
    while(~scanf("%d", &n)){
        init();
        while(n--){
            scanf("%s", in);
            Insert(in);
        }
        Get_fail();
        while(~scanf("%s", text))///"ctrl+Z" to quit
            printf("%d\n", AC_Automaton(text));
    }
    return 0;
}