hdu 2222 Keywords Search

hdu 2222 Keywords Search

题意：

给出n个单词，然后给你一篇文章，问给出的单词在文章中出现的次数。

限制：

0 <= n <= 1e4; 单词只包含小写字母; 长度小于50;

文章长度 <= 1e6

思路：

ac自动机模板题

/*hdu 2222 Keywords Search
题意：
给出n个单词，然后给你一篇文章，问给出的单词在文章中出现的次数。
限制：
0 <= n <= 1e4; 单词只包含小写字母; 长度小于50;
文章长度 <= 1e6
思路：
ac自动机模板题
*/
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=500001;
int n;
int que
;
char str[2*N];
struct acm{
int cnt;
int next
[26],sum
,fail
;
bool vis
;
void clear(){
for(int i=1;i<=cnt;++i){
vis[i]=sum[i]=fail[i]=0;
for(int j=0;j<26;++j)
next[i][j]=0;
}
cnt=1;
for(int i=0;i<26;++i) next[0][i]=1;
}
void insert(char str[]){
int now=1;
int len=strlen(str);
for(int i=0;i<len;++i){
if(!next[now][str[i]-'a']) next[now][str[i]-'a']=++cnt;
now=next[now][str[i]-'a'];
}
++sum[now];
}
void buildFail(){
int head=0,tail=1;
que[0]=1;
fail[1]=0;
while(head!=tail){
int now=que[head];head++;
for(int i=0;i<26;++i){
int ch=next[now][i];
if(!ch) continue;
int tmp=fail[now];
while(!next[tmp][i]) tmp=fail[tmp];
fail[ch]=next[tmp][i];
que[tail++]=ch;
}
}
}
void gao(char str[]){
int ans=0,now=1,len=strlen(str);
for(int i=0;i<len;++i){
vis[now]=1;
while(!next[now][str[i]-'a']) now=fail[now];
now=next[now][str[i]-'a'];
if(!vis[now])
for(int j=now;j;j=fail[j]){
ans+=sum[j];
sum[j]=0;
}
}
printf("%d\n",ans);
}
void debug(){
for(int i=0;i<=cnt;++i){
printf("id=%3d,fail=%3d,sum=%3d,chi=[",i,fail[i],sum[i]);
puts("");
for(int j=0;j<26;++j)
printf("%2c",j+'a');
puts("");
for(int j=0;j<26;j++)
printf("%2d",next[i][j]);
puts("");
printf("]\n");
}
}
}acm;
int main(){
int T;
scanf("%d",&T);
while(T--){
scanf("%d",&n);
acm.clear();
for(int i=0;i<n;++i){
scanf("%s",str);
acm.insert(str);
}
acm.buildFail();
//acm.debug();
scanf("%s",str);
acm.gao(str);
}
return 0;
}