hdu5030 Rabbit's String 後綴數組

把一個字符串最多分爲k段，所有每一段的最大子串的最大值最小。

二分答案串是排名多少的字串，根據sum可以定位該串的位置，在t之後的後綴中有些串是需要切割的，這些串加上後一個字符會大於答案串，把這些串的位置求出後貪心最少切多少刀。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
#include<vector>
#include<set>
#include<map>
#include<queue>
#include<bitset>
#define fi first
#define se second
#define pii pair<int,int>
#define ll long long
#define inf 1<<30
#define eps 1e-8
using namespace std;
const int maxn=100005;
int n,k;
char str[maxn];
int s[maxn];
ll sum[maxn];
int cc[maxn];
int t1[maxn],t2[maxn],c[maxn];
int sa[maxn],ra[maxn],height[maxn];
void buildSa(int s[],int n,int m)//最大值小於m
{
    int i,j,p,*x=t1,*y=t2;
    for(i=0;i<m;i++)c[i]=0;
    for(i=0;i<n;i++)c[x[i]=s[i]]++;
    for(i=1;i<m;i++)c[i]+=c[i-1];
    for(i=n-1;i>=0;i--)sa[--c[x[i]]]=i;
    for(j=1;j<=n;j<<=1)
    {
        p=0;
        for(i=n-j;i<n;i++)y[p++]=i;
        for(i=0;i<n;i++)if(sa[i]>=j)y[p++]=sa[i]-j;
        for(i=0;i<m;i++)c[i]=0;
        for(i=0;i<n;i++)c[x[y[i]]]++;
        for(i=1;i<m;i++)c[i]+=c[i-1];
        for(i=n-1;i>=0;i--)sa[--c[x[y[i]]]]=y[i];
        swap(x,y);
        p=1;x[sa[0]]=0;
        for(i=1;i<n;i++)
            x[sa[i]]=y[sa[i-1]]==y[sa[i]] && y[sa[i-1]+j]==y[sa[i]+j]?p-1:p++;
        if(p>=n)break;
        m=p;
    }
}
void getHeight(int s[],int n)
{
    int i,j,k=0;
    for(i=0;i<=n;i++)ra[sa[i]]=i;
    for(i=0;i<n;i++)
    {
        if(k)k--;
        j=sa[ra[i]-1];
        while(s[i+k]==s[j+k])k++;
        height[ra[i]]=k;
    }
}
bool ok(ll mid)
{
    int t=lower_bound(sum,sum+n+1,mid)-sum;
    int l=sa[t],r=n-sum[t]+mid-1;
    int len=r-l+1;
    for(int i=0;i<=n;i++)
        cc[i]=-1;
    if(l+len-1<n-1)
        cc[l]=len;
    int ans=0;
    for(int i=t+1;i<=n;i++) {
        len=min(len,height[i]);
        if(len==0) return false;
        if(sa[i]+len-1<n-1)
            cc[sa[i]]=len;
    }
    int m=n;
    for(int i=0;i<n;i++) {
        if(i==m) {
            ans++;
            if(ans>=k) return false;
            m=n;
        }
        if(cc[i]!=-1) {
            m=min(m,i+cc[i]);
        }
    }
    return ans+1<=k;
}
void solve()
{
    sum[0]=0;
    for(int i=1;i<=n;i++)
        sum[i]=sum[i-1]+n-sa[i]-height[i];
    ll l=1,r=sum[n];
    while(l<r) {
        ll mid=(l+r)/2;
        if(ok(mid))
            r=mid;
        else
            l=mid+1;
    }
    int t=lower_bound(sum,sum+n+1,l)-sum;
    int le=sa[t];
    int rr=n-sum[t]+l-1;
    for(int i=le;i<=rr;i++)
        putchar(str[i]);
    puts("");
}
int main()
{
    while(~scanf("%d",&k) && k) {
        scanf("%s",str);
        n=strlen(str);
        for(int i=0;i<=n;i++)
            s[i]=str[i];
        buildSa(s,n+1,128);
        getHeight(s,n);
        solve();
    }
    return 0;
}