题目描述
喜欢钻研问题的JS 同学,最近又迷上了对加密方法的思考。一天,他突然想出了一种他认为是终极的加密办法:把需要加密的信息排成一圈,显然,它们有很多种不同的读法。
例如‘JSOI07’,可以读作: JSOI07 SOI07J OI07JS I07JSO 07JSOI 7JSOI0 把它们按照字符串的大小排序: 07JSOI 7JSOI0 I07JSO JSOI07 OI07JS SOI07J 读出最后一列字符:I0O7SJ,就是加密后的字符串(其实这个加密手段实在很容易破解,鉴于这是突然想出来的,那就^^)。 但是,如果想加密的字符串实在太长,你能写一个程序完成这个任务吗?
说明
对于40%的数据字符串的长度不超过10000。
对于100%的数据字符串的长度不超过100000。
思路
一看到子串排序,就能想到后缀数组
将原串复制一遍接到后面,然后就没了
代码
#include<iostream> #include<cstdio> #include<cstring> #include<algorithm> #define maxn 200010 using namespace std; int n, m; char s[maxn]; int tax[maxn], rk[maxn], tp[maxn], sa[maxn], M = 200; void rsort() { for (int i = 0; i <= M; ++i) tax[i] = 0; for (int i = 1; i <= n; ++i) ++tax[rk[i]]; for (int i = 1; i <= M; ++i) tax[i] += tax[i - 1]; for (int i = n; i; --i) sa[tax[rk[tp[i]]]--] = tp[i]; } int H[maxn]; void SA() { for (int i = 1; i <= n; ++i) rk[i] = s[i], tp[i] = i; int c1 = 0; rsort(); for (int k = 1; k <= n; k *= 2) { if (c1 == n) break; M = c1; c1 = 0; for (int i = n - k + 1; i <= n; ++i) tp[++c1] = i; for (int i = 1; i <= n; ++i) if (sa[i] > k) tp[++c1] = sa[i] - k; rsort(); swap(tp, rk); rk[sa[1]] = c1 = 1; for (int i = 2; i <= n; ++i) { if (tp[sa[i - 1]] != tp[sa[i]] || tp[sa[i - 1] + k] != tp[sa[i] + k]) ++c1; rk[sa[i]] = c1; } } int lcp = 0; for (int i = 1; i <= n; ++i) { if (lcp) --lcp; int j = sa[rk[i] - 1]; while (s[j + lcp] == s[i + lcp]) ++lcp; H[rk[i]] = lcp; } } int ans[maxn * 2]; int main() { scanf("%s", s + 1); m = n = strlen(s + 1); for (int i = n + 1; i <= 2 * n; ++i) s[i] = s[i - n]; n *= 2; s[n + 1] = '\0'; SA(); for (int i = 1; i <= m; ++i) ans[rk[i]] = s[i + m - 1]; for (int i = 1; i <= n; ++i) if (ans[i]) printf("%c", (char) ans[i]); return 0; }