[bzoj3998]弦论

建立后缀自动机，对于一个节点，根据len排序后依次统计（t为0时每一个点只有1的贡献，t为1时每个点有|right|的贡献）子树内的子串数量，然后不断确定字符即可。

#include<bits/stdc++.h>
using namespace std;
#define N 500005
int V,last,k,sum,a[N<<1],sz[N<<1],id[N<<1],ans[N<<1],len[N<<1],fa[N<<1],ch[N<<1][31];
char s[N];
void add(int c){
    int p=last,np=last=++V;
    len[np]=len[p]+1;
    sz[np]=1;
    for(;(p)&&(!ch[p][c]);p=fa[p])ch[p][c]=V;
    if (!p)fa[np]=1;
    else{
        int q=ch[p][c];
        if (len[q]==len[p]+1)fa[np]=q;
        else{
            int nq=++V;
            len[nq]=len[p]+1;
            memcpy(ch[nq],ch[q],sizeof(ch[q]));
            fa[nq]=fa[q];
            fa[q]=fa[np]=nq;
            for(;(p)&&(ch[p][c]==q);p=fa[p])ch[p][c]=nq;
        }
    }
}
int main(){
    int t,k;
    scanf("%s%d%d",s,&t,&k);
    V=last=1;
    for(int i=0;s[i];i++)add(s[i]-'a');
    for(int i=1;i<=V;i++)a[len[i]]++;
    for(int i=1;s[i-1];i++)a[i]+=a[i-1];
    for(int i=V;i;i--)id[a[len[i]]--]=i;
    for(int i=V;i;i--)sz[fa[id[i]]]+=sz[id[i]];
    if (!t)
        for(int i=1;i<=V;i++)sz[i]=1;
    memcpy(a,sz,sizeof(a));
    sz[0]=a[0]=sz[1]=0;
    for(int i=V;i;i--)
        for(int j=0;j<26;j++)a[id[i]]+=a[ch[id[i]][j]];
    if (a[1]<k){
        printf("-1");
        return 0;
    }
    for(int i=1;k>0;k-=sz[i])
        for(int j=0;j<26;j++)
            if (k>a[ch[i][j]])k-=a[ch[i][j]];
            else{
                printf("%c",j+'a');
                i=ch[i][j];
                break;
            }
}