BZOJ1031: [JSOI2007]字符加密Cipher

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

这道题其实是一个叫做Burrows-Wheeler变换的东西。
后缀数组即可。但是要注意，题目里不是要跟我们玩循环吗，那就干脆把字符串先复制一遍，n*2来解决循环的问题，然后再用后缀数组，否则其顺序会出现问题。

#include
#include
using namespace std;
const int N=200010;
char ch[N];int n,k,p=0,q=1;
int sa[2][N],rank[2][N],c[N];
void solve(int sa1[N],int rank1[N],int sa2[N],int rank2[N])
{
    for (int i=1;i<=n;++i) c[rank1[sa1[i]]]=i;
    for (int i=n;i;--i) if (sa1[i]>k) sa2[c[rank1[sa1[i]-k]]--]=sa1[i]-k;
    for (int i=n-k+1;i<=n;++i) sa2[c[rank1[i]]--]=i;
    for (int i=1;i<=n;++i)
    rank2[sa2[i]]=rank2[sa2[i-1]]+(rank1[sa2[i-1]]!=rank1[sa2[i]]||rank1[sa2[i-1]+k]!=rank1[sa2[i]+k]);
}
int main()
{
    scanf("%s",ch+1);n=strlen(ch+1);
    for (int i=1;i<=n;++i) ch[i+n]=ch[i];n<<=1;
    for (int i=1;i<=n;++i) ++c[(int)(ch[i])];
    for (int i=1;i<=256;++i) c[i]+=c[i-1];
    for (int i=1;i<=n;++i) sa[p][c[(int)(ch[i])]--]=i;
    for (int i=1;i<=n;++i) rank[p][sa[p][i]]=rank[p][sa[p][i-1]]+(ch[sa[p][i]]!=ch[sa[p][i-1]]);
    for (k=1;k1) solve(sa[p],rank[p],sa[q],rank[q]),p^=1,q^=1;
    for (int i=1;i<=n;++i) if (sa[p][i]<=(n>>1)) putchar(ch[sa[p][i]+(n>>1)-1]);
    return 0;
}