代码

//include <bits/stdc++.h>
#include <cstdio>
#include <algorithm>
#include <cstring> 
#define rep(i,x,y) for (int i=x;i<=y;++i)
using namespace std;
const int N  =  2e5+233;
//const int mod = ;
struct Suffix_Array
{
	static const int M=300;
	int rk[N],tp[N],cnt[M],h[N],sa[N],n,m,p,i,j,k;
	int * x=rk,*y=tp;
	char str[N];
	inline void Init()
	{
		scanf("%s",str+1);
		n=strlen(str+1);
	}
	void Sort()
	{	
		//memset(cnt,0,sizeof(cnt)) ;
		rep(i,0,m) cnt[i]=0;
		rep(i,1,n) ++cnt[x[y[i]]];
		rep(i,1,m) cnt[i]+=cnt[i-1];
		for (int i=n;i>=1;--i)sa[ cnt[x[y[i]]]--] =y[i];
	}
	inline int cmp(int *f,int x,int y,int w){return f[x]==f[y]&&f[x+w]==f[y+w];}
	inline void SA()
	{
		rep(i,1,n) x[i]=str[i],y[i]=i;
		m=256;Sort();
		for (int w=1;w<=n;w<<=1,m=p)
		{
			for (p=0,i=n-w+1;i<=n;++i) y[++p]=i;
			rep(i,1,n) if (sa[i]>w) y[++p]=sa[i]-w;
			Sort(),swap(x,y),x[sa[1]]=p=1;
			for (i=2;i<=n;++i) x[sa[i]]=cmp(y,sa[i],sa[i-1],w) ?p :++p;
		}
		rep(i,1,n) rk[sa[i]]=i;
		for (k=0,i=1;i<=n;h[rk[i++]]=k)
		  for (k? --k :0 ,j=sa[x[i]-1];str[j+k]==str[i+k];++k);
	}
}S;
signed main()
{
	S.Init();
	S.SA()	;
	rep(i,1,S.n) printf("%d ",S.sa[i]);puts("");
	rep(i,2,S.n) printf("%d ",S.h[i]) ;
	return 0;	
}

评测结果