SPOJ-SUBLEX Lexicographical Substring Search（字典序第k小子串，后缀自动机+预处理）

题意：

给定一个字符串S，求S本质不同排名第k小的子串

数据范围：|S|<=9e4

解法：

对S建立后缀自动机，然后预处理每条路径后面有多少种本质不同的子串，
然后就像权值线段树找第k大那样做就行了。

code：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxm=2e6+5;
char s[maxm];
int n;
struct SAM{
    int ch[maxm][26];
    int fa[maxm],l[maxm];//l[]是等价类的最长字符串长度len
    int last=1,tot=1;//tot是节点数量
    void add(int c){
        int p=last,np=++tot;last=np;l[np]=l[p]+1;
        for(;p&&!ch[p][c];p=fa[p])ch[p][c]=np;
        if(!p)fa[np]=1;
        else{
            int q=ch[p][c];
            if(l[p]+1==l[q])fa[np]=q;
            else{
                int nq=++tot;l[nq]=l[p]+1;
                memcpy(ch[nq],ch[q],sizeof ch[q]);
                fa[nq]=fa[q];fa[q]=fa[np]=nq;
                for(;ch[p][c]==q;p=fa[p])ch[p][c]=nq;
            }
        }
    }
    //本题
    int cnt[maxm];
    int idx[maxm];
    int sz[maxm];
    void cal_sz(){
        for(int i=1;i<=tot;i++)sz[i]=1;
        //基数排序
        for(int i=1;i<=tot;i++)cnt[l[i]]++;
        for(int i=1;i<=tot;i++)cnt[i]+=cnt[i-1];
        for(int i=1;i<=tot;i++)idx[cnt[l[i]]--]=i;
        //
        for(int i=tot;i>=1;i--){
            for(int j=0;j<26;j++){
                sz[idx[i]]+=sz[ch[idx[i]][j]];
            }
        }
    }
    void solve(int k){
        int x=1;
        while(k){
            for(int j=0;j<26;j++){
                if(ch[x][j]){
                    if(sz[ch[x][j]]>=k){
                        printf("%c",'a'+j);
                        x=ch[x][j];
                        k--;
                        break;
                    }else{
                        k-=sz[ch[x][j]];
                    }
                }
            }
        }
        puts("");
    }
    //
}S;
signed main(){
    //
    scanf("%s",s+1);
    n=strlen(s+1);
    for(int i=1;i<=n;i++){
        S.add(s[i]-'a');
    }
    S.cal_sz();
    //
    int q;scanf("%d",&q);
    while(q--){
        int k;scanf("%d",&k);
        S.solve(k);
    }
    return 0;
}

---