HDU 5008 求第k小子串

本题要求第k小的distinct子串，可以根据height数组，二分出这个第k小子串所在后缀的位置信息。由于题目要求子串起始下标尽可能小。所以再在rank数组中，二分出与当前后缀LCP大于等于所求子串长度的范围。通过RMQ求出这个范围中最小的sa。

 #include <iostream>
 #include <vector>
 #include <algorithm>
 #include <string>
 #include <string.h>
 #include <stdio.h>
 #include <queue>
 #include <stack>
 #include <map>
 #include <set>
 #include <cmath>
 #include <ctime>
 #include <cassert>
 #include <sstream>
 using namespace std;

 const int N=;

 int MIN(int a,int b){return a<b?a:b;}
 int MAX(int a,int b){return a>b?a:b;}

 int val[N];
 struct RMQ {
     int dp[N][];
     int (*cmp) (int,int);
     void setMin(){cmp=MIN;}
     void setMax(){cmp=MAX;}
     void init(int n,int *val) {
         for (int i=; i<=n; i++)
             dp[i][]=val[i];
         for (int j=; (<<j)<=n; j++) {
             int k=<<(j-);
             for (int i=; i+k<=n; i++)
                 dp[i][j]=cmp(dp[i][j-],dp[i+k][j-]);
         }
     }
     int query(int a,int b) {
         if (a>b) swap(a,b);
         int dis=b-a+;
         int k=log((double)dis)/log(2.0);
         return cmp(dp[a][k],dp[b-(<<k)+][k]);
     }
 }rmq;
 char s[N];
 struct SuffixArray {;
     int sa[N];
     int t1[N],t2[N],c[N];
     int rk[N],height[N];
     long long sum[N];
     inline int cmp(int *r,int a,int b,int l){
         return r[a]==r[b]&&r[a+l]==r[b+l];
     }
     void calcSA (char *s,int n,int m) {
         int i,j,p,*x=t1,*y=t2;
         for(i=;i<m;i++)c[i]=;
         for(i=;i<n;i++)c[x[i]=s[i]]++;
         for(i=;i<m;i++)c[i]+=c[i-];
         for(i=n-;i>=;i--)sa[--c[x[i]]]=i;
         for(j=;j<=n;j<<=){
             p=;
             for(i=n-j;i<n;i++)y[p++]=i;
             for(i=;i<n;i++)if(sa[i]>=j)y[p++]=sa[i]-j; // 排名从小到大，如果pos比j大，则suffix(sa[i]-j)的第二关键字为p
             for(i=;i<m;i++)c[i]=;
             for(i=;i<n;i++)c[x[y[i]]]++;
             for(i=;i<m;i++)c[i]+=c[i-];
             for(i=n-;i>=;i--)sa[--c[x[y[i]]]]=y[i]; // 根据第二关键字从大到小，确定新一轮sa
             swap(x,y);
             p=;x[sa[]]=;
             for(i=;i<n;i++)
                 x[sa[i]]=cmp(y,sa[i-],sa[i],j)?p-:p++;
             if(p>=n)break;
             m=p;
         }
     }
     void calcHeight(char *s,int n) {
         int i,j,k=;
         for(i=;i<=n;i++)rk[sa[i]]=i;
         for(i=;i<n;i++){
             if(k)k--; // h[i]>=h[i-1]-1
             j=sa[rk[i]-]; // suffix(j)排名在suffix(i)前一位
             while(s[i+k]==s[j+k])k++; // 暴力计算lcp
             height[rk[i]]=k;
         }
         sum[]=;
         for (int i=;i<=n;i++) sum[i]=sum[i-]+n-sa[i]-height[i];
     }
     int lcp(int a,int b,int len) {
         if (a==b) return len-a;
         int ra=rk[a],rb=rk[b];
         if (ra>rb) swap(ra,rb);
         return queryST(ra+,rb);
     }
     int st[N][];
     void initST(int n) {
         for (int i=;i<=n;i++)
             st[i][]=height[i];
         for (int j=;(<<j)<=n;j++) {
             int k=<<(j-);
             for (int i=; i+k<=n; i++)
                 st[i][j]=min(st[i][j-],st[i+k][j-]);
         }
     }
     int queryST(int a,int b) {
         if (a>b) swap(a,b);
         int dis=b-a+;
         int k=log((double)dis)/log(2.0);
         return min(st[a][k],st[b-(<<k)+][k]);
     }
     void solve(int &l,int &r,long long k,int n) {
         if (k<||k>sum[n]) {
             l=;r=;
             return;
         }
         int t=lower_bound(sum,sum+n+,k)-sum;
         assert(t>=&&t<=n);
         long long now=sum[t-];
         int need=k-now;
         l=sa[t],r=sa[t]+height[t]+need-;
         int len=r-l+;
         int le=t,ri=n,ret=l;
         while (le<=ri) {
             int mid=(le+ri)/;
             if (lcp(sa[mid],l,n)>=len) {
                 le=mid+;
                 ret=mid;
             } else ri=mid-;
         }
         l=rmq.query(t,ret);
         l++;
         r=l+len-;
     }
 }suf;

 int main () {
     while (scanf("%s",s)!=EOF) {
         int n=strlen(s);
         suf.calcSA(s,n+,);
         suf.calcHeight(s,n);
         suf.initST(n);
         rmq.setMin();
         rmq.init(n,suf.sa);
         int Q;
         scanf("%d",&Q);
         int l=,r=; //int cnt=0;
         while (Q--) {
             long long k;
             scanf("%I64d",&k);
             k^=(l^r);
             k++;
             //k=++cnt;
             suf.solve(l,r,k,n);
             printf("%d %d\n",l,r);
         }
     }
     return ;
 }