【bzoj3998】[TJOI2015]弦论 后缀自动机

Description

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

Input

第一行是一个仅由小写英文字母构成的字符串S

第二行为两个整数T和K，T为0则表示不同位置的相同子串算作一个。T=1则表示不同位置的相同子串算作多个。K的意义如题所述。
Output

输出仅一行，为一个数字串，为第K小的子串。如果子串数目不足K个，则输出-1

Sample Input

aabc
0 3
Sample Output

aab
HINT

N<=5*10^5

T<2
K<=10^9

题解
T=0 right[i]=1
T=0 right[fa[i]]+=right[i]表示这个子串出现几次
处理出每个节点之后能有多少种子串，dfs计算即可。

代码

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
#define ll long long
using namespace std;
inline 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*10+ch-'0';ch=getchar();}
    return x*f;
}
int T,n,k;
char s[500005];
int last,cnt,S;
int trans[1000005][26],fa[1000005],l[1000005],Right[1000005],sum[1000005];
int v[500005],q[1000005];
void add(int x)
{
    int p=last,np=++cnt,c=s[x]-'a';last=cnt;l[np]=x;Right[np]=1;
    for (;p&&!trans[p][c];p=fa[p]) trans[p][c]=np;
    if (!p) fa[np]=S;
    else
    {
        int q=trans[p][c];
        if (l[p]+1==l[q]) fa[np]=q;
        else
        {
            int nq=++cnt;l[nq]=l[p]+1;
            memcpy(trans[nq],trans[q],sizeof trans[q]);
            fa[nq]=fa[q];fa[q]=fa[np]=nq;
            for (;trans[p][c]==q;p=fa[p]) trans[p][c]=nq;
        }
    }
}
void pre()
{
    for (int i=1;i<=cnt;i++) v[l[i]]++;
    for (int i=1;i<=n;i++) v[i]+=v[i-1];
    for (int i=cnt;i;i--) q[v[l[i]]--]=i;
    for (int i=cnt;i;i--)
    {
        int t=q[i];
        if (T==1) Right[fa[t]]+=Right[t];
        else Right[t]=1;
    }
    Right[1]=0;
    for (int i=cnt;i;i--)
    {
        int t=q[i];sum[t]=Right[t];
        for (int j=0;j<26;j++)
            sum[t]+=sum[trans[t][j]];
    }
}
void dfs(int x,int k)
{
    if (k<=Right[x]) return;
    k-=Right[x];
    for (int i=0;i<26;i++)if (int t=trans[x][i])
    {
        if (k<=sum[t])
        {
            putchar('a'+i);
            dfs(t,k);
            return;
        }
        k-=sum[t];
    }
}
int main()
{
    scanf("%s",s+1);T=read();k=read();
    last=S=++cnt;n=strlen(s+1);
    for (int i=1;i<=n;i++) add(i);
    pre();if (k>sum[1])puts("-1");else dfs(S,k);
    return 0;
}