/*
寻找都有哪些子串
不能保证是字母或数字,所以子节点有差不多130个
*/
#include <iostream>
#include<string.h>
#include<stdlib.h>
using namespace std;
int biaoshi[510];
const int kind = 130;
int dulist[10];
struct node{
node *fail; //失败指针
node *next[kind];
int count; //是否为该单词的最后一个节点
node(){ //构造函数初始化
fail=NULL;
count=0;
memset(next,NULL,sizeof(next));
}
}*q[5000000]; //队列,方便用于bfs构造失败指针
char keyword[510]; //输入的单词
char str[1000010]; //模式串
int head,tail; //队列的头尾指针
void insert(char *str,node *root,int no){
node *p=root;
int i=0,index;
while(str[i]){
index=str[i]-31;
if(p->next[index]==NULL) p->next[index]=new node();
p=p->next[index];
i++;
}
p->count=no;//初始化为0,++后为1,表示是一个单词的结尾
}
void build_ac_automation(node *root)
{
int i;
root->fail=NULL;
q[head++]=root;
while(head!=tail)
{
node *temp=q[tail++];//拿到一个节点
node *p=NULL;
for(i=0;i<130;i++)
{
if(temp->next[i]!=NULL)//若其i孩子非空
{
if(temp==root) //他自己是头,其孩子的失败指针指向头
temp->next[i]->fail=root;
else{ //普通节点
p=temp->fail; //指向自己的失败指针
while(p!=NULL)
{
if(p->next[i]!=NULL)//失败指针有i孩子
{
temp->next[i]->fail=p->next[i]; //当前节点的i孩子的失败指针指向失败指针的i孩子,然后跳出
break;
}
p=p->fail; //继续找失败指针
}
if(p==NULL) //若失败指针为空
temp->next[i]->fail=root; //当前节点的i孩子的失败指针指向头
}
q[head++]=temp->next[i]; //进去的都是定义过失败指针的,故此过程是给其孩子定义失败指针
}
}
}
}
int query(node *root)
{
int i=0,cnt=0,index,len=strlen(str);
node *p=root;
while(str[i])
{
index=str[i]-31; //计算孩子的位置
while(p->next[index]==NULL && p!=root)
p=p->fail; //若没有i孩子节点,通过失败指针找与这之前相同的有i孩子的节点
p=p->next[index]; //指向其i孩子
p=(p==NULL)?root:p;
node *temp=p;
while(temp!=root && temp->count)
{
if(temp->count&&!biaoshi[temp->count])//主要是改这里
{
cnt++;
biaoshi[temp->count]=1;
}
temp=temp->fail;
}
i++;
}
return cnt;
}
int cmp(const void *a,const void *b)
{
int *c=(int*)a,*d=(int*)b;
return *c-*d;
}
void print(int ji,int na,int n)
{
printf("web %d:",na);
for(int i=1;i<=n;++i)
{
if(biaoshi[i])
{
printf(" %d",i);
}
}
printf("\n");
}
int main(){
int n,t,i,j=0;
while(scanf("%d",&n)!=EOF)
{
head=tail=0;
node *root=new node();
getchar();
for(i=1;i<=n;i++)
{
gets(keyword);
insert(keyword,root,i);
}
build_ac_automation(root);
scanf("%d",&t);
for(i=1;i<=t;++i)
{
memset(biaoshi,0,sizeof(biaoshi));
scanf("%s",str);
int ge=query(root);
if(ge)
{
print(ge,i,n);
j++;
}
}
printf("total: %d\n",j);
}
return 0;
}
hdu 2896 病毒侵袭 -- AC自动机
最新推荐文章于 2020-07-27 19:43:52 发布