这是blog主sb时,打上的后缀平衡树版本的code
#include<cstdio>
#include<cstring>
#define N 200010
#define LL long long
int n,m,len,x,y,t,k;
int ch[N][2],rd[N],cl[N],cnt,root,rank[N],sa[N],height[N];
LL rk[N];
int s[N];
inline int randlmy(){
static int seed=142857;
return seed=int(seed*48271LL%2147483647);
}
void rebuild(int &rt,LL l,LL r){
if(!rt)return;
rk[rt]=l+r;
LL mid=(l+r)>>1;
rebuild(ch[rt][0],l,mid);
rebuild(ch[rt][1],mid,r);
}
void rotate(int &p,int d,LL l,LL r){
int q=ch[p][d^1];
ch[p][d^1]=ch[q][d];
ch[q][d]=p;
p=q;
rebuild(p,l,r);
}
void insert(int &rt,LL l,LL r){
LL mid=(l+r)>>1;
if(!rt){rt=cnt;rk[rt]=l+r;return;}
if(cl[cnt]<cl[rt]||(cl[cnt]==cl[rt]&&rk[cnt-1]<rk[rt-1])){insert(ch[rt][0],l,mid);if(rd[ch[rt][0]]>rd[rt])rotate(rt,1,l,r);}
else{insert(ch[rt][1],mid,r);if(rd[ch[rt][1]]>rd[rt])rotate(rt,0,l,r);}
}
void insert(int v){
cl[++cnt]=v;
rd[cnt]=randlmy()*randlmy();
insert(root,1,1LL<<61);
}
void SA(int rt){
if(!rt)return;
SA(ch[rt][0]);
sa[t]=len-rt;
rank[len-rt]=t++;
SA(ch[rt][1]);
}
void make_height(){
for(register int i=0;i<len;++i){
if(!rank[i]) continue;
int j=sa[rank[i]-1];
if(k)--k;
while(s[i+k]==s[j+k])++k;
height[rank[i]]=k;
}
}
int main(){
char ch=getchar();while(ch>'z'||ch<'a')ch=getchar();
while(ch>='a'&&ch<='z')s[len++]=ch,ch=getchar();
for(register int i=1;i<=len;++i)
insert(s[len-i]-'a');
SA(root);
make_height();
for(register int i=0;i<len;++i)
printf("%d ",sa[i]+1);printf("\n");
for(register int i=1;i<len;i++)
printf("%d ",height[i]);
return 0;
}