bzoj4310 跳蚤（SA+二分答案+贪心）

原题描述有误，应该是把原串分成最多k个子串，然后在每个子串中挑出字典序最大的子串，使得这些串中最大的字典序最小。

我们二分答案是第K小子串，用SA搞出这个串的位置，然后去贪心判定是否可行，即是否存在一种划分使得所有子串都小于等于这个串。

如何贪心判定呢？首先如果有一个s[i]>s[ansl]，则肯定怎么划分都不行。不然的话我们从后往前判断，如果当前串的大小超过了答案串，则必须砍一刀，我们贪心的砍在最前面即可。如果砍了超过K-1刀则gg

复杂度$O(nlogn)$

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 100010
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int K,n,m=26,sa[N],cnt[N],rk[N<<1],rk1[N<<1],tmp[N],h[N],a[N];
char s[N];int st[N][18],Log[N],ansl,ansr;
inline void getkth(ll k){
    for(int i=1;i<=n;++i){
        ll now=n-sa[i]+1-h[i];
        if(k>now){k-=now;continue;}
        ansl=sa[i];ansr=sa[i]+h[i]+k-1;return;
    }
}
inline int lcp(int x,int y){
    if(x==y) return n-x+1;x=rk[x];y=rk[y];
    if(x>y) swap(x,y);++x;int d=Log[y-x+1];
    return min(st[x][d],st[y-(1<<d)+1][d]);
}
inline bool Less(int x1,int y1,int x2,int y2){
    int l1=y1-x1+1,l2=y2-x2+1,l3=lcp(x1,x2);
    if(l3>=l1&&l2>l1) return 1;
    if(l3>=l2&&l1>=l2) return 0;
    return a[x1+l3]<a[x2+l3];
}
inline bool jud(){
    int lst=n,res=0;
    for(int i=n;i>=1;--i){
        if(a[i]>a[ansl]) return 0;
        if(Less(ansl,ansr,i,lst)) lst=i,++res;
        if(res>=K) return 0;
    }return 1;
}
inline void initrmq(){
    Log[0]=-1;
    for(int i=1;i<=n;++i) Log[i]=Log[i>>1]+1,st[i][0]=h[i];
    for(int i=1;i<=Log[n];++i)
        for(int j=1;j<=n;++j){
            if(j+(1<<i-1)>n) break;
            st[j][i]=min(st[j][i-1],st[j+(1<<i-1)][i-1]);
        }
}
int main(){
//  freopen("a.in","r",stdin);
    K=read();scanf("%s",s+1);n=strlen(s+1);
    for(int i=1;i<=n;++i) a[i]=s[i]-'a'+1;
    for(int i=1;i<=n;++i) cnt[a[i]]=1;
    for(int i=1;i<=m;++i) cnt[i]+=cnt[i-1];
    for(int i=n;i>=1;--i) rk[i]=cnt[a[i]];
    int k=0;
    for(int p=1;k!=n;m=k,p+=p){
        for(int i=1;i<=m;++i) cnt[i]=0;
        for(int i=1;i<=n;++i) cnt[rk[i+p]]++;
        for(int i=1;i<=m;++i) cnt[i]+=cnt[i-1];
        for(int i=n;i>=1;--i) tmp[cnt[rk[i+p]]--]=i;
        for(int i=1;i<=m;++i) cnt[i]=0;
        for(int i=1;i<=n;++i) cnt[rk[tmp[i]]]++;
        for(int i=1;i<=m;++i) cnt[i]+=cnt[i-1];
        for(int i=n;i>=1;--i) sa[cnt[rk[tmp[i]]]--]=tmp[i];
        memcpy(rk1,rk,sizeof(rk)>>1);rk[sa[1]]=k=1;
        for(int i=2;i<=n;++i){
            if(rk1[sa[i]]!=rk1[sa[i-1]]||rk1[sa[i]+p]!=rk1[sa[i-1]+p]) ++k;
            rk[sa[i]]=k;
        }
    }for(int i=1;i<=n;++i){
        if(rk[i]==1){h[1]=0;continue;}
        if(i==1||h[rk[i-1]]<=1) k=0;
        if(k) --k;
        while(i+k<=n&&a[i+k]==a[sa[rk[i]-1]+k]) ++k;
        h[rk[i]]=k;
    }initrmq();ll l=1,r=0;
    for(int i=1;i<=n;++i) r+=n-sa[i]+1-h[i];
    while(l<=r){
        ll mid=l+r>>1;getkth(mid);
        if(jud()) r=mid-1;else l=mid+1;
    }getkth(r+1);for(int i=ansl;i<=ansr;++i) putchar(s[i]);
    return 0;
}