hdu 5008 Boring String Problem(后缀数组+rmq)

题目链接：hdu 5008 Boring String Problem

题意：

给你一个字符串，有q个询问，每次询问该字符串所有的子串中字典序第k小的是哪个串，输出位置，如果有多个位置，输出最靠左的那个。

题解：

后缀数组求完height后，求一下字典序第i的后缀有多少个不同的子串，然后求一下前缀和。

然后就可以每对于一个k，可以二分求到排序后第一个字典序为k的后缀，然后用rmq求一下最小就行了。

#include<bits/stdc++.h>
#define mst(a,b) memset(a,b,sizeof(a))
#define F(i,a,b) for(int i=(a);i<=(b);++i)
#define ___ freopen("c:\code\in.txt","r",stdin);
inline int RT(int l,int r){return l+r|l!=r;}
using namespace std;
typedef long long ll;
typedef pair<int,int>P;

namespace suffixarray{    
    #define FN(n) for(int i=0;i<n;i++)
    const int N =1E5+7;
    int rnk[N],sa[N],height[N],c[N];char s[N];
    void getsa(int n,int m,int *x=rnk,int *y=height){
        FN(m)c[i]=0;FN(n)c[x[i]=s[i]]++;FN(m)c[i+1]+=c[i];
        for(int i=n-1;i>=0;i--)sa[--c[x[i]]]=i;
        for(int k=1,p;p=0,k<=n;k=p>=n?N:k<<1,m=p){
            for(int i=n-k;i<n;i++)y[p++]=i;
            FN(n)if(sa[i]>=k)y[p++]=sa[i]-k;
            FN(m)c[i]=0;FN(n)c[x[y[i]]]++;FN(m)c[i+1]+=c[i];
            for(int i=n-1;i>=0;i--)sa[--c[x[y[i]]]]=y[i];
            swap(x,y),p=1,x[sa[0]]=0;
            for(int i=1;i<n;i++)
            x[sa[i]]=y[sa[i-1]]==y[sa[i]]&&y[sa[i-1]+k]==y[sa[i]+k]?p-1:p++;
        }
        FN(n)rnk[sa[i]]=i;
        for(int i=0,j,k=0;i<n-1;height[rnk[i++]]=k)
        for(k=k?k-1:k,j=sa[rnk[i]-1];s[i+k]==s[j+k];k++);
    }
}
using namespace suffixarray;

int n,q;
ll k,sum[N],L,R;

int a[N][18],b[N][18];
void rmq(int *a,int f[][18])
{
    for(int i=1;i<=n;i++)f[i][0]=a[i];
    for(int j=1;1<<j<n;j++)for(int i=1;i<=n;i++)
    if(i+(1<<j)-1<= n)f[i][j]=min(f[i][j-1],f[i+(1<<j-1)][j-1]);
    else break;
}
inline int find(int l,int r,int f[][18])
{
    int k=31-__builtin_clz(r-l+1);
    return min(f[l][k],f[r-(1<<k)+1][k]);
}

void getkth(ll k)
{
    if(k>sum[n]){L=R=0;return;}
    int l=0,r=n,ans,mid;
    while(l<=r)
    {
        mid=l+r>>1;
        if(sum[mid]>=k)ans=mid,r=mid-1;
        else l=mid+1;
    }
    k=k-sum[ans-1]+height[ans];
    int pos=ans;
    l=2,r=n-pos+1,ans=1;
    while(l<=r)
    {
        mid=l+r>>1;
        if(find(pos+1,pos+mid-1,a)>=k)l=mid+1,ans=mid;
        else r=mid-1;
    }
    L=find(pos,pos+ans-1,b)+1,R=L+k-1;
}

int main(){
    while(~scanf("%s",s))
    {
        L=R=0,getsa((n=strlen(s))+1,150);
        F(i,1,n)sum[i]=n-sa[i]-height[i]+sum[i-1];
        rmq(height,a),rmq(sa,b);
        scanf("%d",&q);
        while(q--)
        {
            scanf("%lld",&k);
            k=(k^L^R)+1,getkth(k);
            printf("%lld %lld
",L,R);
        }
    }
    return 0;
}