bzoj1031 [JSOI2007]字符加密Cipher

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\) 。

Solution

后缀数组裸题。答案就是 \(s[sa[i] + n - 1](s=input+input,sa[i]<=n)\)

#include<cstring>
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
#include<cmath>
using namespace std;

#define N 200001
#define rep(i, a, b) for (int i = a; i <= b; i++)
#define drp(i, a, b) for (int i = a; i >= b; i--)
#define ll long long

inline int read() {
	int x = 0, flag = 1; char ch = getchar(); while (!isdigit(ch)) { if (!(ch ^ '-')) flag = -1; ch = getchar(); }
	while (isdigit(ch)) x = (x << 1) + (x << 3) + ch - '0', ch = getchar(); return x * flag;
}


int n;
char in[N]; int s[N];
int sa[N], t1[N], t2[N], c[N], rk[N], height[N];

void get_sa() {
	int *x = t1, *y = t2, m = 54; 
	rep(i, 1, n) c[x[i] = s[i]]++;
	rep(i, 1, m) c[i] += c[i - 1];
	drp(i, n, 1) sa[c[x[i]]--] = i;
	for (int k = 1; k <= n; k <<= 1) {
		int p = 0;
		rep(i, n - k + 1, n) y[++p] = i;
		rep(i, 1, n) if (sa[i] > k) y[++p] = sa[i] - k;
		memset(c, 0, sizeof(c));
		rep(i, 1, n) c[x[i]]++;
		rep(i, 1, m) c[i] += c[i - 1];
		drp(i, n, 1) sa[c[x[y[i]]]--] = y[i];
		swap(x, y), p = 0, x[sa[1]] = ++p;
		rep(i, 2, n) x[sa[i]] = y[sa[i - 1]] == y[sa[i]] && y[sa[i - 1] + k] == y[sa[i] + k] ? p : ++p;
		if ((m = p) == n) break;
	}
}

void get_height() {
	rep(i, 1, n) rk[sa[i]] = i;
	int k = 0;
	rep(i, 1, n) {
		if (k) k--;
		int p = sa[rk[i] - 1];
		while (s[i + k] == s[p + k]) k++;
		height[rk[i]] = k;
	}
}

int main() {
	scanf("%s", in + 1); n = strlen(in + 1);
	rep(i, 1, n) in[i + n] = in[i], s[n + i] = s[i] = (in[i] >= 'a' && in[i] <= 'z') ? in[i] - 'a' + 1 : in[i] - 'A' + 1 + 26;
	n += n; get_sa();
	rep(i, 1, n) if (sa[i] <= n / 2) printf("%c", in[sa[i] + n / 2 - 1]);
	return 0;
}