bzoj 3998

我们分成两种情况来分析这个问题：t=0和t=1

t=1时，每一个子串出现的次数就是他在parent树上所在子树内前缀节点的个数，这一点我们已经说的很清楚了

利用SAM有向无环的性质，我们可以在parent树上统计完之后在后缀自动机上dfs，对每个点累计以他为开头的所有子串的总数

然后在查询的时候直接在SAM上跑，如果以当前点为开头的子串总数小于k，则将k减去这个总数后向他的兄弟节点查询，否则输出这个节点的字符，然后向他的子节点上查询，直到k小于当前点对应子串出现次数即说明输出的答案已满足要求

t=0时，由于只区分字典序，所以每一个节点都是一个子串，字典序相同的子串不会被重复统计（这利用了后缀自动机可以不重不漏地识别原串的每个子串的优秀性质）

剩余操作同上：在后缀自动机上dfs，然后再查询

代码：

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;
char ch[500005];
int T,k,n;
struct SAM
{
    int tranc[27];
    int endpos;
    int len;
    int pre;
}s[1000005];
struct Edge
{
    int next;
    int to;
}edge[1000005];
int head[1000005];
int val[1000005];
int sum[1000005];
int cnt=1;
int las,siz;
void init()
{
    memset(head,-1,sizeof(head));
    cnt=1;
}
void add(int l,int r)
{
    edge[cnt].next=head[l];
    edge[cnt].to=r;
    head[l]=cnt++;
}
void ins(int c)
{
    int nwp=++siz;
    s[nwp].len=s[las].len+1;
    s[nwp].endpos=1;
    int lsp;
    for(lsp=las;lsp&&!s[lsp].tranc[c];lsp=s[lsp].pre)s[lsp].tranc[c]=nwp;
    if(!lsp)
    {
        s[nwp].pre=1;
    }else
    {
        int lsq=s[lsp].tranc[c];
        if(s[lsq].len==s[lsp].len+1)
        {
            s[nwp].pre=lsq;
        }else
        {
            int nwq=++siz;
            s[nwq]=s[lsq];
            s[nwq].endpos=0;
            s[nwq].len=s[lsp].len+1;
            s[lsq].pre=s[nwp].pre=nwq;
            while(s[lsp].tranc[c]==lsq)
            {
                s[lsp].tranc[c]=nwq;
                lsp=s[lsp].pre;
            }
        }
    }
    las=nwp;
}
void buildtree()
{
    init();
    for(int i=2;i<=siz;i++)add(s[i].pre,i);
}
void dfs(int x)
{
    if(T)val[x]=s[x].endpos;
    else val[x]=1;
    for(int i=head[x];i!=-1;i=edge[i].next)
    {
        int to=edge[i].to;
        dfs(to);
        if(T)val[x]+=val[to];
    }
}
void redfs(int x)
{
    sum[x]=val[x];
    for(int i=1;i<=26;i++)
    {
        int to=s[x].tranc[i];
        if(!to)continue;
        if(!sum[to])redfs(to);
        sum[x]+=sum[to];
    }
}
void qdfs(int x)
{
    if(k<=val[x])return;
    k-=val[x];
    for(int i=1;i<=26;i++)
    {
        int to=s[x].tranc[i];
        if(!to)continue;
        if(k<=sum[to])
        {
            printf("%c",i+'a'-1);
            qdfs(to);
            return;
        }else k-=sum[to];
    }
}
int main()
{   
//  freopen("data.in","r",stdin);
    scanf("%s",ch+1);
    scanf("%d%d",&T,&k);
    n=strlen(ch+1);
    las=++siz;
    for(int i=1;i<=n;i++)ins(ch[i]-'a'+1);
    buildtree();
    dfs(1);
    val[1]=0;
    redfs(1);
    if(k>sum[1])printf("-1");
    else qdfs(1);
    printf("\n");
    return 0;
}