后缀数组

本代码只求出sa[ ], rank[ ], height[ ]

#include <cstdio>
#include <cstring>
const int maxn=200;
char s[maxn];//假设字符串中出现的都是大写英文字母
int sa[maxn],rank[maxn],height[maxn],wa[maxn],wb[maxn],wv[maxn],ws[maxn],n;
int cmp(int *r,int a,int b,int l)
{
	if(r[a]!=r[b]) return 1;
	if(a+l>=n &&b+l>=n) return 0;
	if(a+l<n && b+l<n) return r[a+l]!=r[b+l];
	return 1;
}
void build_sa(int m)//m表示大于 数组s中出现的最大值
{
	int i,j,p,*x=wa,*y=wb,*t;
	for(i=0;i<m;i++) ws[i]=0;
	for(i=0;i<n;i++) ws[x[i]=s[i]]++;
	for(i=1;i<m;i++) ws[i]+=ws[i-1];
	for(i=n-1;i>=0;i--) sa[--ws[x[i]]]=i;
	
	for(j=1,p=1;p<n;j<<=1,m=p)//p表示x数组中出现不同数值的个数，j表示关键字代表的字符的个数
	{
		//首先对第二关键排序
		p=0;
		for(i=n-j;i<n;i++) y[p++]=i;//先排序没有第二关键字的
		for(i=0;i<n;i++) if(sa[i]>=j) y[p++]=sa[i]-j;//利用sa数组来对有第二关键字的排序，y数组的含义：y[i]为第i个（i>=0）第二关键字
		                                                 //所对应的字符串的第一关键字的下标
		for(i=0;i<n;i++) wv[i]=x[y[i]];//wv表示经过对第二关键字的排序后处在i位置的字符串对应的第一关键字的值
		//对第一关键字排序
		for(i=0;i<m;i++) ws[i]=0;
		for(i=0;i<n;i++) ws[wv[i]]++;
		for(i=1;i<m;i++) ws[i]+=ws[i-1];
		for(i=n-1;i>=0;i--) sa[--ws[wv[i]]]=y[i];
		t=x;x=y;y=t;x[sa[0]]=0;p=1;
		for(i=1;i<n;i++) x[sa[i]]=cmp(y,sa[i-1],sa[i],j)==0?p-1:p++;
	}
}
void get_height()
{
	int i,j,k=0;
	for(i=0;i<n;i++) rank[sa[i]]=i;
	for(i=0;i<n;i++)
	{
		if(k) k--;
		j=rank[i]-1;
		if(j==-1) 
		{
			height[0]=0;
			continue;
		}
		j=sa[j];
		while(s[i+k]==s[j+k]) k++;
		height[rank[i]]=k;
	}
}
int main()
{
	int i,m;
	while(scanf("%s",s)!=EOF)
	{
		n=strlen(s);
		for(m=0,i=0;i<n;i++) m=m>s[i]?m:s[i];
		build_sa(m+1);
		get_height();
		for(i=0;i<n;i++)
			printf("%5d",sa[i]);
		printf("\n");
		for(i=0;i<n;i++)
			printf("%5d",rank[i]);
		printf("\n");
		for(i=0;i<n;i++)
			printf("%5d",height[i]);
		printf("\n");
	}
	return 0;
}