bzoj3998 TJOI2015 弦论

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#define LL long long
using namespace std;
//by zfy =.=
const int N=1000005;
int p,last=1,np,cnt=1,q,nq,l
,val
,sum
,tr
[26],fa
,T,k,len,wc
,sa
;
char s[N/2];
void insert(int x){
p=last;np=last=++cnt;l[np]=l[p]+1;val[np]=1;
for(;p&&!tr[p][x];p=fa[p]) tr[p][x]=np;
if(!p) fa[np]=1;
else {
q=tr[p][x];
if(l[q]==l[p]+1) fa[np]=q;
else {
nq=++cnt;l[nq]=l[p]+1;
memcpy(tr[nq],tr[q],sizeof tr[q]);
fa[nq]=fa[q];fa[q]=fa[np]=nq;
for(;tr[p][x]==q;p=fa[p]) tr[p][x]=nq;
}
}
}
void dfs(int x,int k){
if(k<=val[x]) return;k-=val[x];
for(int i=0;i<26;i++){
if(k<=sum[tr[x][i]]){
printf("%c",i+'a');dfs(tr[x][i],k);return;
}else k-=sum[tr[x][i]];
}
}
int main(){
scanf("%s",s+1);scanf("%d%d",&T,&k);len=strlen(s+1);
for(int i=1;i<=len;i++) insert(s[i]-'a');
for(int i=1;i<=cnt;i++) wc[l[i]]++;
for(int i=1;i<=len;i++) wc[i]+=wc[i-1];
for(int i=1;i<=cnt;i++) sa[wc[l[i]]--]=i;
for(int i=cnt;i;i--){
if(T==0) val[sa[i]]=1;
else {
val[fa[sa[i]]]+=val[sa[i]];
}
sum[sa[i]]=val[sa[i]];
for(int j=0;j<26;j++) sum[sa[i]]+=sum[tr[sa[i]][j]];
}sum[1]-=val[1];val[1]=0;
if(k>sum[1]) puts("-1");
else dfs(1,k);
return 0;
}