bzoj3998 弦论 后缀自动机

题目大意：

对于给定的一个长度为n的字符串S，求其第k小的子串。询问有两种，一种不重复（本质不同）的子串，另一种要计算重复（位置不同)的子串。

题解：

首先我们可以建一个后缀自动机。
然后每条路径走到每个点都是一个串，它们是有字典序的。
我们只需要统计出往每个点走之后都有多少串($sum[i]$)就好了。
$sum[i]=\sum sum[son[i]]+val[i]$
对于不计重复的情况下，$val[i]$=1
对于计算重复的情况下，$val[i]$就是该状态的right集合大小。
询问时dfs即可。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
using namespace std;

const int N=500005;
int T,n,k;
char s[N];
struct sam
{
    int last,cnt;
    int son[N<<1][26],fa[N<<1],mx[N<<1],val[N<<1],sum[N<<1];
    int v[N],q[N<<1];
    sam()
    {
        last=++cnt;
    }

    void extend(int c)
    {
        int p=last,np=last=++cnt;
        mx[np]=mx[p]+1;val[np]=1;
        while(p&&!son[p][c])son[p][c]=np,p=fa[p];
        if(!p)fa[np]=1;
        else 
        {
            int q=son[p][c];
            if(mx[q]==mx[p]+1)fa[np]=q;
            else 
            {
                int nq=++cnt;mx[nq]=mx[p]+1;
                memcpy(son[nq],son[q],sizeof(son[q]));
                fa[nq]=fa[q];
                fa[q]=fa[np]=nq;
                while(son[p][c]==q)son[p][c]=nq,p=fa[p];
            }
        }
    }

    void pre()
    {
        for(int i=1;i<=cnt;i++)v[mx[i]]++;
        for(int i=1;i<=n;i++)v[i]+=v[i-1];
        for(int i=cnt;i>=1;i--)q[v[mx[i]]--]=i;
        for(int i=cnt;i>=1;i--)
        {
            int t=q[i];
            if(T==1)val[fa[t]]+=val[t];
            else val[t]=1;
        }
        val[1]=0;
        for(int i=cnt;i>=1;i--)
        {
            int t=q[i];sum[t]=val[t];
            for(int j=0;j<26;j++)
                sum[t]+=sum[son[t][j]];
        }
    }

    void dfs(int x,int k)
    {
        if(k<=val[x])return;
        k-=val[x];
        for(int i=0;i<26;i++)
            if(int t=son[x][i])
            {
                if(k<=sum[t])
                {
                    putchar(i+'a');
                    dfs(t,k);
                    return;
                }
                else k-=sum[t];
            }
    }
}sam;

int main()
{
    //freopen("lx.in","r",stdin);
    scanf("%s",s+1);
    n=strlen(s+1);
    scanf("%d%d",&T,&k);
    for(int i=1;i<=n;i++)
        sam.extend(s[i]-'a');
    sam.pre();
    if(k>sam.sum[1])puts("-1");
    else sam.dfs(1,k);
    return 0;
}