bzoj 3998(后缀自动机)

3998: [TJOI2015]弦论

Time Limit: 10 Sec   Memory Limit: 256 MB
Submit: 1511   Solved: 517
[ Submit][ Status][ Discuss]

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来确定当前点下有几个答案，然后dfs来查找（类似二叉搜索树的查找）。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
char c[500005];
int last,cnt,n,T,K;
int sl[1000005],ch[1000005][28],fa[1000005],val[1000005],sum[1000005];
int v[500005],q[1000005];


void add(int s)
 {
  int p=last; int np=last=++cnt; sl[np]=sl[p]+1; val[np]=1;
  while (p && !ch[p][s]) ch[p][s]=np,p=fa[p];
if (!p) fa[np]=1;else
{
int q=ch[p][s];
if (sl[q]==sl[p]+1) fa[np]=q;else
{
++cnt; int nq=cnt; sl[nq]=sl[p]+1;
   fa[nq]=fa[q]; 
   memcpy(ch[nq],ch[q],sizeof(ch[q]));
   fa[np]=nq; fa[q]=nq;
   while (ch[p][s]==q)ch[p][s]=nq,p=fa[p];
}
 } 
 }


void pre()
 {
  for (int i=1;i<=cnt;++i) v[sl[i]]++;
  for (int i=1;i<=n;++i) v[i]+=v[i-1];
  for (int i=cnt;i>=1;--i) q[v[sl[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=1;j<=26;++j)
sum[t]+=sum[ch[t][j]];
}
 }


void dfs(int now,int su)
 {
  if (val[now]>=su) 
  {
  return; 
 }
su-=val[now];
for (int i=1;i<=26;++i)
if (ch[now][i])
{
int t=ch[now][i];
if (sum[t]>=su)
{
printf("%c",i+'a'-1);
dfs(t,su);
return;
 }
su-=sum[t];
}
 }




int main()
{
last=cnt=1; 
scanf("%s",c);
n=strlen(c);
for (int i=1;i<=n;++i)
{
add(int(c[i-1])-96);
}
scanf("%d %d",&T,&K);
pre();
    if (K>sum[1]) printf("-1");else
     {
      dfs(1,K);
}
}