hdu 2222 Keywords Search（ac自动机）

Posted 2020-08-20 Gealo

题目链接 http://acm.hdu.edu.cn/showproblem.php?pid=2222

题意：给你一系列子串，再给你一个主串问你主串一共有几个匹配子串

原来使用字典树写的但数据有点大TLE了，然后就开始学习ac自动机了，ac自动机就像是多串匹配的kmp原理也是类似的。

这题可以练一下手写ac自动机模版。

 

#include <iostream>
#include <cstring>
#include <cstdio>
#include <queue>
using namespace std;
const int M = 5e5 + 50;
int Next[M][26] , fail[M] , End[M] , root , L , cnt;
char str[60] , sl[M * 2];
int newnode() {
    for(int i = 0 ; i < 26 ; i++) {
        Next[L][i] = -1;
    }
    End[L++] = 0;
    return L - 1;
}
void init() {
    L = 0;
    root = newnode();
}
void build(char s[]) {
    int len = strlen(s);
    int now = root;
    for(int i = 0 ; i < len ; i++) {
        int id = s[i] - ‘a‘;
        if(Next[now][id] == -1) {
            Next[now][id] = newnode();
        }
        now = Next[now][id];
    }
    ++End[now];
}
void get_fail() {
    queue<int>q;
    fail[root] = root;
    for(int i = 0 ; i < 26 ; i++) {
        if(Next[root][i] == -1) {
            Next[root][i] = root;
        }
        else {
            fail[Next[root][i]] = root;
            q.push(Next[root][i]);
        }
    }
    while(!q.empty()) {
        int now = q.front();
        q.pop();
        for(int i = 0 ; i < 26 ; i++) {
            if(Next[now][i] == -1) {
                Next[now][i] = Next[fail[now]][i];
            }
            else {
                fail[Next[now][i]] = Next[fail[now]][i];
                q.push(Next[now][i]);
            }
        }
    }
}
void match(char s[]) {
    int now = root;
    int len = strlen(s);
    for(int i = 0 ; i < len ; i++) {
        int id = s[i] - ‘a‘;
        now = Next[now][id];
        int temp = now;
        while(temp != root) {
            cnt += End[temp];
            End[temp] = 0;
            temp = fail[temp];
        }
    }
}
int main()
{
    int t;
    scanf("%d" , &t);
    while(t--) {
        int n;
        scanf("%d" , &n);
        init();
        for(int i = 0 ; i < n ; i++) {
            scanf("%s" , str);
            build(str);
        }
        get_fail();
        scanf("%s" , sl);
        cnt = 0;
        match(sl);
        printf("%d\n" , cnt);
    }
    return 0;
}