做了这题,才知道,所谓的keyword可能出现好多次。
#include<stdio.h>
#include<string.h>
#include<queue>
using namespace std;
char a[51];
#define maxn 500001
char c[1000010];
int ch[maxn][26];
int fail[maxn],last[maxn];
int val[maxn];
int T,n,tot,root,nex,cur,ans;
queue<int>p;
int newnode()
{
memset(ch[tot],0,sizeof(ch[tot]));
val[tot]=0;
return tot++;
}
void insert(char *x)
{
int len=strlen(x);
int now=root;
for(int i=0;i<len;i++)
{
int &temp=ch[now][x[i]-'a'];
if(!temp)
temp=newnode();
now=temp;
}
val[now]++;
}
void getfail()
{
fail[root]=root;
last[root]=root;
int i;
for(i=0;i<26;i++)
{
nex=ch[root][i];
if(nex)
{
fail[nex]=last[nex]=root;
p.push(nex);
}
}
while(!p.empty())
{
cur=p.front();
p.pop();
for(i=0;i<26;i++)
{
nex=ch[cur][i];
if(!nex)
ch[cur][i]=ch[fail[cur]][i];
else
{
fail[nex]=ch[fail[cur]][i];
last[nex]=val[fail[nex]]?fail[nex]:last[fail[nex]];
p.push(nex);
}
}
}
}
void query(char *x)
{
int len=strlen(x),i;
ans=0;
cur=root;
for(i=0;i<len;i++)
{
cur=ch[cur][x[i]-'a'];
int temp=cur;
while(temp!=root&&val[temp])
{
ans+=val[temp];//不是ans++;
val[temp]=0;
temp=last[temp];
}
}
printf("%d\n",ans);
}
int main()
{
scanf("%d",&T);
while(T--)
{
scanf("%d",&n);
int i;
tot=0;
root=newnode();
for(i=0;i<n;i++)
{
scanf("%s",a);
insert(a);
}
getfail();
scanf("%s",c);
query(c);
}
return 0;
}