题意:给出字符串,求其中字典序第k小的子串。
我们设Ans[s]为状态s之后一共有多少子串,这个在建完图之后倒叙更新一遍就可以了。
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<string>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<ctime>
using namespace std;
int n,T,K;
char s[1000010];
struct node{
int tot,son[1000010][26],sum[1000010],pre[1000010],val[1000010];
int last,step[1000010],num[1000010],v[1000010],Ans[1000010];
node () { last=++tot; }
void insert(int x) {
int now=last,np=last=++tot;step[np]=step[now]+1;val[np]=1;
while(son[now][x]==0&&now!=0) son[now][x]=np,now=pre[now];
if(now==0) pre[np]=1;
else
{
int q=son[now][x];
if(step[q]==step[now]+1) pre[np]=q;
else
{
int newq=++tot;
step[newq]=step[now]+1;
for(int i=0;i<26;i++) son[newq][i]=son[q][i];
pre[newq]=pre[q];
pre[q]=pre[np]=newq;
while(son[now][x]==q) son[now][x]=newq,now=pre[now];
}
}
}
void calc() {
for(int i=1;i<=tot;i++) num[step[i]]++;
for(int i=1;i<=n;i++) num[i]+=num[i-1];
for(int i=tot;i>=1;i--) v[num[step[i]]--]=i;
for(int i=tot;i>=1;i--)
{
if(T==1) val[pre[v[i]]]+=val[v[i]];
else val[v[i]]=1;
}
val[1]=0;
for(int i=tot;i>=1;i--)
{
Ans[v[i]]=val[v[i]];
for(int j=0;j<26;j++)
Ans[v[i]]+=Ans[son[v[i]][j]];
}
}
void dfs(int x,int G) {
if(val[x]>=G) return ;
G-=val[x];
for(int i=0;i<26;i++)
{
if(son[x][i]!=0&&G<=Ans[son[x][i]])
{
printf("%c",'a'+i);
dfs(son[x][i],G);
return ;
}
G-=Ans[son[x][i]];
}
}
};node SAM;
int main()
{
scanf("%s",s+1);
n=strlen(s+1);
scanf("%d%d",&T,&K);
for(int i=1;i<=n;i++) SAM.insert(s[i]-'a');
SAM.calc();
if(K>SAM.Ans[1]) printf("-1");
else SAM.dfs(1,K);
return 0;
}