首页 > 代码库 > 【BZOJ】1031: [JSOI2007]字符加密Cipher(后缀数组)

【BZOJ】1031: [JSOI2007]字符加密Cipher(后缀数组)

http://www.lydsy.com/JudgeOnline/problem.php?id=1031

很容易想到这就是将字符串复制到自己末尾然后后缀数组搞出sa然后按区间输出即可。

然后换了下模板,将基数排序放到外边

#include <cstdio>#include <cstring>#include <cmath>#include <string>#include <iostream>#include <algorithm>#include <queue>#include <set>#include <map>using namespace std;typedef long long ll;#define pii pair<int, int>#define mkpii make_pair<int, int>#define pdi pair<double, int>#define mkpdi make_pair<double, int>#define pli pair<ll, int>#define mkpli make_pair<ll, int>#define rep(i, n) for(int i=0; i<(n); ++i)#define for1(i,a,n) for(int i=(a);i<=(n);++i)#define for2(i,a,n) for(int i=(a);i<(n);++i)#define for3(i,a,n) for(int i=(a);i>=(n);--i)#define for4(i,a,n) for(int i=(a);i>(n);--i)#define CC(i,a) memset(i,a,sizeof(i))#define read(a) a=getint()#define print(a) printf("%d", a)#define dbg(x) cout << (#x) << " = " << (x) << endl#define error(x) (!(x)?puts("error"):0)#define printarr2(a, b, c) for1(_, 1, b) { for1(__, 1, c) cout << a[_][__]; cout << endl; }#define printarr1(a, b) for1(_, 1, b) cout << a[_] << ‘\t‘; cout << endlinline const int getint() { int r=0, k=1; char c=getchar(); for(; c<‘0‘||c>‘9‘; c=getchar()) if(c==‘-‘) k=-1; for(; c>=‘0‘&&c<=‘9‘; c=getchar()) r=r*10+c-‘0‘; return k*r; }inline const int max(const int &a, const int &b) { return a>b?a:b; }inline const int min(const int &a, const int &b) { return a<b?a:b; }const int N=200005;char s[N];int t1[N], t2[N], sa[N], c[N];void st(int *x, int *y, int n, int m) {	int i;	for(i=0; i<m; ++i) c[i]=0;	for(i=0; i<n; ++i) c[ x[y[i]] ]++;	for(i=1; i<m; ++i) c[i]+=c[i-1];	for(i=n-1; i>=0; --i) sa[ --c[x[y[i]]] ]=y[i];}void hz(char *a, int n, int m) {	int i, j, p, *x=t1, *y=t2, *t;	for(i=0; i<n; ++i) x[i]=s[i], y[i]=i;	st(x, y, n, m);	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;		st(x, y, n, m);		for(t=x, x=y, y=t, p=1, x[sa[0]]=0, i=1; i<n; ++i)			x[sa[i]]=y[sa[i]]==y[sa[i-1]]&&y[sa[i]+j]==y[sa[i-1]+j]?p-1:p++;	}}int main() {	scanf("%s", s+1);	int n=strlen(s+1);	for1(i, 1, n) s[i+n]=s[i];	s[0]=0;	hz(s, n*2+1, 256);	int tot=1, i=1;	while(tot<=n) {		//dbg(sa[i]);		if(sa[i]<=n) {			printf("%c", s[sa[i]+n-1]);			++tot;		}		++i;	}	return 0;}

原版的后缀数组:

#include <cstdio>#include <cstring>#include <cmath>#include <string>#include <iostream>#include <algorithm>#include <queue>#include <set>#include <map>using namespace std;typedef long long ll;#define pii pair<int, int>#define mkpii make_pair<int, int>#define pdi pair<double, int>#define mkpdi make_pair<double, int>#define pli pair<ll, int>#define mkpli make_pair<ll, int>#define rep(i, n) for(int i=0; i<(n); ++i)#define for1(i,a,n) for(int i=(a);i<=(n);++i)#define for2(i,a,n) for(int i=(a);i<(n);++i)#define for3(i,a,n) for(int i=(a);i>=(n);--i)#define for4(i,a,n) for(int i=(a);i>(n);--i)#define CC(i,a) memset(i,a,sizeof(i))#define read(a) a=getint()#define print(a) printf("%d", a)#define dbg(x) cout << (#x) << " = " << (x) << endl#define error(x) (!(x)?puts("error"):0)#define printarr2(a, b, c) for1(_, 1, b) { for1(__, 1, c) cout << a[_][__]; cout << endl; }#define printarr1(a, b) for1(_, 1, b) cout << a[_] << ‘\t‘; cout << endlinline const int getint() { int r=0, k=1; char c=getchar(); for(; c<‘0‘||c>‘9‘; c=getchar()) if(c==‘-‘) k=-1; for(; c>=‘0‘&&c<=‘9‘; c=getchar()) r=r*10+c-‘0‘; return k*r; }inline const int max(const int &a, const int &b) { return a>b?a:b; }inline const int min(const int &a, const int &b) { return a<b?a:b; }const int N=200005;char s[N];int t1[N], t2[N], sa[N], c[N];void hz(char *a, int n, int m) {	int i, j, p, *x=t1, *y=t2, *t;	for(i=0; i<m; ++i) c[i]=0;	for(i=0; i<n; ++i) c[ x[i]=a[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<m; ++i) c[i]=0;		for(i=0; i<n; ++i) c[ x[y[i]] ]++;		for(i=1; i<m; ++i) c[i]+=c[i-1];		for(i=n-1; i>=0; --i) sa[ --c[x[y[i]]] ]=y[i];		for(t=x, x=y, y=t, p=1, x[sa[0]]=0, i=1; i<n; ++i)			x[sa[i]]=y[sa[i]]==y[sa[i-1]]&&y[sa[i]+j]==y[sa[i-1]+j]?p-1:p++;	}}int main() {	scanf("%s", s+1);	int n=strlen(s+1);	for1(i, 1, n) s[i+n]=s[i];	s[0]=0;	hz(s, n*2+1, 256);	int tot=1, i=1;	while(tot<=n) {		//dbg(sa[i]);		if(sa[i]<=n) {			printf("%c", s[sa[i]+n-1]);			++tot;		}		++i;	}	return 0;}

  

 


 

 

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。

 

Source

【BZOJ】1031: [JSOI2007]字符加密Cipher(后缀数组)