【LGP5108】仰望半月的夜空

题目

我还会写\(SA\)和 \(ST\)表真是令人感动

发现这是一个思博题

我们开一个指针，标记一下当前合法的字典序最小的后缀排名在哪里，刚开始自然是\(1\)

我们发现这个后缀不能为我们提供\(i\)的长度我们就右移这个指针

之后我们二分+\(St\)表找到从这个后缀往右扩展的最大距离，查一下这里面最小的\(sa\)就好了

代码

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define re register
#define LL long long
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
const int maxn=300005;
inline int read() {
    char c=getchar();int x=0;while(c<'0'||c>'9') c=getchar();
    while(c>='0'&&c<='9') x=(x<<3)+(x<<1)+c-48,c=getchar();return x;
}
char S[maxn];
int sa[maxn],het[maxn],rk[maxn],tp[maxn],tax[maxn];
int c[maxn],a[maxn];
int n,opt,m,sz;
int log_2[maxn],St[maxn][20],v[maxn][20];
inline int find(int x) {
    int l=1,r=sz;
    while(l<=r) {
        int mid=l+r>>1;
        if(c[mid]==x) return mid;
        if(c[mid]<x) l=mid+1;else r=mid-1;
    }
    return 0;
}
inline void qsort() {
    for(re int i=0;i<=m;i++) tax[i]=0;
    for(re int i=1;i<=n;i++) tax[rk[i]]++;
    for(re int i=1;i<=m;i++) tax[i]+=tax[i-1];
    for(re int i=n;i;i--) sa[tax[rk[tp[i]]]--]=tp[i];
}
inline int query(int l,int r) {
    int k=log_2[r-l+1];
    return min(St[l][k],St[r-(1<<k)+1][k]);
}
inline int ask(int l,int r) {
    int k=log_2[r-l+1]; 
    return min(v[l][k],v[r-(1<<k)+1][k]);
}
int main() {
    opt=read(),n=read();
    if(opt==26) {scanf("%s",S+1);for(re int i=1;i<=n;i++) a[i]=S[i]-'a';}
    else for(re int i=1;i<=n;i++) a[i]=read();
    for(re int i=1;i<=n;i++) c[i]=a[i];
    std::sort(c+1,c+n+1);sz=std::unique(c+1,c+n+1)-c-1;
    for(re int i=1;i<=n;i++) a[i]=find(a[i]);
    for(re int i=1;i<=n;i++) rk[i]=a[i],tp[i]=i;
    m=sz;qsort();
    for(re int w=1,p=0;p<n;m=p,w<<=1) {
        p=0;
        for(re int i=1;i<=w;i++) tp[++p]=n-w+i;
        for(re int i=1;i<=n;i++) if(sa[i]>w) tp[++p]=sa[i]-w;
        qsort();
        for(re int i=1;i<=n;i++) std::swap(rk[i],tp[i]);
        rk[sa[1]]=p=1;
        for(re int i=2;i<=n;i++)
            rk[sa[i]]=(tp[sa[i]]==tp[sa[i-1]]&&tp[sa[i]+w]==tp[sa[i-1]+w])?p:++p;
    }
    int k=0;
    for(re int i=1;i<=n;i++) {
        if(k) --k;
        int j=sa[rk[i]-1];
        while(a[i+k]==a[j+k]) ++k;
        het[rk[i]]=k;
    }
    for(re int i=1;i<=n;i++) St[i][0]=het[i];
    for(re int i=2;i<=n;i++) log_2[i]=log_2[i>>1]+1;
    for(re int j=1;j<=log_2[n];j++) 
        for(re int i=1;i+(1<<j)-1<=n;i++)
            St[i][j]=min(St[i][j-1],St[i+(1<<(j-1))][j-1]);
    for(re int i=1;i<=n;i++) v[i][0]=sa[i];
    for(re int j=1;j<=log_2[n];j++)
        for(re int i=1;i+(1<<j)-1<=n;i++)
            v[i][j]=min(v[i][j-1],v[i+(1<<(j-1))][j-1]);
    int now=1;
    for(re int i=1;i<=n;i++) {
        while(sa[now]+i-1>n) now++;
        int l=2,r=n-now+1;
        int ans=1;
        while(l<=r) {
            int mid=l+r>>1;
            if(query(now+1,now+mid-1)>=i) ans=mid,l=mid+1;
                else r=mid-1;
        }
        printf("%d ",ask(now,now+ans-1));
    }
    return 0;
}