bzoj1031:[JSOI2007]字符加密Cipher

传送门

后缀排序裸题,SA上就是了
代码:

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
void read(int &x) {
	char ch; bool ok;
	for(ok=0,ch=getchar(); !isdigit(ch); ch=getchar()) if(ch=='-') ok=1;
	for(x=0; isdigit(ch); x=x*10+ch-'0',ch=getchar()); if(ok) x=-x;
}
#define rg register
const int maxn=2e5+10;
char p[maxn];
int n,m=900,now,sa[maxn],x[maxn],q[maxn],y[maxn],a[maxn];
int main()
{
	scanf("%s",p+1);n=strlen(p+1);
	for(rg int i=1;i<=n;i++)p[n+i]=p[i];n<<=1;
	for(rg int i=1;i<=n;i++)a[x[i]=p[i]]++;
	for(rg int i=1;i<=m;i++)a[i]+=a[i-1];
	for(rg int i=n;i>=1;i--)sa[a[x[i]]--]=i;
	for(rg int k=1;k<=n;k<<=1,now=0)
	{
		for(rg int i=n-k+1;i<=n;i++)q[++now]=i;
		for(rg int i=1;i<=n;i++)if(sa[i]>k)q[++now]=sa[i]-k;
		for(rg int i=1;i<=m;i++)a[i]=0;
		for(rg int i=1;i<=n;i++)a[x[i]]++;
		for(rg int i=1;i<=m;i++)a[i]+=a[i-1];
		for(rg int i=n;i>=1;i--)sa[a[x[q[i]]]--]=q[i];
		for(rg int i=1;i<=n;i++)q[i]=x[i];
		now=1;x[sa[1]]=1;
		for(rg int i=2;i<=n;i++)
			if(q[sa[i]]!=q[sa[i-1]]||q[sa[i]+k]!=q[sa[i-1]+k])x[sa[i]]=++now;
			else x[sa[i]]=now;
		if(now>=n)break;m=now;
	}
	for(rg int i=1;i<=n;i++)if(sa[i]<=n/2)printf("%c",p[sa[i]+n/2-1]);
}
posted @ 2019-02-27 20:58  蒟蒻--lichenxi  阅读(68)  评论(0编辑  收藏  举报