【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 << endl
inline 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 << endl
inline 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