【AC自动机】Keywords Search

【题目链接】

https://loj.ac/problem/10057

【题意】

原题来自：HDU 2222

给定  n 个长度不超过 50 的由小写英文字母组成的单词准备查询，以及一篇长为 m 的文章，问：文中出现了多少个待查询的单词。多组数据。

【题解】

模板题

【代码】

#pragma GCC optimize(2)

#include<queue>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int N = 5e5+1e4;
const int M = 1e6+10;
queue < int > Q ;
typedef struct Aho_Corasick_Automaton{
    int Son[N][26],End[N],Fail[N],idx;
    void Init(){
        idx = 0 ;
        while( !Q.empty() ) Q.pop() ;
        memset(Son , 0 , sizeof Son );
        memset(End , 0 , sizeof End );
        memset(Fail, 0 , sizeof Fail );
    }
    void Insert(char s[]){
        int p = 0 ;
        for(int i=0;s[i];i++){
            int t = s[i] - 'a';
            if( !Son[p][t] )
                Son[p][t] = ++idx;
            p = Son[p][t] ;
        }
        End[p] ++ ;
    }

    void Build(){
        for(int i=0;i<26;i++){
            if( Son[0][i] )
                Fail[Son[0][i]] = 0 ,Q.push(Son[0][i]);
        }

        while( !Q.empty() ){
            int u = Q.front() ;
            Q.pop();

            for(int i=0;i<26;i++){
                if( Son[u][i] ){
                    Fail[Son[u][i]] = Son[Fail[u]][i] ;
                    Q.push( Son[u][i] );
                }else{
                    Son[u][i] = Son[Fail[u]][i];
                }
            }
        }
    }

    void Query(char s[]){
        int p = 0,res = 0;
        for(int i=0;s[i];i++){
            int t = s[i] - 'a';
            p = Son[p][t] ;
            for(int j=p ; j && ~End[j] ; j = Fail[j] ){
                res += End[j] ;
                End[j] = -1 ;
            }
        }
        //printf("%d
",res);
        cout << res << endl ;
    }
}AC_Machine ;
AC_Machine AC ;
char s[M];
int n;
int main(){

    ios_base :: sync_with_stdio(false);
    cin.tie(NULL) , cout.tie(NULL);
    int T;
    cin >> T ;
    while(T--){
        cin >> n ;
        AC.Init();
        for(int i=0;i<n;i++){
            //scanf("%s",s);
            cin >> s ;
            AC.Insert(s);
        }
        AC.Build();
        //scanf("%s",s);
        cin >> s ;
        AC.Query(s);
    }
    return 0;
}