题目描述
题解
AC自动机模板题。
代码
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
using namespace std;
#define N 100005
char s[N];
int n,sz,ans[N],tot;
int ch[N][130],is_end[N],fail[N];
bool vis[N];
queue <int> q;
void insert(int id)
{
int len=strlen(s),now=0;
for (int i=0;i<len;++i)
{
int x=s[i];
if (!ch[now][x]) ch[now][x]=++sz;
now=ch[now][x];
}
is_end[now]=id;
}
void make_fail()
{
while (!q.empty()) q.pop();
for (int i=0;i<=128;++i)
if (ch[0][i]) q.push(ch[0][i]);
while (!q.empty())
{
int now=q.front();q.pop();
for (int i=0;i<=128;++i)
{
if (!ch[now][i])
{
ch[now][i]=ch[fail[now]][i];
continue;
}
fail[ch[now][i]]=ch[fail[now]][i];
q.push(ch[now][i]);
}
}
}
void ac()
{
int len=strlen(s),now=0;
for (int i=0;i<len;++i)
{
vis[now]=1;
int x=s[i];
int y=ch[now][x];
while (y&&!vis[y])
{
vis[y]=1;
if (is_end[y]) ans[++ans[0]]=is_end[y];
y=fail[y];
}
now=ch[now][x];
}
}
int main()
{
freopen("input.in","r",stdin);
scanf("%d\n",&n);
for (int i=1;i<=n;++i)
{
gets(s);
insert(i);
}
scanf("%d\n",&n);
make_fail();
for (int i=1;i<=n;++i)
{
gets(s);
ans[0]=0;
memset(vis,0,sizeof(vis));
ac();
if (ans[0])
{
printf("web %d: ",i);
sort(ans+1,ans+ans[0]+1);
for (int i=1;i<=ans[0];++i)
printf("%d%c",ans[i]," \n"[i==ans[0]]);
++tot;
}
}
printf("total: %d\n",tot);
}
总结
①多组数据要清零。
②AC自动机中如果单词没有重复,则这个is_end位置只可能有一种单词的情况。