uoj P35 后缀排序

传送门

妥妥的后缀数组裸题(主要是楼主菜只会打这种题目),还有拓展lucas还没调对qwq菜飞了。。。

求sa和lcp,就是后缀数组求sa和height。

代码:

#include 
#include
#include
using namespace std;
const int Maxn=100005;
char a[Maxn];
int tax[Maxn],tp[Maxn],sa[Maxn],rank[Maxn],height[Maxn];
int m=127,n;
void Rsort()
{
	for(int i=1;i<=m;i++)
		tax[i]=0;
	for(int i=1;i<=n;i++)
		tax[rank[tp[i]]]++;
	for(int i=1;i<=m;i++)
		tax[i]+=tax[i-1];
	for(int i=n;i>=1;i--)
		sa[tax[rank[tp[i]]]--]=tp[i];
}
int cmp(int *f,int x,int y,int w)
{
	return (f[x]==f[y])&&(f[x+w]==f[y+w]);
}
void suffix()
{
	for(int i=1;i<=n;i++)
		rank[i]=a[i],tp[i]=i;
	Rsort();
	for(int w=1,p=1,i;pw)
				tp[++p]=sa[i]-w;
		Rsort();
		swap(rank,tp);
		rank[sa[1]]=p=1;
		for(int i=2;i<=n;i++)
			rank[sa[i]]=cmp(tp,sa[i],sa[i-1],w)?p:++p;
	}
	for(int i=1,j,p;i<=n;height[rank[i]]=j,i++)
		for(j=max(height[rank[i-1]]-1,0),p=sa[rank[i]-1];a[p+j]==a[i+j];j++);
}
int main()
{
	scanf("%s",a+1);
	n=strlen(a+1);
	suffix();
	for(int i=1;i<=n;i++)
		printf("%d ",sa[i]);
	printf("\n");
	for(int i=2;i<=n;i++)
		printf("%d ",height[i]);
	return 0;
}


你可能感兴趣的:(后缀数组,uoj)