代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cctype>
using namespace std;
const int MAXN=100010;
#define rg register int
char s[MAXN];
int c[MAXN],x[MAXN],y[MAXN],n,m,sa[MAXN],h[MAXN],rk[MAXN];
inline void suffix_sort()
{
	n=strlen(s+1);
	m='z';
	for(rg i=1;i<=n;i++)c[x[i]=s[i]]++;
	for(rg i=1;i<=m;i++)c[i]+=c[i-1];
	for(rg i=n;i;i--)sa[c[x[i]]--]=i;
	for(rg k=1;k<=n;k<<=1)
	{
		rg num=0;
		for(rg i=n-k+1;i<=n;i++)y[++num]=i;
		for(rg i=1;i<=n;i++)
			if(sa[i]>k)y[++num]=sa[i]-k;
		for(rg i=1;i<=m;i++)c[i]=0;
		for(rg i=1;i<=n;i++)c[x[i]]++;
		for(rg i=1;i<=m;i++)c[i]+=c[i-1];
		for(rg i=n;i;i--)sa[c[x[y[i]]]--]=y[i],y[i]=0;
		swap(x,y);
		x[sa[1]]=num=1;
		for(rg i=2;i<=n;i++)
			x[sa[i]]=(y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k])?num:++num;
		if(num==n)break ;
		m=num;
	}
	for(rg i=1;i<=n;i++)rk[sa[i]]=i;
	for(rg i=1,k=0;i<=n;i++)
	{
		if(k)k--;
		rg j=sa[rk[i]-1];
		while(j+k<=n&&i+k<=n&&s[j+k]==s[i+k])k++;
		h[rk[i]]=k;
	}
}
int main()
{
	scanf("%s",s+1);
	suffix_sort();
	for(rg i=1;i<=n;i++)
		printf("%d ",sa[i]);
	printf("\n");
	for(rg i=2;i<=n;i++)
		printf("%d ",h[i]);
}

评测结果