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

正解：后缀自动机。

我们可以使用后缀自动机来求出每一个后缀，那么可以知道，一个字符串的出现次数就是它的$right$集合大小，也就是它在后缀自动机的$parent$树上的子树大小。

那么如果$T=1$，就把一个点的$sz$统计成它的$parent$子树，否则它的$sz$就是$1$，然后我们贪心地找出第$k$小子串就行了。

#include <bits/stdc++.h>
#define il inline
#define RG register
#define ll long long
#define N (1000010)

using namespace std;

int ch[N][26],l[N],fa[N],sz[N],sum[N],c[N],id[N],n,t,k,la,tot,len;
char s[N];

il void add(RG int c){
  RG int p=la,np=++tot; la=np,l[np]=l[p]+1,sz[np]=1;
  for (;p && !ch[p][c];p=fa[p]) ch[p][c]=np;
  if (!p){ fa[np]=1; return; } RG int q=ch[p][c];
  if (l[q]==l[p]+1) fa[np]=q; else{
    RG int nq=++tot; l[nq]=l[p]+1;
    fa[nq]=fa[q],fa[q]=fa[np]=nq;
    memcpy(ch[nq],ch[q],sizeof(ch[q]));
    for (;ch[p][c]==q;p=fa[p]) ch[p][c]=nq;
  }
  return;
}

il void dfs(RG int x,RG int k){
  if (k<=sz[x]) return; RG int c=sz[x];
  for (RG int i=0;i<26;++i)
    if (k<=c+sum[ch[x][i]]){
      putchar('a'+i);
      return dfs(ch[x][i],k-c);
    } else c+=sum[ch[x][i]];
  return;
}

int main(){
#ifndef ONLINE_JUDGE
  freopen("kth.in","r",stdin);
  freopen("kth.out","w",stdout);
#endif
  scanf("%s",s+1),len=strlen(s+1),cin>>t>>k,tot=la=1;
  for (RG int i=1;i<=len;++i) add(s[i]-'a');
  for (RG int i=1;i<=tot;++i) ++c[l[i]];
  for (RG int i=1;i<=tot;++i) c[i]+=c[i-1];
  for (RG int i=1;i<=tot;++i) id[c[l[i]]--]=i;
  if (t) for (RG int i=tot;i;--i) sz[fa[id[i]]]+=sz[id[i]];
  else for (RG int i=tot;i;--i) sz[id[i]]=1; sz[1]=0;
  for (RG int i=tot;i;--i){
    sum[id[i]]=sz[id[i]];
    for (RG int j=0;j<26;++j)
      sum[id[i]]+=sum[ch[id[i]][j]];
  }
  if (sum[1]<k) puts("-1"); else dfs(1,k); return 0;
}