终于找到教程了,现在来写一发
题目:Jzoj4072(bzoj3998)弦论 第一问
其实非常好写,连空间都不需要多开一倍,只需要sa数组rank数组和height数组就好了,dfs一次可以求出
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<algorithm>
#define N 1000010
using namespace std;
char str[N]; long long k,c[N];
int s[N][26],f[N],sz[N],l[N],mx[N];
int n,m,cnt=1,lst=1,tot=0,sa[N],r[N],h[N];
inline int extend(char c,int pos){
int p=lst,np=lst=++cnt,q,nq; c-='a';
mx[np]=mx[p]+1; sz[np]=1; l[np]=pos;
for(;p&&!s[p][c];p=f[p]) s[p][c]=np;
if(!p) return f[np]=1;
q=s[p][c];
if(mx[q]==mx[p]+1) f[np]=q;
else{
nq=++cnt;
mx[nq]=mx[p]+1; l[nq]=l[q];
f[nq]=f[q]; f[q]=f[np]=nq;
memcpy(s[nq],s[q],26<<2);
for(;p&&s[p][c]==q;p=f[p]) s[p][c]=nq;
}
}
void dfs(int x){
if(sz[x]) sa[++tot]=l[x],r[l[x]]=tot;
for(int i=0;i<26;++i) if(s[x][i]) dfs(s[x][i]);
}
int main(){
scanf("%s%d%lld",str+1,&m,&k);
for(int i=n=strlen(str+1);i;--i) extend(str[i],i);
memset(s,0,sizeof s);
for(int i=2;i<=cnt;++i) s[f[i]][str[l[i]+mx[f[i]]]-'a']=i;
dfs(1);
for(int i=1,j,k=0;i<=n;++i)
if(r[i]>1){
for(k?--k:0,j=sa[r[i]-1];str[i+k]==str[j+k];++k);
h[r[i]]=k;
}
if(m) for(int i=1;i<=n;++i) c[i]=c[i-1]+n-sa[i]+1;
else for(int i=1;i<=n;++i) c[i]=c[i-1]+n-sa[i]+1-h[i];
if(k>c[n]) puts("-1"); else {
int j=lower_bound(c+1,c+1+n,k)-c;
while(c[j]>k) --j; str[k-c[j]+sa[j+1]+h[j+1]]=0;
printf("%s",str+sa[j+1]);
}
}