[luogu] P3809 【模板】后缀排序 (SA)

板子,照着题解打的倍增版。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;const int N=1e6+7;
char s[N];int n,m,x[N],y[N],bot[N],sa[N];
int main()
{
	gets(s+1);n=strlen(s+1);m=122; 
	for(int i=1;i<=n;i++) bot[x[i]=s[i]]++;
	for(int i=1;i<=m;i++) bot[i]+=bot[i-1];
	for(int i=n;i>=1;i--) sa[bot[x[i]]--]=i;
	for(int k=1;k<=n;k<<=1)
	{
		int num=0;
		for(int i=n-k+1;i<=n;i++) y[++num]=i;
		for(int i=1;i<=n;i++) if(sa[i]>k) y[++num]=sa[i]-k;
		for(int i=1;i<=m;i++) bot[i]=0;
		for(int i=1;i<=n;i++) bot[x[i]]++;
		for(int i=1;i<=m;i++) bot[i]+=bot[i-1];
		for(int i=n;i>=1;i--) sa[bot[x[y[i]]]--]=y[i],y[i]=0;
		swap(x,y);x[sa[1]]=1;num=1;
		for(int 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(int i=1;i<=n;i++) cout<<sa[i]<<" ";
	return 0;
}
posted @ 2019-12-24 14:21  chtomede  阅读(165)  评论(0编辑  收藏  举报