比赛 清华集训2017模板练习 评测结果 AAAAAAATTA
题目名称 后缀排序 最终得分 80
用户昵称 FoolMike 运行时间 1.735 s
代码语言 C++ 内存使用 21.74 MiB
提交时间 2017-07-17 18:30:36
显示代码纯文本
#include<bits/stdc++.h>
using namespace std;
const int N=1e6+10;
char s[N];
int n,sa[N],Wa[N*2],Wb[N*2],Ws[N];
bool cmp(int *r,int x,int y,int l){
	return r[x]==r[y]&&r[x+l]==r[y+l];
}
void getsa(){
	int m=1000,*x=Wa,*y=Wb;
	for (int i=1;i<=n;i++) Ws[x[i]=s[i]]++;
	for (int i=1;i<=m;i++) Ws[i]+=Ws[i-1];
	for (int i=n;i;i--) sa[Ws[x[i]]--]=i;
	for (int l=1,p=0;p<n;m=p,l<<=1){
		p=0;
		for (int i=n-l+1;i<=n;i++) y[++p]=i;
		for (int i=1;i<=n;i++)
			if (sa[i]>l) y[++p]=sa[i]-l;
		for (int i=0;i<=m;i++) Ws[i]=0;
		for (int i=1;i<=n;i++) Ws[x[y[i]]]++;
		for (int i=1;i<=m;i++) Ws[i]+=Ws[i-1];
		for (int i=n;i;i--) sa[Ws[x[y[i]]]--]=y[i];
		swap(x,y);
		x[sa[1]]=p=1;
		for (int i=2;i<=n;i++)
			x[sa[i]]=cmp(y,sa[i],sa[i-1],l)?p:++p; 
	}
}
void print(int x){
	if (!x) putchar('0');
	else{
		static char s[20];
		int l=0;
		for (;x;x/=10) s[++l]=x%10;
		for (;l;putchar(s[l--]+'0'));
	}
	putchar(' ');
}
int main()
{
	freopen("sais.in","r",stdin);
	freopen("sais.out","w",stdout);
	scanf("%s",s+1);
	n=strlen(s+1);
	getsa();
	for (int i=1;i<=n;i++) print(sa[i]-1);
	puts("");
	return 0;
}