bzoj4310: 跳蚤

二分+后缀数组；

#include<bits/stdc++.h>
#define rep(i,k,n) for(int i=k;i<(n);i++)
#define rep2(i,k,n)for(int i=k;i>=(n);i--)
using namespace std;
const int maxn=100105;
typedef long long ll;
int K,sa[maxn],he[maxn],p1[maxn],p2[maxn],c[maxn],rank[maxn],sig=500,n,rmq[maxn][18],pw[18],Log[maxn],top=0,id[maxn],len;
ll g[maxn];
char s[maxn];
void build_sa(){
    int* x=p1;int* y=p2;
    rep(i,0,n)c[x[i]=s[i]]++;
    rep(i,1,sig)c[i]+=c[i-1];
    rep2(i,n-1,0)sa[--c[x[i]]]=i;
    for(int h=1;h<=n;h<<1){int t=0;
        rep(i,n-h,n)y[t++]=i;rep(i,0,n)if(sa[i]>=h)y[t++]=sa[i]-h;
        rep(i,0,sig)c[i]=0;
        rep(i,0,n)c[x[y[i]]]++;
        rep(i,1,sig)c[i]+=c[i-1];
        rep2(i,n-1,0)sa[--c[x[y[i]]]]=y[i];
        swap(x,y);t=0;
        x[sa[0]]=0;
        rep(i,1,n)x[sa[i]]=y[sa[i]]==y[sa[i-1]] && y[sa[i]+h]==y[sa[i-1]+h] ? t : ++t;
        t++;if(t>=n)break;
        sig=t;
    }
    rep(i,0,n)rank[sa[i]]=i;
    int j=0;
    rep(i,0,n){if(!rank[i])continue;
    if(j)j--;
        int k=sa[rank[i]-1];
        while(s[i+j]==s[k+j])j++;
        he[rank[i]]=j;
    }
}
void init(){pw[0]=1;
    rep(i,1,17)pw[i]=pw[i-1]<<1;
    rep(i,2,maxn)Log[i]=Log[i>>1]+1;
    rep(i,0,n)rmq[i][0]=he[i];
    rep(i,1,17)rep(j,0,n)if(j+pw[i-1]<n)rmq[j][i]=min(rmq[j][i-1],rmq[j+pw[i-1]][i-1]); //if(j+pw[i-1]<n)
    ll ss=0;
    rep(i,0,n){ss+=n-sa[i]-he[i];g[++top]=ss;id[top]=i;}
}
int lcp(int x,int y){if(x==y)return n-x; //
x=rank[x],y=rank[y];
if(x>y)swap(x,y);
    int t=Log[y-x];
    return min(rmq[x+1][t],rmq[y-pw[t]+1][t]); //
}
int nowl,nowr,ansl,ansr;
void kth(ll k){int t=lower_bound(g+1,g+top+1,k)-g;
    nowl=sa[id[t]];nowr=nowl+he[id[t]]+k-g[t-1]-1;len=nowr-nowl+1;
}
bool pan(int l,int r){
    int t=min(lcp(l,nowl),min(len,r-l+1));
    if(t==r-l+1 && t<=len)return 1;
    if(t==len)return 0;
    return s[l+t]<=s[nowl+t];
}

bool ok(){int k=0;
    int i=n-1,j=n-1;
    while(i>=0){
        while(j>=0 && pan(j,i))j--;
        if(i==j)return 0;
        k++;i=j;
    }
    return (k<=K);
}
int main(){//freopen("in.in","r",stdin);
    scanf("%d",&K);
    scanf("%s",s);n=strlen(s);
    build_sa();
    init();
    ll l=0,r=g[top];  //
    while(l<=r){
        ll mid=(l+r)>>1;
        kth(mid);
        if(ok()){ansl=nowl,ansr=nowr,r=mid-1;}
        else l=mid+1;
    }
    rep(i,ansl,ansr+1)putchar(s[i]);
    return 0;
}