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
思路
后缀数组,先破环成链,注意需要延长一倍,然后对后缀进行排序。这样就求出了序列的每一个开始位置,将开始位置的前一个字符按顺序输出即可。
代码
#include <cstdio>
#include <cstring>
#include <algorithm>
const int maxn=400000;
struct suffix_array//后缀数组的模板
{
int sa[maxn+10],rank[maxn+10],tmp[maxn+10],sum[maxn+10];
int suffix_sort(char* s)
{
int* x=rank;
int* y=tmp;
int n=strlen(s+1),m=127;
for(register int i=1; i<=n; ++i)
{
x[i]=s[i];
y[i]=i;
++sum[x[i]];
}
for(register int i=1; i<=m; ++i)
{
sum[i]+=sum[i-1];
}
for(register int i=n; i; --i)
{
sa[sum[x[i]]]=i;
--sum[x[i]];
}
int len=1,p=1;
while(p<n)
{
m=p;
p=0;
for(register int i=n-len+1; i<=n; ++i)
{
++p;
y[p]=i;
}
for(register int i=1; i<=n; ++i)
{
if(sa[i]-len>0)
{
++p;
y[p]=sa[i]-len;
}
}
for(register int i=0; i<=m; ++i)
{
sum[i]=0;
}
for(register int i=1; i<=n; ++i)
{
++sum[x[y[i]]];
}
for(register int i=1; i<=m; ++i)
{
sum[i]+=sum[i-1];
}
for(register int i=n; i; i--)
{
sa[sum[x[y[i]]]]=y[i];
--sum[x[y[i]]];
}
std::swap(x,y);
p=1;
x[sa[1]]=1;
for(register int i=2; i<=n; ++i)
{
if(!((y[sa[i-1]]==y[sa[i]])&&(y[sa[i-1]+len]==y[sa[i]+len])))
{
++p;
}
x[sa[i]]=p;
}
len<<=1;
}
p=0;
for(int i=1; i<=n; ++i)
{
++p;
rank[sa[i]]=p;
}
return 0;
}
int work(char* s)
{
int n=strlen(s+1);
for(register int i=1; i<=n; ++i)//将s扩大一倍
{
s[i+n]=s[i];
}
suffix_sort(s);//对s的后缀进行排序
for(register int i=1; i<=n<<1; ++i)//输出答案
{
int x=sa[i]-1;//这一个串的末尾字符就是这个串开头字符的前一个字符
if(x<n)
{
if(!x)
{
x=n;
}
putchar(s[x]);
}
}
return 0;
}
};
suffix_array s;
char ch[maxn+10];
int main()
{
scanf("%s",ch+1);
s.work(ch);
return 0;
}