【USACO06DEC】—牛奶模式Milk Patterns(后缀自动机)

传送门

后缀自动机板子题了吧

只需要求ParentTreeParent-Tree上非叶子节点的最长的lenlen就可以了

因为字符集太大又不想离散化

就直接用mapmap来存nxtnxt就可以了

#include<bits/stdc++.h>
using namespace std;
inline int read(){
    char ch=getchar();
    int res=0;
    while(!isdigit(ch))ch=getchar();
    while(isdigit(ch))res=(res<<3)+(res<<1)+(ch^48),ch=getchar();
    return res;
}
const int N=40005;
map<int,int>nxt[N];
int tot,last,k,n,len[N],link[N],siz[N],A[N],p[N];
inline void sa_extend(int c){
    int cur=++tot,p=last;last=cur;
    len[cur]=len[p]+1,siz[cur]=1;
    for(;p&&!nxt[p][c];p=link[p])nxt[p][c]=cur;
    if(!p)link[cur]=1;
    else{
        int q=nxt[p][c];
        if(len[q]==len[p]+1)link[cur]=q;
        else{
            int clo=++tot;
            nxt[clo]=nxt[q],link[clo]=link[q],len[clo]=len[p]+1;
            for(;p&&nxt[p][c]==q;p=link[p])nxt[p][c]=clo;
            link[q]=link[cur]=clo;
        }
    }
}
inline bool cmp(int x,int y){
    return len[x]>len[y];
}
int main(){
    n=read(),k=read();last=tot=1;
    for(int i=1;i<=n;++i){
        int c=read();sa_extend(c);
    }
    for(int i=1;i<=tot;++i)++A[len[i]];
    for(int i=1;i<=tot;++i)A[i]+=A[i-1];
    for(int i=1;i<=tot;++i)p[A[len[i]]--]=i;
    for(int i=tot;i;--i)siz[link[p[i]]]+=siz[p[i]];
    int ans=0;
    for(int i=tot;i;i--){
        if(siz[i]>=k)ans=max(ans,len[i]);
    }
    cout<<ans;
}

原文地址:https://www.cnblogs.com/stargazer-cyk/p/10366372.html