AC自动机(加强版)

#include<bits/stdc++.h>
using namespace std;
const int N=2e4+10;
int n,tot,tr[N][30],fail[N],end[N];
string s[200];char m[1000010];
struct node{int pos,num;}ans[N];
bool cmp(node a,node b){
    return (a.num>b.num||(a.num==b.num&&a.pos<b.pos));
}
void insert(int k){
    int len=s[k].size(),now=0;
    for(int i=0;i<len;++i){
        int ch=s[k][i]-'a';
        if(!tr[now][ch]) tr[now][ch]=++tot;
        now=tr[now][ch];
    }end[now]=k;
}
void build(){
    queue<int>q;
    for(int i=0;i<26;++i){
        if(tr[0][i]) q.push(tr[0][i]),fail[tr[0][i]]=0;
    }
    while(q.size()){
        int x=q.front();q.pop();
        for(int i=0;i<26;++i){
            if(tr[x][i]){
                int v=tr[x][i];
                fail[v]=tr[fail[x]][i];
                q.push(v);
            }else tr[x][i]=tr[fail[x]][i];
        }
    }
}
void query(){
    int len=strlen(m+1),now=0;
    for(int i=1;i<=len;++i){
        int ch=m[i]-'a';
        now=tr[now][ch];
        for(int t=now;t;t=fail[t])
        ans[end[t]].num++;
    }
}
int main(){
    while(scanf("%d",&n)){
        if(n==0) break;
        memset(ans,0,sizeof(ans));
        memset(tr,0,sizeof(tr));
        memset(end,0,sizeof(end));
        memset(fail,0,sizeof(fail));
        tot=0;
        for(int i=1;i<=n;++i) cin>>s[i],insert(i),ans[i].pos=i;
        build();
        scanf("%s",m+1);
        query();
        sort(ans+1,ans+n+1,cmp);
        cout<<ans[1].num<<endl;
        for(int i=1;i<=n;++i)
            if(ans[i].num==ans[1].num) cout<<s[ans[i].pos]<<endl;
            else break;
    }
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/kgxw0430/p/10252626.html