Keywords Search（hdu 2222）

题意：给出n个单词，一篇文章，询问有几个单词在文章中出现过。

/*
  AC自动机的裸题。
  题目标号牛的一比。
*/
#include<cstdio>
#include<cstring>
#include<iostream>
#define N 500010
#define M 1000010
using namespace std;
int a[N][],danger[N],point[N],mark[N],q[N],n,size;
char ch[M];
void insert(){
    int len=strlen(ch),now=;
    for(int i=;i<len;i++){
        int t=ch[i]-'a';
        if(!a[now][t])a[now][t]=++size;
        now=a[now][t];
    }
    danger[now]++;
}
void acmach(){//构建自动机
    int h=,t=;
    q[]=;point[]=;
    while(h<t){
        int now=q[++h];
        for(int i=;i<=;i++){
            if(!a[now][i])continue;
            int k=point[now];
            while(!a[k][i])k=point[k];
            point[a[now][i]]=a[k][i];//类似于kmp的失败数组
            q[++t]=a[now][i];
        }
    }
}
void solve(){
    int len=strlen(ch),k=,ans=;
    for(int i=;i<len;i++){
        mark[k]=;
        int t=ch[i]-'a';
        while(!a[k][t])k=point[k];
        k=a[k][t];
        if(mark[k])continue;
        for(int j=k;j;j=point[j]){
            ans+=danger[j];
            danger[j]=;//每个单词只记录一次
        }
    }
    printf("%d\n",ans);
}
int main(){
    int T;scanf("%d",&T);
    while(T--){
        memset(a,,sizeof(a));
        memset(point,,sizeof(point));
        memset(danger,,sizeof(danger));
        memset(mark,,sizeof(mark));
        size=;
        for(int i=;i<=;i++)a[][i]=;
        scanf("%d",&n);
        for(int i=;i<=n;i++){
            scanf("%s",ch);
            insert();
        }
        acmach();
        scanf("%s",ch);
        solve();
    }
    return ;
}