HDU 2222(ac自动机)

最新推荐文章于 2020-09-14 18:22:41 发布

zzulihrs

最新推荐文章于 2020-09-14 18:22:41 发布

阅读量199

点赞数 2

分类专栏：字符串

本文链接：https://blog.youkuaiyun.com/weixin_44070289/article/details/104065676

版权

字符串专栏收录该内容

3 篇文章

订阅专栏

本文深入探讨了AC自动机的原理与应用，通过详细解释其数据结构和算法流程，包括插入单词、获取失败指针及查询过程。文章提供了一个完整的C++实现代码示例，展示了如何构建AC自动机来高效地进行字符串匹配。

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

ac自动机

参考博客：https://blog.youkuaiyun.com/bestsort/article/details/82947639

#include<queue>
#include<cstdlib>
#include<cmath>
#include<cstdio>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cstdio>
#define debug(x) cout<<#x<<" = "<<x<<endl
#define gg cout<<"-----QAQ-----"<<endl
using namespace std;
typedef long long ll;
const int maxn = 500010;

int trie[maxn][26];
int cntword[maxn];
int fail[maxn];
int cnt = 0;

void insertWords(string s)
{
    int root = 0;
    for(int i = 0; i <s.size(); ++i) {
        int next = s[i]-'a';
        if(!trie[root][next])
            trie[root][next] = ++cnt;
        root = trie[root][next];
    }
    cntword[root]++;
}
void getFail()
{
    queue<int> q;
    for(int i = 0; i < 26; ++i) {
        if(trie[0][i]) {
            fail[trie[0][i]] = 0;
            q.push(trie[0][i]);
        }
    }


    while(!q.empty()){
        int now = q.front();q.pop();
        for(int i = 0; i < 26; ++i)  {
            if(trie[now][i]) {
                fail[trie[now][i]] = trie[fail[now]][i];
                q.push(trie[now][i]);
            }
            else trie[now][i] = trie[fail[now]][i];
        }
    }
}

int query(string s)
{
    int now = 0, ans = 0;
    for(int i = 0; i < s.size(); ++i) {
        now = trie[now][s[i]-'a'];
        for(int j = now; j&&cntword[j]!=-1;j=fail[j]) {
            ans += cntword[j];
            cntword[j] = -1;
        }
    }
    return ans;
}
void init()
{
    cnt = 0;
    for(int i = 0; i < 26; ++i) {
        for(int j = 0; j < maxn; ++j)
            trie[j][i] = 0;
    }
    memset(fail, 0, sizeof(fail));
    memset(cntword, 0, sizeof(cntword));
}
void solve()
{
    init();
    int n;string s;
    cin >> n;
    for(int i = 1; i <= n; ++i) {
        cin >> s;
        insertWords(s);
    }
    getFail();
    cin >> s;
    cout<<query(s)<<endl;

}
int main()
{
    int t;
    cin >> t;
    while(t--)
        solve();
    return 0;
}