AC自动机讲解
ac快乐机模板题……
对照模板写完以后……内心崩溃.jpg
注意insert和cal_fail都放在输入m前面,插入和形成指针只要一次就好了
快乐提交之后……
首先runtime error,上网查了一下不能s[i]-'a’而要直接s[i]转化成数字。对啊不然下标就变成负数了。。amazing
然后presentation error
最后是没有空格的!!
oh也就提交了十次吧(kk
#include<iostream>
#include<queue>
#include<algorithm>
#include<cstring>
#include<cstdio>
using namespace std;
#define maxn 500 * 200 + 10
typedef struct node
{
int son[128];
int fail;
int flag;
}tri;
tri trie[maxn];
char s[maxn];//要插入的字符串
int cnt=1;
queue<int> que;
int vis[maxn];
void insert(char* s,int num)
{
int u=1;
int n=strlen(s);
for(int i=0;i<n;i++)
{
int v=s[i];
if(!trie[u].son[v]) trie[u].son[v]=++cnt;
u=trie[u].son[v];
}
trie[u].flag=num;
}
void cal_fail()
{
for(int i=0;i<128;i++) trie[0].son[i]=1;
que.push(1);
int u;
while(que.size()!=0)
{
u=que.front();que.pop();
for(int i=0;i<128;i++)
{
int v=trie[u].son[i];
int Fail=trie[u].fail;
if(!trie[u].son[i]) {trie[u].son[i]=trie[Fail].son[i];continue;}
trie[v].fail=trie[Fail].son[i];
que.push(v);
}
}
}
int search(char*s)
{
int u=1;
int n=strlen(s);
int v,k;
int ans=0;
for(int i=0;i<n;i++)
{
v=s[i];
k=trie[u].son[v];
while(k>1)
{
if(trie[k].flag>0) {ans++; vis[trie[k].flag]++;}
k=trie[k].fail;
}
u=trie[u].son[v];
}
return ans;
}
int main()
{
int n,m;
cin>>n;
getchar();
for(int i=1;i<=n;i++)
{
gets(s);
insert(s,i);
}
cin>>m;
getchar();
int ans=0;
cal_fail();
for(int i=1;i<=m;i++)
{
memset(vis,0,sizeof(vis));
gets(s);
if(search(s)){ cout<<"web "<<i<<":";ans++;}
else continue;
for(int j=1;j<=n;j++)
{
if(vis[j]!=0)
{
cout<<" "<<j;
}
}
cout<<endl;
}
cout<<"total: "<<ans<<endl;
}