后缀数组

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=1e6+10;
char s[maxn];
int n,p,sz,sa[maxn],tmp[maxn],bin[maxn],rk[maxn],h[maxn];
void SA()
{
	sz=max(n,150);
	for(int i=0;i<n;i++) rk[i]=s[i];
	for(int i=0;i<sz;i++) bin[i]=0;
	for(int i=0;i<n;i++) ++bin[rk[i]];
	for(int i=1;i<sz;i++) bin[i]+=bin[i-1];
	for(int i=n-1;i>=0;i--) sa[--bin[rk[i]]]=i;
	for(int j=1;j<=n;j*=2){
		p=0;
		for(int i=n-j;i<n;i++) tmp[p++]=i;
		for(int i=0;i<n;i++) if(sa[i]-j>=0) tmp[p++]=sa[i]-j;
		for(int i=0;i<sz;i++) bin[i]=0;
		for(int i=0;i<n;i++) ++bin[rk[i]];
		for(int i=1;i<sz;i++) bin[i]+=bin[i-1];
		for(int i=n-1;i>=0;i--) sa[--bin[rk[tmp[i]]]]=tmp[i];
		p=0;tmp[sa[0]]=0;
		for(int i=1;i<n;i++){
			int v0=sa[i-1],v1=sa[i],v01,v00;
			if(v0+j<n) v00=rk[v0+j]; else v00=-1;
			if(v1+j<n) v01=rk[v1+j]; else v01=-1;
			if(rk[v0]==rk[v1]&&v00==v01) tmp[sa[i]]=p;
			else tmp[sa[i]]=++p;
		}
		for(int i=0;i<n;i++) rk[i]=tmp[i];
	}
}
int calc(int x,int y,int res)
{
	for(;x+res<n&&y+res<n&&s[x+res]==s[y+res];) res++;
	return res;
}
void calc_height()
{
	if(rk[0]==0) h[0]=0; else h[0]=calc(0,sa[rk[0]-1],0);
	for(int i=1;i<n;i++){
		if(rk[i]==0) h[i]=0;
		else h[i]=calc(i,sa[rk[i]-1],max(0,h[i-1]-1));
	}
}
int main()
{
	scanf("%s",s);
	n=strlen(s);	
	SA();
	for(int i=0;i<n;i++) printf("%d ",sa[i]+1);
	printf("\n");
	calc_height();
	for(int i=1;i<n;i++) printf("%d ",h[sa[i]]);
	return 0;
}