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

解题思路：

sa裸题，将字符串倍长求sa

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
const int N=1000000;
int rnk[N];
int tmr[N];
int has[N];
int sa[N];
int ln[N];
char a[N];
int n;
int len;
int cnt;
bool Same(int a,int b,int l)
{
    if(a+l>len||b+l>len)
        return false;
    return (rnk[a+l]==rnk[b+l])&&(rnk[a]==rnk[b]);
}
int main()
{
    scanf("%s",a+1);
    n=strlen(a+1);
    len=2*n;
    for(int i=1;i<=n;i++)
        ln[i]=ln[i+n]=a[i];
    for(int i=1;i<=len;i++)
        has[ln[i]]++;
    for(int i=0;i<128;i++)
        if(has[i])
            tmr[i]=++cnt;
    for(int i=1;i<128;i++)
        has[i]+=has[i-1];
    for(int i=1;i<=len;i++)
    {
        rnk[i]=tmr[ln[i]];
        sa[has[ln[i]]--]=i;
    }
    for(int k=1;cnt!=len;k<<=1)
    {
        cnt=0;
        for(int i=0;i<=len;i++)
            has[i]=0;
        for(int i=1;i<=len;i++)
            has[rnk[i]]++;
        for(int i=1;i<=len;i++)
            has[i]+=has[i-1];
        for(int i=len;i;i--)
            if(sa[i]>k)
                tmr[sa[i]-k]=has[rnk[sa[i]-k]]--;
        for(int i=1;i<=k;i++)
            tmr[len-i+1]=has[rnk[len-i+1]]--;
        for(int i=1;i<=len;i++)
            sa[tmr[i]]=i;
        for(int i=1;i<=len;i++)
            if(Same(sa[i],sa[i-1],k))
                tmr[sa[i]]=cnt;
            else
                tmr[sa[i]]=++cnt;
        for(int i=1;i<=len;i++)
            rnk[i]=tmr[i];
    }
    for(int i=1;i<=len;i++)
        if(sa[i]<=n)
            printf("%c",a[(sa[i]-2+n)%n+1]);
    puts("");
    return 0;
}