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=;
 /*#ifndef ONLINE_JUDGE
         freopen("1.txt","r",stdin);
 #endif */

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

 int cnt=;

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

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

 void build(string s,int flag){///建trie树
     int len=s.length();
     int now=;
     for(int i=;i<len;i++){
         if(ac[now].vis[s[i]-'a']==){
             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[].fail=;
     for(int i=;i<;i++){
         if(ac[].vis[i]){
             ac[ac[].vis[i]].fail=;
             Q.push(ac[].vis[i]);
         }
     }
     while(!Q.empty()){
         int u=Q.front();
         Q.pop();
         for(int i=;i<;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=;
     for(int i=;i<len;i++){
         now=ac[now].vis[s[i]-'a'];
         for(int t=now;t&&ac[t].num!=-;t=ac[t].fail){
             ans[ac[t].num].num++;
         }
     }
 }

 string s[];

 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();
         for(int i=;i<=n;i++){
             cin>>s[i];
             build(s[i],i);
             ans[i].num=;
             ans[i].pos=i;
         }
         get_fail();
         cin>>s[];
         ac_query(s[]);
         sort(ans+,ans+n+);
         cout<<ans[].num<<endl;
         cout<<s[ans[].pos]<<endl;
         for(int i=;i<=n;i++){
             if(ans[i].num==ans[i-].num){
                 cout<<s[ans[i].pos]<<endl;
             }
             else{
                 break;
             }
         }
     }
 }