HDU - 2222 Keywords Search

题目传送门：23333

AC自动机的裸题（虽然我看了好长时间写了2H才2333的A掉QAQ）

开始没看到多组数据=幸好有cl大的提醒

参考：AC自动机详解

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#define rep(j,k,l) for (int j=k;j<=l;j++)
#define N 10005
#define M 1000005

using namespace std;
int n,m,cnt,head,tail,ch[50*N][26],sum[50*N],fail[50*N],queue[50*N];
char s[M];

void add(){

int rt=0,len=strlen(s);
rep(i,0,len-1){

if (!ch[rt][s[i]-'a']) ch[rt][s[i]-'a']=++cnt;
rt=ch[rt][s[i]-'a'];

}
sum[rt]++;

}

void getfail(){

fail[0]=-1;head=0;tail=1;
while (head<tail){

int x=queue[++head];
rep(i,0,25) if (ch[x][i])
queue[++tail]=ch[x][i],
fail[ch[x][i]]=x==0?0:ch[fail[x]][i];
else ch[x][i]=x==0?0:ch[fail[x]][i];

}

}

int getans(){

int ans=0,len=strlen(s),rt=0;
rep(i,0,len-1){

rt=ch[rt][s[i]-'a'];
int k=rt;
while (k!=0){

ans+=sum[k];
sum[k]=0;
k=fail[k];

}

}
return ans;

}

int main(){

scanf("%d",&n);
while (n--){

memset(sum,0,sizeof(sum));
memset(ch,0,sizeof(ch));
scanf("%d",&m);
while (m--){

scanf("%s",s);
add();

}
getfail();
scanf("%s",s);
printf("%d\n",getans());

}

}