这题可以参考平衡树求第k大的过程,需要预处理一下从当前节点往下走能走出多少个子串。
原本准备存个图用反向的topsort,发现极为麻烦,看了别人的代码后发现,他们按step大小用了基排,省了很多麻烦。
仔细看了一下构造过程,貌似确实有这样的性质,后创建且step大的节点在fail树上处于儿子的地位,在原图上处于被转移点的地位。
如果性质成立,那么这里也就说得通了。
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<string>
#include<cstring>
#include<cmath>
#include<ctime>
#include<algorithm>
#include<iomanip>
#include<set>
#include<map>
#include<queue>
using namespace std;
#define mem1(i,j) memset(i,j,sizeof(i))
#define mem2(i,j) memcpy(i,j,sizeof(i))
#define LL long long
#define up(i,j,n) for(int i=(j);i<=(n);i++)
#define FILE "dealing"
#define poi vec
#define eps 1e-10
#define db double
const int maxn=500010,inf=1000000000,mod=1000000007;
int read(){
int x=0,f=1,ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch<='9'&&ch>='0'){x=(x<<1)+(x<<3)+ch-'0',ch=getchar();}
return f*x;
}
bool cmax(int& a,int b){return a<b?a=b,true:false;}
bool cmin(int& a,int b){return a>b?a=b,true:false;}
char s[maxn];int f[maxn];
namespace SAM{
int cnt=1,now=1;
int pre[maxn],c[maxn][26],len[maxn],ct[maxn],sa[maxn];
void extend(int x){
int np,p,q,nq;
p=now;np=++cnt;len[np]=len[p]+1;now=np;
while(p&&!c[p][x])c[p][x]=np,p=pre[p];
if(!p)pre[np]=1;
else {
q=c[p][x];
if(len[q]==len[p]+1)pre[np]=q;
else {
len[nq=++cnt]=len[p]+1;
mem2(c[nq],c[q]);
pre[nq]=pre[q];
pre[q]=pre[np]=nq;
while(p&&c[p][x]==q)c[p][x]=nq,p=pre[p];
}
}
}
void getsort(){
up(i,1,cnt)ct[len[i]]++;
up(i,1,cnt)ct[i]+=ct[i-1];
for(int i=cnt;i>=1;i--)sa[ct[len[i]]--]=i;
for(int i=cnt;i>=1;i--){
f[sa[i]]=1;
up(j,0,25)f[sa[i]]+=f[c[sa[i]][j]];
}
}
void build(char* s){
cnt=1,now=1;int n=strlen(s+1);
up(i,1,n)extend(s[i]-'a');
}
};
int main(){
freopen(FILE".in","r",stdin);
freopen(FILE".out","w",stdout);
scanf("%s",s+1);
SAM::build(s);
SAM::getsort();
int Q=read();
while(Q--){
int k=read();
int now=1,len=0;
while(k)up(i,0,25){
if(SAM::c[now][i]){
if(f[SAM::c[now][i]]>=k){
putchar('a'+i);
now=SAM::c[now][i];
k--;//不可省
break;
}
else k-=f[SAM::c[now][i]];
}
}
printf("\n");
}
return 0;
}