AC-自动机（AC-Automachine）

最新推荐文章于 2024-04-02 09:44:53 发布

inrence

最新推荐文章于 2024-04-02 09:44:53 发布

阅读量738

点赞数

CC 4.0 BY-SA版权

分类专栏： Algorithms 文章标签：字符串

本文链接：https://blog.youkuaiyun.com/inrence/article/details/51719993

Algorithms 专栏收录该内容

15 篇文章

订阅专栏

本文介绍了一种使用字典树（Trie）的数据结构实现，并在此基础上扩展了AC自动机的功能。主要内容包括字典树的基本定义、插入操作、获取失败指针的方法以及查询过程。通过这些操作，可以高效地进行字符串匹配和搜索任务。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

代码

struct trie{

    int size, indx[maxs][26], word[maxs], fail[maxs];
    bool jump[maxs];

    int idx(char ff){return ff - 'a';}

    void insert(char s[]){

        int u = 0;
        for(int i = 0; s[i]; ++i){
            int k = idx(s[i]);
            if(!indx[u][k]) indx[u][k] = ++size;
            u = indx[u][k];
        }
        word[u] = 1;
        jump[u] = true;
    }

    void get_fail(){

        queue<int> que;
        int head = 0, tail = 0;
        que.push(0);

        while(!que.empty()){
            int u = que.front();
            que.pop();
            for(int k = 0; k < 26; ++k){
                if(!indx[u][k]) continue;
                int v = indx[u][k];
                int p = fail[u];
                while(p && !indx[p][k]) p = fail[p];
                if(indx[p][k] && indx[p][k] != v) p = indx[p][k];
                fail[v] = p;
                jump[v] |= jump[p];
                que.push(v);
            }
        }
    }

    int query(char s[]){

        int rtn = 0, p = 0;
        int flag[maxs];
        memcpy(flag, word, sizeof flag);
        for(int i = 0; s[i]; ++i){
            int k = idx(s[i]);
            while(p && !indx[p][k]) p = fail[p];
            p = indx[p][k];
            int v = p;
            while(jump[v]){
                rtn += flag[v];
                flag[v] = 0;
                v = fail[v];
            }
        }
        return rtn;
    }

} dict;