后缀数组

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#define N 1000010
using namespace std;
char s[N];
int len;
//--------begin--------
struct SA {
	int rnk[N],sa[N],tme[N],sec[N],height[N],n,sigma;
	void Sort() {
		for(int i=0; i<=sigma; i++)tme[i]=0;
		for(int i=1; i<=n; i++)tme[rnk[i]]++;
		for(int i=1; i<=sigma; i++)tme[i]+=tme[i-1];
		for(int i=n; i>=1; i--)sa[tme[rnk[sec[i]]]--]=sec[i];
		memset(sec,0,sizeof(sec));
	}
	void GetSa(char *s) {
		for(int i=1; i<=n; i++)rnk[i]=s[i],sec[i]=i;
		Sort();
		for(int w=1,p=0; w<=n&&p<n; w<<=1,sigma=p) {
			p=0;
			for(int i=n-w+1; i<=n; i++)sec[++p]=i;
			for(int i=1; i<=n; i++)if(sa[i]>w)sec[++p]=sa[i]-w;
			Sort();
			swap(sec,rnk);
			rnk[sa[1]]=p=1;
			for(int i=2; i<=n; i++) {
				rnk[sa[i]]=(sec[sa[i-1]]==sec[sa[i]]&&sec[sa[i-1]+w]==sec[sa[i]+w])? p:++p;
			}
		}
	}
	void GetHeight(char *s) {
		int k=0;
		for(int i=1; i<=n; i++) {
			if(k)k--;
			int j=sa[rnk[i]-1];
			while(s[i+k]==s[j+k])k++;
			height[rnk[i]]=k;
		}
	}
	int init(char *s) {
		sigma=300;
		scanf("%s",s+1);
		n=strlen(s+1);
		GetSa(s);
		GetHeight(s);
		return n;
	}
}tree;
//-------end----------
int main() {
	len=tree.init(s);
	for(int i=1;i<=len;i++)cout<<tree.rnk[i]<<' ';
	puts("");
	for(int i=1;i<=len;i++)cout<<tree.height[i]<<' ';
}