Luogu3796AC自动机模版
题目描述
有NN 个由小写字母组成的模式串以及一个文本串TT 。每个模式串可能会在文本串中出现多次。你需要找出哪些模式串在文本串TT 中出现的次数最多。
输入(样例)
2
aba
bab
ababababac
6
beta
alpha
haha
delta
dede
tata
dedeltalphahahahototatalpha
0
输出
4
aba
2
alpha
haha
Solution
AC自动机
一些调代码的经验:
数组大小要开够,memset的时候(尤其是AC自动机)要小心被卡成TLE
因为trie数组有很多没用到的地方,多组数据清空要注意
多组数据cnt要清零
每次init()只清空数组,不清cnt,本地调试根本调不出来
#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;
#define maxn 105010
int trie[maxn][30];
int val[maxn],fail[maxn];
int cnt[maxn];
char ss[1000010];
char str[200][200];
int cnnt=0;
void ins(char *s,int n,int x){
int now=0;
for(int i=1;i<=n;++i){
int c=s[i]-'a'+1;
if(!trie[now][c])trie[now][c]=++cnnt;
now=trie[now][c];
}
val[now]=x;
}
void Get_fail(){
queue<int> q;
for(int i=1;i<=26;++i){
if(trie[0][i]) q.push(trie[0][i]);
}
while(!q.empty()){
int x=q.front();
q.pop();
for(int i=1;i<=26;++i){
int v=trie[x][i];
if(v){
fail[v]=trie[fail[x]][i];
q.push(v);
}
else trie[x][i]=trie[fail[x]][i];
}
}
}
void query(char *s,int n,int m){
int u=0;
for(int i=1;i<=n;++i){
int v=u=trie[u][s[i]-'a'+1];
while(v){
if(val[v]) ++cnt[val[v]];
v=fail[v];
}
}
int maxx=0;
for(int i=1;i<=m;++i){
maxx=max(maxx,cnt[i]);
}
printf("%d\n",maxx);
for(int i=1;i<=m;++i){
if(cnt[i]==maxx) printf("%s\n",str[i]+1);
}
}
void init(){
memset(str,0,sizeof(str));
memset(fail,0,sizeof(fail));
memset(val,0,sizeof(val));
memset(cnt,0,sizeof(cnt));
memset(ss,0,sizeof(ss));
for(int i=0;i<=400000;++i){
for(int j=0;j<=28;++j) trie[i][j]=0;
}
cnnt=0;
}
int main(){
int n;
while(scanf("%d",&n)!=EOF&&n){
init();
for(int i=1;i<=n;++i){
scanf("%s",str[i]+1);
int len=strlen(str[i]+1);
ins(str[i],len,i);
}
Get_fail();
scanf("%s",ss+1);
int len=strlen(ss+1);
query(ss,len,n);
}
return 0;
}