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;
}