题意:
在目标串种找出单词出现的种类数。
题解:
直接搞。
#include<iostream>
#include<math.h>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<vector>
#include<map>
#include<set>
using namespace std;
#define B(x) (1<<(x))
typedef long long ll;
const int oo=0x3f3f3f3f;
const ll OO=1LL<<61;
const ll MOD=10007;
const int maxn=500005;
char str[1000005];
struct AC
{
int next[500005][26],fail[500005],end[500005],Q[500005];
int root,cnt;
void Init()
{
cnt=0;
root=newNode();
}
int newNode()
{
for(int i=0;i<26;i++)
next[cnt][i]=-1;
end[cnt++]=0;
return cnt-1;
}
void Insert(char buff[])
{
int now=root;
int len=strlen(buff);
for(int i=0,k;i<len;i++)
{
k=buff[i]-'a';
if(next[now][k]==-1)
next[now][k]=newNode();
now=next[now][k];
}
end[now]++;///计算这种单词的数量
}
void build()
{
int front,rear;
front=rear=0;
int now=root;
for(int i=0;i<26;i++)
{
if(next[now][i]==-1)
next[now][i]=root;
else
{
fail[next[now][i]]=root;
Q[rear++]=next[now][i];
}
}
while(front<rear)
{
now=Q[front++];
for(int i=0;i<26;i++)
{
if(next[now][i]==-1)
next[now][i]=next[fail[now]][i];///同下一样,只不过因为下个是空所以只接用next[now][i],非控就用fail来指向
else
{
fail[next[now][i]]=next[fail[now]][i];///适配只想fail指针对应节点的相同字母处,注意是相同字母处
Q[rear++]=next[now][i];
}
}
}
}
int Search(char buff[])
{
int now=root,res=0;
int len=strlen(buff);
for(int i=0;i<len;i++)
{
now=next[now][buff[i]-'a'];
int temp=now;
while(temp!=root)///扫描
{
res+=end[temp];
end[temp]=0;///防止重复计算
temp=fail[temp];
}
}
return res;
}
void Debug()
{
for(int i=0;i<cnt;i++)
{
printf("id=%3d fail=%3d end=%3d child=[",i,fail[i],end[i]);
for(int j=0;j<26;j++)
printf(" %3d",next[i][j]);
printf(" ]\n");
}
}
};
AC ac;
int main()
{
int T,n;
scanf("%d",&T);
while(T--)
{
ac.Init();
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%s",str);
ac.Insert(str);
}
ac.build();
scanf("%s",str);
printf("%d\n",ac.Search(str));
}
return 0;
}