bzoj 3998: [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

建出后缀自动机，然后在后缀自动机上跑一下就可以。
维护后面有几种情况
切记虚点处置为0。。

#include<cstdio>
#include<string>
#include<cstring>
#include<iostream>
using namespace std;
int T;
int lx;
struct sam
{
int son[1000005][26],pre[1000005],step[1000005];
int v[1000005],q[1000005];
int s[1000005],sum[1000005];
int last,tot;
sam()
{
last=1;
tot=1;
}
inline void push_back(int v)
{
tot++;
step[tot]=v;
}
inline void extend(int x)
{
push_back(step[last]+1);
int p=last,np=tot;s[np]=1;
while(son[p][x]==0&&p!=0)
{
son[p][x]=np;
p=pre[p];
}
if(p==0)
pre[np]=1;
else
{
int q=son[p][x];
if(step[q]!=step[p]+1)
{
push_back(step[p]+1);
//tot++;
int nq=tot;
int i;
for(i=0;i<=25;i++)
son[nq][i]=son[q][i];
pre[nq]=pre[q];
pre[q]=pre[np]=nq;
while(son[p][x]==q)
{
son[p][x]=nq;
p=pre[p];
}
}
else
pre[np]=q;
}
last=np;
}
inline void build()
{
string x;
cin>>x;
lx=x.size();
int i;
for(i=0;i<=lx-1;i++)
extend(x[i]-'a');
}
inline void prep()
{
int i,j;
for(i=1;i<=tot;i++)
v[step[i]]++;
for(i=1;i<=tot;i++)
v[i]+=v[i-1];
for(i=tot;i>=1;i--)
q[v[step[i]]--]=i;
// for(i=1;i<=tot;i++)
// s[i]=1;
for(i=tot;i>=1;i--)
{
if(T==1)
s[pre[q[i]]]+=s[q[i]];
else
s[q[i]]=1;
}
s[1]=0;
for(i=tot;i>=1;i--)
{
sum[q[i]]+=s[q[i]];
for(j=0;j<=25;j++)
sum[q[i]]+=sum[son[q[i]][j]];
}
}
inline void dfs(int d,int k)
{
if(k<=s[d])
return ;
k-=s[d];
int i;
for(i=0;i<=25;i++)
{
int t=son[d][i];
if(t!=0)
{
if(k<=sum[t])
{
printf("%c",i+'a');
dfs(t,k);
return ;
}
k-=sum[t];
}
}
}
}sam;
int main()
{
//freopen("data.in","r",stdin);
//freopen("data.out","w",stdout);
sam.build();
int k;
scanf("%d%d",&T,&k);
sam.prep();
if(k>sam.sum[1])
printf("-1\n");
else
{
sam.dfs(1,k);
printf("\n");
}
return 0;
}