BZOJ 1031 字符加密

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

后缀数组大裸题，将原串接在后面，求一遍sa，再遍历一遍sa即可。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cstdlib>
using namespace std;

#define maxn 200010
char s[maxn];
int sa[maxn],t[maxn];
int t2[maxn],c[maxn],n;

inline void build(int m)
{
    int i,*x = t,*y = t2;
    memset(c,0,4*(m+1));
    for (i = 0;i < n;++i) c[x[i] = s[i]]++;
    for (i = 1;i < m;++i) c[i] += c[i-1];
    for (i = n-1;i >= 0;--i) sa[--c[x[i]]] = i;
    for (int k = 1;k < n;k <<= 1)
    {
        int p = 0;
        for (i = n - k;i < n;++i) y[p++] = i;
        for (i = 0;i < n;++i) if (sa[i] >= k) y[p++] = sa[i]-k;
        memset(c,0,4*(m+1));
        for (i = 0;i < n;++i) c[x[y[i]]]++;
        for (i = 1;i < m;++i) c[i] += c[i-1];
        for (i = n - 1;i >= 0;--i) sa[--c[x[y[i]]]] = y[i];
        swap(x,y);
        p = 1; x[sa[0]] = 0;
        for (i = 1;i < n;++i)
            x[sa[i]] = (y[sa[i-1]] == y[sa[i]] && y[sa[i-1]+k] == y[sa[i]+k] ? p - 1:p++);
        if (p >= n) break;
        m = p;
    }
}

int main()
{
    freopen("1031.in","r",stdin);
    freopen("1031.out","w",stdout);
    scanf("%s",s);
    n = strlen(s);
    for (int i = n;i < 2*n-1;++i) s[i] = s[i - n];
    n <<= 1;
    build(128);
    for (int i = 0;i < n;++i) if (sa[i] < (n >> 1)) printf("%c",s[sa[i]+(n >> 1) - 1]);
    fclose(stdin); fclose(stdout);
    return 0;
}