AC自动机模板 2

题目：AC自动机（加强版）

资料：yyb大佬的博客

注意：这个题为了防止tle，需要加上优化，即 if(!ch[u][i]) ch[u][i]=ch[fail[u]][i]

代码：#include<bits/stdc++.h>
using namespace std;

#define maxn 150
#define maxd 70
#define maxm maxn*maxd
#define maxa 1000000
#define s 26

int n;

struct AC {
int ch[maxm+5][s+5],m;
int val[maxm+5];
char a[maxa+5];
int fail[maxm+5];
char lst[maxn+5][maxd+5];
int mp[maxn+5];
int sum[maxm+5];

void clr() {
memset(ch,0,sizeof(ch));
memset(val,0,sizeof(val));
memset(fail,0,sizeof(fail));
memset(mp,0,sizeof(mp));
memset(sum,0,sizeof(sum));
m=0;
}

void insert(char* x,int len,int j) {
int u=0;
for(int i=0; i<len; i++) {
int y=x[i]-'a'+1;
if(!ch[u][y]) {
ch[u][y]=++m;
}
u=ch[u][y];
}
val[u]++;
mp[j]=u;
}

void make_fail() {
queue<int> que;
for(int i=1; i<=s; i++) {
if(ch[0][i]) que.push(ch[0][i]);
}
while(!que.empty()) {
int u=que.front();
que.pop();
for(int i=1; i<=s; i++) {
if(ch[u][i]) {
que.push(ch[u][i]);
fail[ch[u][i]]=ch[fail[u]][i];
} else {
ch[u][i]=ch[fail[u]][i];
}
}
}
}

void find() {
int len=strlen(a);
int j=0;
for(int i=0; i<len; i++) {
int x=a[i]-'a'+1;
j=ch[j][x];
for(int k=j; k; k=fail[k]) sum[k]++;
}
}

void readin() {
for(int i=1; i<=n; i++) {
scanf("%s",lst[i]);
int y=strlen(lst[i]);
insert(lst[i],y,i);
}
make_fail();
scanf("%s",a);
}

void print() {
int ans=0;
for(int i=1; i<=m; i++) {
if(val[i]) ans=max(ans,sum[i]);
}
printf("%d\n",ans);
for(int i=1; i<=n; i++) {
if(sum[mp[i]]==ans) {
printf("%s\n",lst[i]);
}
}
}
};

AC ac;

int main() {
while(~scanf("%d",&n)&&n) {
ac.clr();
ac.readin();
ac.find();
ac.print();
}

return 0;
}