hdoj 2222-Keywords Search解题报告

#include<stdio.h>
#include<string.h>
struct node
{
int cnt;
node *next[26],*fail;
}*rt,qq[500005],*que[500005];//不去新建节点，而是用qq数组去存储，节省时间
char key[55];
char mes[1000005];
int k;
void insert(char *str)//创建字典树的过程
{
int i;
int temp;
node *p=rt;
for(i=0;str[i];i++)
{
temp=str[i]-'a';
if(p->next[temp]==NULL)
{
qq[k].cnt=0;
qq[k].fail=NULL;
memset(qq[k].next,NULL,sizeof(qq[k].next));
p->next[temp]=&qq[k++];
}
p=p->next[temp];
}
p->cnt++;
}
void ac()//类似于kmp的getnext函数
{
int f=0,r=0;
int i;
node *p,*temp;
rt->fail=NULL;
que[r++]=rt;
while(f<r)
{
temp=que[f++];
for(i=0;i<26;i++)
{
if(temp->next[i]==NULL)
continue;
if(temp==rt)
temp->next[i]->fail=rt;
else
{
for(p=temp->fail;p!=NULL;p=p->fail)
{
if(p->next[i]!=NULL)
{
temp->next[i]->fail=p->next[i];
break;
}
}
if(p==NULL)
{
temp->next[i]->fail=rt;
}
}
que[r++]=temp->next[i];
}
}
}
int acs()
{
int ans=0;
int i,id;
node *p=rt,*temp;
for(i=0;mes[i];i++)
{
id=mes[i]-'a';
while(p->next[id]==NULL&&p!=rt)
p=p->fail;
p=p->next[id];
if(p==NULL)
p=rt;
for(temp=p;temp!=NULL&&temp->cnt!=-1;temp=temp->fail)
{
ans+=temp->cnt;
temp->cnt=-1;
}
}
return ans;
}
int main()
{
int t;
scanf("%d",&t);
int n;
int i,j;
while(t--)
{
k=0;
rt=&qq[k++];
memset(rt->next,NULL,sizeof(rt->next));
scanf("%d",&n);
for(i=0;i<n;i++)
{
scanf("%s",key);
insert(key);
}
ac();
scanf("%s",mes);
printf("%d\n",acs());
}
return 0;
}