3998: [TJOI2015]弦论 后缀自动机

Description

对于一个给定长度为N的字符串，求它的第K小子串是什么。

题解：

这道题感觉真的很入门啊……建出后缀自动机，对于$T=0$，默认所有节点的$right$集合为1，对于$T=1$，就直接正常求出$right$集合，然后对每个点求个$Sum[x]$，表示$x$节点开始有多少个子串，然后类似主席树找第$k$小那样，dfs一次就行了。

代码：

#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define pa pair<int,int>
const int Maxn=1000010;
const int inf=2147483647;
int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
    return x*f;
}
int last=1,tot=1;
int son[Maxn][26],par[Maxn],mx[Maxn];
int sum[Maxn],to[Maxn],Right[Maxn],Sum[Maxn];
void extend(int x)
{
    int p=last,np=++tot;mx[np]=mx[p]+1;Right[np]=1;
    while(p&&!son[p][x])son[p][x]=np,p=par[p];
    if(!p)par[np]=1;
    else
    {
        int q=son[p][x];
        if(mx[p]+1==mx[q])par[np]=q;
        else
        {
            int nq=++tot;mx[nq]=mx[p]+1;
            for(int i=0;i<26;i++)son[nq][i]=son[q][i];
            par[nq]=par[q];
            par[q]=par[np]=nq;
            while(son[p][x]==q)son[p][x]=nq,p=par[p];
        }
    }
    last=np;
}
char s[Maxn];
int T,K,len;
void dfs(int now,int k)
{
    if(k<=Right[now])return;
    k-=Right[now];
    for(int i=0;i<26;i++)
    {
        int x=son[now][i];
        if(!x)continue;
        if(k<=Sum[x])
        {
            printf("%c",i+'a'),dfs(x,k);
            break;
        }
        else k-=Sum[x];
    }
}
int main()
{
    scanf("%s",s);
    T=read(),K=read();
    len=strlen(s);for(int i=0;i<len;i++)extend(s[i]-'a');
    for(int i=1;i<=tot;i++)sum[mx[i]]++;
    for(int i=1;i<=len;i++)sum[i]+=sum[i-1];
    for(int i=1;i<=tot;i++)to[sum[mx[i]]--]=i;
    if(!T){for(int i=1;i<=tot;i++)Right[i]=1;}
    else
    {
        for(int i=tot;i;i--)
        {
            int x=to[i];
            Right[par[x]]+=Right[x];
        }
    }
    Right[1]=0;
    for(int i=tot;i;i--)
    {
        int x=to[i];Sum[x]=Right[x];
        for(int j=0;j<26;j++)
        if(son[x][j])Sum[x]+=Sum[son[x][j]];
    }
    if(K>Sum[1])return puts("-1"),0;
    dfs(1,K);
}