luogu3796AC自动机模版

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;
}