题目意思很简单,给定几个模式串,看是否出现在主串中。
AC自动机可解。解法很简单,用所给的模式串构建AC自动机,然后用主串去匹配。每个模式串的末尾记录该串的标号,而中间的字符的标号为0即可,最后跑一次AC自动机就行了。
#include<cstdio>
#include<cstring>
#include<iostream>
#include<utility>
#include<string>
#include<vector>
#include<stack>
#include<algorithm>
#include<queue>
#include<cstdlib>
#include<cmath>
using namespace std;
const int M=1000200;
const int N=550;
const int kind=94;
struct Node
{
int num;
Node *next[kind],*fail;
void init(int a)
{
num=a;
fail=NULL;
memset(next,NULL,sizeof(next));
};
}*root,pp[3*M],*que[2*M];
bool vis[N];
char vir[300];
char web[10100];
int nc;
int n,m;
void init()
{
nc=0;
pp[nc].init(0);
root=&pp[nc++];
}
void insert(char s[],int num)
{
Node *p=root;
for(int i=0;s[i];i++)
{
if(p->next[s[i]-' ']==NULL)
{
pp[nc].init(0);
p->next[s[i]-' ']=&pp[nc++];
}
p=p->next[s[i]-' '];
}
p->num=num;
}
void ac_auto()
{
int front=0,rear=0;
Node *cur,*ptr;
que[rear++]=root;
root->fail=NULL;
while(front<rear)
{
cur=que[front++];
for(int i=0;i<kind;i++)
{
if(cur->next[i]==NULL) continue;
if(cur==root)
{
cur->next[i]->fail=root;
que[rear++]=cur->next[i];
continue;
}
else
{
for(ptr=cur->fail;ptr!=NULL;ptr=ptr->fail)
{
if(ptr->next[i]!=NULL)
{
cur->next[i]->fail=ptr->next[i];
break;
}
}
if(ptr==NULL)
{
cur->next[i]->fail=root;
}
que[rear++]=cur->next[i];
}
}
}
}
int query()
{
Node *r=root,*p,*tmp;
int i=0,cnt=0;
while(web[i])
{
int index=web[i]-' ';
while(r->next[index]==NULL && r!=root)
{
r=r->fail;
}
r=r->next[index];
if(r==NULL) r=root;
tmp=r;
while(tmp!=root)
{
vis[tmp->num]=1;
if(tmp->num!=0) cnt++;
tmp=tmp->fail;
}
i++;
}
return cnt;
}
int main()
{
while(scanf("%d",&n)==1)
{
init();
int cnt=0;
for(int i=1;i<=n;i++)
{
scanf("%s",vir);
insert(vir,i);
}
ac_auto();
scanf("%d",&m);
for(int i=1;i<=m;i++)
{
scanf("%s",web);
memset(vis,0,sizeof(vis));
int temp=query();
if(temp!=0)
{
cnt++;
printf("web %d:",i);
for(int j=1;j<=n;j++)
if(vis[j])
printf(" %d",j);
printf("\n");
}
}
printf("total: %d\n",cnt);
}
return 0;
}