【BZOJ 2251】[2010Beijing Wc]外星联络 后缀数组

我的模板是用的大白上面的，但是比较坑的是，我发现例如aaaa求出来的sa数组并不正确，而这一道题中用哪个版也是过不了，找了很久发现在重新更新x数组的时候是需要判断是否越界的，改过来就对了，有点。。。。

思路很简单，够着sa数组和h数组然后暴力统计

#include<cstdio>
#include<cstring>
#include<iostream>
#define maxn 3010
using namespace std;
int n,t1[maxn],t2[maxn],sa[maxn],c[maxn],h[maxn],rank[maxn];
char s[maxn];

bool cmp(int* y,int a,int b,int k){
	int aa=a+k>=n?-1:y[a+k];
	int bb=b+k>=n?-1:y[b+k];
	return y[a]==y[b]&&aa==bb;
}

void make(int n,int m){
	int *x=t1,*y=t2,i,p=0;
	for(i=0;i<n;i++)c[x[i]=s[i]]++;
	for(i=1;i<=m;i++)c[i]+=c[i-1];
	for(i=n-1;i>=0;i--)sa[--c[x[i]]]=i;
	for(int k=1;k<=n;k<<=1){
		p=0;
		for(i=n-k;i<n;i++)y[p++]=i;
		for(i=0;i<n;i++)if(sa[i]>=k)y[p++]=sa[i]-k;
		
		for(i=0;i<=m;i++)c[i]=0;
		for(i=0;i<n;i++)c[x[y[i]]]++;
		for(i=1;i<=m;i++)c[i]+=c[i-1];
		for(i=n-1;i>=0;i--)sa[--c[x[y[i]]]]=y[i];
		
		swap(x,y);p=1;x[sa[0]]=0;
		for(i=1;i<n;i++)
			x[sa[i]]=cmp(y,sa[i],sa[i-1],k)? p-1 : p++;
		if(p>=n)break;m=p;
	}
}

void get_h(){
	for(int i=0;i<n;i++)rank[sa[i]]=i;
	for(int k=0,i=0,j;i<n;i++){
		if(k)k--;
		if(!rank[i])continue;
		j=sa[rank[i]-1];
		while(s[i+k]==s[j+k])k++;
		h[rank[i]]=k;
	}
}

void solve(){
	int k=0,l;
	for(int i=0;i<n;i++){
		for(int j=h[i]+1;j+sa[i]-1<n;j++){
			l=i;
			while(h[l+1]>=j)l++;
			if(l>i)printf("%d\n",l-i+1);
		}
	}
}

int main(){
	scanf("%d%s",&n,s);
	make(n,129);
	get_h();
	solve();
	return 0;
}