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。
一道后缀数组模板题,给出自己写的模板。
后缀数组我是在百度上学的。
注意桶排序的时候要从后往前生成字符串,因为在生成完后面的时候,前面的是有序的。然而从前往后生成时后面是无序的。
#include<algorithm>
#include<iostream>
#include<fstream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<vector>
#include<cmath>
#include<set>
#include<map>
#define N 200010
using namespace std;
char ch[N];
int sa[N],v[N];
int len;
inline bool pd_same(int x,int y,int l,int len)
{
return v[x] == v[y] &&
((x + l >= len && y + l >= len) || (x + l < len && y + l < len && v[x + l] == v[y + l]));
}
inline void Make_Sa()
{
int lim = 256, cnt[N], val[N], q[N],j;
for(int i = 0; i < lim; i ++) cnt[i] = 0;
for(int i = 0; i < len; i ++) ++ cnt[ v[i] = ch[i] ];
for(int i = 1; i < lim; i ++) cnt[i] += cnt[i - 1];
for(int i = len - 1; i >= 0; i --) sa[-- cnt[ v[i] ] ] = i;
for(int d = 1; ; d ++)
{
int top = 0;
int l = 1 << (d - 1);
for(int i = 0; i < len; i ++)
if(sa[i] + l >= len)
q[top ++] = sa[i];
for(int i = 0; i < len; i ++)
if(sa[i] >= l) q[top ++] = sa[i] - l;
for(int i = 0; i < lim; i ++) cnt[i] = 0;
for(int i = 0; i < len; i ++) ++ cnt[ v[ q[i] ] ];
for(int i = 1; i < lim; i ++) cnt[i] += cnt[i - 1];
for(int i = len - 1; i >= 0; i --) sa[ -- cnt[ v[ q[i] ] ] ] = q[i];
lim = 0;
for(int i = 0; i < len; lim ++)
{
j = i;
while(j < len - 1 && pd_same(sa[j], sa[j + 1], l, len))
j ++;
for(; i <= j; i ++) val[sa[i]] = lim;
}
for(int i = 0; i < len; i ++) v[i] = val[i];
if(lim == len) break;
}
}
int main()
{
scanf("%s",ch);
len = strlen(ch);
for(int i = len; i < len << 1; i ++)
ch[i] = ch[i - len];
ch[len << 1] = '\0';
len = strlen(ch);
Make_Sa();
for(int i = 0; i < len; i ++)
if(sa[i] < (len >> 1))
putchar(ch[sa[i] + (len >> 1) - 1]);
return 0;
}