HDU2222：Keywords Search——题解

http://acm.hdu.edu.cn/showproblem.php?pid=2222

题目大意：

给定 n 个长度不超过 50 的由小写英文字母组成的单词,以及一篇长为 m 的文章,问有多少个单词在文章中出现了,多组数据。

——————————————————————

AC自动机裸题。

（C++AC，G++TLE我能说什么……）

#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
#include<iostream>
#include<queue>
using namespace std;
const int N=1e4+;
const int M=1e6+;
const int L=+;
struct trie{
    int ed;
    int a[];
    int fail;
    void clear(){
    ed=fail=;
    memset(a,,sizeof(a));
    return;
    }
}tree[N*L]={};
char s[M];
int cnt=;
inline void insert(){
    int now=;
    int len=strlen(s);
    for(int i=;i<len;i++){
        int t=s[i]-;
        if(!tree[now].a[t]){
            cnt++;
            tree[now].a[t]=cnt;
        tree[cnt].clear();
        }
        now=tree[now].a[t];
    }
    tree[now].ed++;
    return;
}
int q[N*L];
void getfail(){
    int r=;
    //以下是对第一层的特殊处理
    for(int i=;i<;i++){
        if(tree[].a[i]!=){
            tree[tree[].a[i]].fail=;
        r++;q[r]=tree[].a[i];
        }
    }
    for(int l=;l<=r;l++){
        int u=q[l];
        for(int i=;i<;i++){
            if(tree[u].a[i]!=){
                tree[tree[u].a[i]].fail=tree[tree[u].fail].a[i];
        r++;q[r]=tree[u].a[i];
            }else{
                tree[u].a[i]=tree[tree[u].fail].a[i];
            }
        }
    }
    return;
}
int check(){
    int now=,ans=;
    int len=strlen(s);
    for(int i=;i<len;i++){
        int t=s[i]-;
        now=tree[now].a[t];
        for(int j=now;j&&tree[j].ed!=-;j=tree[j].fail){
            ans+=tree[j].ed;
            tree[j].ed=-;
        }
    }
    return ans;
}
int main(){
    int t;
    cin>>t;
    while(t--){
    tree[].clear();
    cnt=;
    int n;
    cin>>n;
    for(int i=;i<=n;i++){
        cin>>s;
        insert();
    }
    getfail();
    cin>>s;
    cout<<check()<<endl;
    }
    return ;
}