有位仁兄和我错的一模一样但是他过了（看不到代码QAQ）

代码见下

#include<bits/stdc++.h>
using namespace std;
inline int rd(){
	int f=1,j=0;
	char w=getchar();
	while(w>'9'||w<'0'){
		if(w=='-')f=-1;
		w=getchar();
	}
	while(w>='0'&&w<='9'){
		j=(j<<3)+(j<<1)+w-'0';
		w=getchar();
	}
	return f*j;
}
const int N=500001;
int cnt[N],id[N*2],sc[N*2],sa[N],hei[N];
int n,p[N],st[N][19],fro[N];
long long bac[N];
char s[N];
void getfa(){
	int m=122;
	for(int i=1;i<=n;i++)cnt[id[i]=s[i]]++;
	for(int i=1;i<=m;i++)cnt[i]+=cnt[i-1];
	for(int i=1;i<=n;i++)sa[cnt[id[i]]--]=i;
	for(int k=1;k<=n;k=(k<<1)){
		int j=0;
		for(int i=n-k+1;i<=n;i++)sc[++j]=i;
		for(int i=1;i<=n;i++)if(sa[i]>k)sc[++j]=sa[i]-k;
		for(int i=1;i<=m;i++)cnt[i]=0;
		for(int i=1;i<=n;i++)cnt[id[i]]++;
		for(int i=1;i<=m;i++)cnt[i]+=cnt[i-1];
		for(int i=n;i>=1;i--)sa[cnt[id[sc[i]]]--]=sc[i],sc[i]=0;
		swap(id,sc);
		id[sa[1]]=1,j=1;
		for(int i=1;i<=n;i++)id[sa[i]]=(sc[sa[i]]==sc[sa[i-1]]&&sa[i]+k<=n&&sa[i-1]+k<=n&&sc[sa[i]+k]==sc[sa[i-1]+k])?j:++j;
		if(j>=n)break;
		m=j;
	}
	return ;
}
void gethei(){
	for(int i=1;i<=n;i++)p[sa[i]]=i;
	int k=0;
	for(int i=1;i<=n;i++){
		if(p[i]==1)continue;
		if(k)k--;
		int j=sa[p[i]-1];
		while(i+k<=n&&j+k<=n&&s[j+k]==s[i+k])k++;
		hei[p[i]]=k;
	}
	return ;
}
int getbac(int pla,int t){
	pla++;
	for(int i=18;i>=0;i--)if(st[pla][i]>=t)pla+=(1<<i);
	return pla;
}
signed main(){
	scanf("%s",s+1);
	n=strlen(s+1);
	getfa(),gethei();
	for(int i=1;i<=n;i++)st[i][0]=hei[i];
	for(int j=1;(1<<j)<=n;j++)for(int i=1;i+(1<<j)-1<=n;i++)st[i][j]=min(st[i][j-1],st[i+(1<<(j-1))][j-1]);
	int t=rd(),k=rd(),pla=0,ans;
	long long nown=0,anns=0;
	for(int i=1;i<=n;i++){
		for(int j=nown+1;j<=hei[i];j++)fro[j]=i-1;
		for(int j=nown;j>hei[i];j--)bac[fro[j]]+=i-fro[j];
		if(i<n)bac[i]=n-sa[i]+1-max(hei[i],hei[i+1]);
	}
	for(int j=nown;j>0;j--)bac[fro[j]]+=n-fro[j]+1;
	for(int i=1;i<=n;i++){
		anns+=(t==0)?n-sa[i]+1-hei[i]:bac[i];
		if(anns>=k){
			anns-=(t==0)?n-sa[i]+1-hei[i]:bac[i];
			pla=i;
			break;
		}
	}
	if(!pla){
		printf("-1");
		return 0;
	}
	if(t==0)ans=k-anns+hei[pla];
	else{
		for(int i=hei[pla]+1;i<=n-sa[pla]+1;i++){
			if(hei[pla+1]<i){
				anns++;
				if(anns==k){ans=i;break;}
			}
			else{
				int rr=getbac(pla,i);
				if(anns+rr-pla>=k){
					ans=i;
					break;
				}
				anns+=rr-pla;
			}
		}
	}
	for(int i=0;i<ans;i++)putchar(s[sa[pla]+i]);//printf("%c",s[sa[pla]+i]);
	return 0;
}