SPOJ 7258 SUBLEX 后缀数组 + 二分答案 + 前缀和

Code:

#include <cstdio>
#include <algorithm>
#include <cstring>
#define setIO(s) freopen(s".in","r",stdin)
#define maxn 1000000
#define ll long long
using namespace std;
char str[maxn];
int arr[maxn],pos[maxn],x[maxn],rk[maxn],sa[maxn],c[maxn],height[maxn];
ll C[maxn],brr[maxn];
int n,m;
struct SA{
    void qsort(){
        for(int i=0;i<=m;++i) c[i]=0;
        for(int i=1;i<=n;++i) ++c[rk[pos[i]]];
        for(int i=1;i<=m;++i) c[i]+=c[i-1];
        for(int i=n;i>=1;--i) sa[c[rk[pos[i]]]--]=pos[i];
    }
    void build(){
        for(int i=1;i<=n;++i) rk[i]=arr[i],pos[i]=i;
        qsort();
        for(int k=1;k<=n;k<<=1){
            int num=0;
            for(int i=n-k+1;i<=n;++i) pos[++num]=i;
            for(int i=1;i<=n;++i) if(sa[i]>k) pos[++num]=sa[i]-k;
            qsort();
            swap(rk,pos);
            rk[sa[1]]=1,num=1;
            for(int i=2;i<=n;++i)
                rk[sa[i]]=(pos[sa[i]]==pos[sa[i-1]]&&pos[sa[i]+k]==pos[sa[i-1]+k])?num:++num;
            if(num==n) break;
            m=num;
        }
    }
    void get_height(){
        int k=1;
        for(int i=1;i<=n;++i) rk[sa[i]]=i;
        for(int i=1;i<=n;++i){
            if(k) --k;
            int j=sa[rk[i]-1];
            while(arr[i+k]==arr[j+k]) ++k;
            height[rk[i]]=k;
        }
    }
}T;
int main(){
    //setIO("input");
    scanf("%s",str),n=strlen(str),m=128;
    for(int i=1;i<=n;++i) arr[i]=str[i-1];
    T.build(),T.get_height();
    for(int i=1;i<=n;++i) brr[i]=n-sa[i]+1-height[i];
    for(int i=1;i<=n;++i) C[i]=C[i-1]+brr[i];
    int q; long long opt;  scanf("%d",&q);
    while(q--){
        scanf("%lld",&opt);
        int l=1,r=n,mid,ans=0;
        while(l<=r) {
            mid=(l+r)>>1;
            if(C[mid]>=opt) ans=mid,r=mid-1;
            else l=mid+1;
        }
        for(int i=sa[ans];i<=n-(C[ans]-opt);++i) printf("%c",str[i-1]);
        printf("\n");
    }
    return 0;
}