BZOJ 1031: [JSOI2007]字符加密Cipher

1031: [JSOI2007]字符加密Cipher

Time Limit: 10 Sec  Memory Limit: 162 MB

Submit: 7544  Solved: 3273

[Submit][Status][Discuss]

Description

　　喜欢钻研问题的JS同学，最近又迷上了对加密方法的思考。一天，他突然想出了一种他认为是终极的加密办法：把需要加密的信息排成一圈，显然，它们有很多种不同的读法。例如下图，可以读作：

JSOI07 SOI07J OI07JS I07JSO 07JSOI 7JSOI0把它们按照字符串的大小排序：07JSOI 7JSOI0 I07JSO JSOI07 OI07JS SOI07J读出最后一列字符：I0O7SJ，就是加密后的字符串（其实这个加密手段实在很容易破解，鉴于这是突然想出来的，那就^^）。但是，如果想加密的字符串实在太长，你能写一个程序完成这个任务吗？

Input

　　输入文件包含一行，欲加密的字符串。注意字符串的内容不一定是字母、数字，也可以是符号等。

Output

　　输出一行，为加密后的字符串。

Sample Input

JSOI07

Sample Output

I0O7SJ

HINT

对于100%的数据字符串的长度不超过100000。

题解

将字符串翻倍接在后面，直接后缀数组排序后输出s[sa[i]+len-1](sa[i]<len)。

代码

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<iostream>
using namespace std;
const int N=100005;
int len;
int sa[N*2],a[N*2],b[N*2],c[N*2],d[N*2];
char s[N*2];
bool cmp(int *y,int a,int b,int l){
	return y[a]==y[b]&&y[a+l]==y[b+l];
}
void get_sa(int n,int m){
	int i,j,p,*t,*x=a,*y=b;
	for(i=0;i<m;i++)c[i]=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(j=1,p=1;p<n;j<<=1,m=p){
		for(p=0,i=n-j;i<n;i++)y[p++]=i;
		for(i=0;i<n;i++)if(sa[i]>=j)y[p++]=sa[i]-j;
		for(i=0;i<n;i++)d[i]=x[y[i]];
		for(i=0;i<m;i++)c[i]=0;
		for(i=0;i<n;i++)c[d[i]]++;
		for(i=1;i<m;i++)c[i]+=c[i-1];
		for(i=n-1;i>=0;i--)sa[--c[d[i]]]=y[i];
		t=x,x=y,y=t;
		for(i=1,p=1,x[sa[0]]=0;i<n;i++)
			x[sa[i]]=cmp(y,sa[i],sa[i-1],j)?p-1:p++;
	}
}
int main(){
	scanf("%s",s);
	len=strlen(s);
	for(int i=0;i<len;i++)s[i+len]=s[i];
	get_sa(len*2+1,128);
	for(int i=1;i<=len*2;i++){
		if(sa[i]<len)printf("%c",s[sa[i]+len-1]);
	}
	printf("
");
	return 0;
}