hdoj--2222 Keywords Search(AC自动机)

最新推荐文章于 2019-08-10 10:40:00 发布

exmy

最新推荐文章于 2019-08-10 10:40:00 发布

阅读量444

点赞数

CC 4.0 BY-SA版权

分类专栏：字符串字符串--AC自动机文章标签： hdoj ac自动机

本文链接：https://blog.youkuaiyun.com/xushao_Movens/article/details/54647625

字符串同时被 2 个专栏收录

17 篇文章

订阅专栏

字符串--AC自动机

2 篇文章

订阅专栏

本文介绍了一种基于AC自动机的多模匹配算法实现。该算法能够高效地在文本中搜索多个模式串，适用于诸如关键词查找等应用场景。文中详细展示了AC自动机的构建过程，包括插入模式串、构建失败指针以及搜索算法的具体实现。

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

problem link

题解

多模匹配。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <string>
#include <cstring>
#include <queue>
using namespace std;

struct Node{
    Node* fail;
    Node* next[26];
    int cnt;
    Node(){
        fail = NULL;
        cnt = 0;
        memset(next, 0, sizeof(next));
    }
};

struct AC_automation{
    Node* root;
    AC_automation(){
        root = new Node();
    }
    void insert(string s){
        Node* p = root;
        for(int i = 0; i < s.length(); ++i){
            int id = s[i] - 'a';
            if(!p->next[id]) p->next[id] = new Node();
            p = p->next[id];
        }
        p->cnt++;
    }

    void build_fail(){
       root->fail = NULL;
        queue<Node*> Q;
        Q.push(root);
        while(!Q.empty()){
            Node* cur = Q.front(); Q.pop();
            for(int i = 0; i < 26; ++i){
                if(cur->next[i]){
                    if(cur == root){
                        cur->next[i]->fail = root;
                    }else{
                        Node* p = cur->fail;
                        while(p){
                            if(p->next[i]){
                                cur->next[i]->fail = p->next[i];
                                break;
                            }
                            p = p->fail;
                        }
                        if(p == NULL){
                            cur->next[i]->fail = root;
                        }
                    }
                    Q.push(cur->next[i]);
                }
            }
        }
    }

    int search(string text){
        int cnt = 0;
        Node* p = root;
        for(int i = 0; i < text.length(); ++i){
            int id = text[i] - 'a';
            while(p->next[id] == NULL && p != root) p = p->fail;
            p = p->next[id];
            if(p == NULL) p = root;

            Node* tmp = p;
            while(tmp != root && tmp->cnt != -1){
                cnt += tmp->cnt;
                tmp->cnt = -1;
                tmp = tmp->fail;
            }
        }
        return cnt;
    }

    void clear(){
        clear(root);
    }

    void clear(Node* p){
        if(p){
            for(int i = 0; i < 26; ++i){
                if(p->next[i]) clear(p->next[i]);
            }
            delete p;
        }
    }

};


int main(){
#ifdef EXMY
freopen("data.in", "r", stdin);
#endif // EXMY

    ios::sync_with_stdio(false);
    int n, t;
    string s;

    cin >> t;
    while(t--){
        AC_automation ac;
        cin >> n;
        for(int i = 0; i < n; ++i){
            cin >> s;
            ac.insert(s);
        }
        ac.build_fail();
        cin >> s;
        cout << ac.search(s) << endl;
    }

    return 0;
}