P3796 【模板】AC自动机（加强版）

P3796 【模板】AC自动机（加强版）

https://www.luogu.org/problemnew/show/P3796

题目描述

有NN个由小写字母组成的模式串以及一个文本串TT。每个模式串可能会在文本串中出现多次。你需要找出哪些模式串在文本串TT中出现的次数最多。

输入输出格式

输入格式：

输入含多组数据。

每组数据的第一行为一个正整数NN，表示共有NN个模式串，1 leq N leq 1501≤N≤150。

接下去NN行，每行一个长度小于等于7070的模式串。下一行是一个长度小于等于10^6106的文本串TT。

输入结束标志为N=0N=0。

输出格式：

对于每组数据，第一行输出模式串最多出现的次数，接下去若干行每行输出一个出现次数最多的模式串，按输入顺序排列。

输入输出样例

输入样例#1： 

2
aba
bab
ababababac
6
beta
alpha
haha
delta
dede
tata
dedeltalphahahahototatalpha
0

输出样例#1： 

4
aba
2
alpha
haha

#include<bits/stdc++.h>
using namespace std;
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define sqr(x) ((x)*(x))
#define maxn 1000005
typedef long long ll;
typedef unsigned long long ull;
const ull MOD=257;
/*#ifndef ONLINE_JUDGE
        freopen("1.txt","r",stdin);
#endif */

struct tree{
    int fail;
    int vis[26];
    int num;
}ac[1000005];

int cnt=0;

struct ANS{
    int num,pos;
    bool operator<(const ANS &b){
        if(num==b.num) return pos<b.pos;
        return num>b.num;
    }
}ans[155];

void Clear(int x){
    memset(ac[x].vis,0,sizeof(ac[0].vis));
    ac[x].num=ac[x].fail=0;
}

void build(string s,int flag){///建trie树
    int len=s.length();
    int now=0;
    for(int i=0;i<len;i++){
        if(ac[now].vis[s[i]-'a']==0){
            ac[now].vis[s[i]-'a']=++cnt;
            Clear(cnt);
        }
        now=ac[now].vis[s[i]-'a'];
    }
    ac[now].num=flag;
}

void get_fail(){///构建成trie图
    queue<int>Q;
    ac[0].fail=0;
    for(int i=0;i<26;i++){
        if(ac[0].vis[i]){
            ac[ac[0].vis[i]].fail=0;
            Q.push(ac[0].vis[i]);
        }
    }
    while(!Q.empty()){
        int u=Q.front();
        Q.pop();
        for(int i=0;i<26;i++){
            if(ac[u].vis[i]){
                ac[ac[u].vis[i]].fail=ac[ac[u].fail].vis[i];
                Q.push(ac[u].vis[i]);
            }
            else{
                ac[u].vis[i]=ac[ac[u].fail].vis[i];///如果当前结点不存在，就指向父亲结点的fail指向的结点的子结点
            }
        }
    }
}

void ac_query(string s){
    int len=s.length();
    int now=0;
    for(int i=0;i<len;i++){
        now=ac[now].vis[s[i]-'a'];
        for(int t=now;t&&ac[t].num!=-1;t=ac[t].fail){
            ans[ac[t].num].num++;
        }
    }
}

string s[155];

int main(){
    #ifndef ONLINE_JUDGE
        freopen("1.txt","r",stdin);
    #endif
    std::ios::sync_with_stdio(false);
    int n;
    while(cin>>n){
        if(!n) break;
        Clear(0);
        for(int i=1;i<=n;i++){
            cin>>s[i];
            build(s[i],i);
            ans[i].num=0;
            ans[i].pos=i;
        }
        get_fail();
        cin>>s[0];
        ac_query(s[0]);
        sort(ans+1,ans+n+1);
        cout<<ans[1].num<<endl;
        cout<<s[ans[1].pos]<<endl;
        for(int i=2;i<=n;i++){
            if(ans[i].num==ans[i-1].num){
                cout<<s[ans[i].pos]<<endl;
            }
            else{
                break;
            }
        }
    }
}