HUD2222(ac自动机)

最新推荐文章于 2022-11-15 11:48:59 发布

原创最新推荐文章于 2022-11-15 11:48:59 发布 · 448 阅读

0 ·

CC 4.0 BY-SA版权

模板同时被 2 个专栏收录

31 篇文章

订阅专栏

字符串

9 篇文章

订阅专栏

本文介绍了一道标准AC自动机模板题的实现方法，详细解释了如何构建Trie树、插入字符串、计算失败指针及查找模式串的过程。

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

标准ac自动机模板题

题目链接:https://vjudge.net/problem/HDU-2222

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

const int MAXNODE = 550000;
const int SIGMA_SIZE = 26;

int fail[MAXNODE]; //失配数组
int last[MAXNODE]; //后缀链接
int ans;

//构造trie
int ch[MAXNODE][SIGMA_SIZE];
int val[MAXNODE];
int sz;
void init()
{
    ans = 0;
    sz = 1;
    memset(ch[0], 0, sizeof(ch[0]));
}

void insert(char *s)
{
    int u = 0;
    int len = strlen(s);

    for(int i = 0; i < len; ++i)
    {
        int c = s[i] - 'a';
        if(!ch[u][c])
        {
            memset(ch[sz], 0, sizeof(ch[sz]));
            val[sz] = 0;
            ch[u][c] = sz++;
        }
        u = ch[u][c];
    }
    val[u]++;
}


//计算fail指针
void getFail()
{
    queue<int> q;
    fail[0] = 0;
    for(int c = 0; c < SIGMA_SIZE; c++)
    {
        int u = ch[0][c];
        if(u)
        {
            fail[u] = 0; q.push(u); last[u] = 0;
        }
    }
    while(!q.empty())
    {
        int r = q.front(); q.pop();
        for(int c = 0;c < 26;c++)
        {
            int u = ch[r][c];
            if(!u){ ch[r][c] = ch[fail[r]][c]; continue;}
            q.push(u);
            int v = fail[r];
            while(v && !ch[v][c]) v = fail[v];
            fail[u] = ch[v][c];
            last[u] = val[fail[u]] ? fail[u] : last[fail[u]];
        }
        //cout << last[r] <<endl;
    }
}

void solve(int j)
{
    //cout << j <<endl;
    if(!j) return;
    if(val[j])
    {
        ans += val[j];
        val[j] = 0;
    }
    solve(last[j]);
}
void find(char* T)
{
    int len = strlen(T),j = 0;
    getFail();
    for(int i = 0; i < len; i++)
    {
        j = ch[j][T[i] - 'a'];
        //cout << val[j] <<endl;
        if(val[j]) solve(j);
        else if(last[j]) solve(last[j]);
    }
}

int main()
{
    int t, n;
    char dic[100],str[1100000];
    scanf("%d", &t);
    while(t--)
    {
        init();
        scanf("%d", &n);
        while(n--)
        {
            scanf("%s", dic);
            insert(dic);
        }
        scanf("%s", str);
        find(str);
        printf("%d\n", ans);
    }
    return 0;
}