bzoj 1031 [JSOI2007]字符加密Cipher

题目描述

喜欢钻研问题的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。

解题思路

把字符串复制一遍（开环），求后缀数组。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
using namespace std;

const int N=2e5+3;
int n,m,len,rank[N],sa[N],height[N],c[N];
char s[N],ans[N];

void SA_Init()
{
    int *x=rank,*y=height,r=256;
    for(int i=0;i<n;i++)c[x[i]=s[i]]++;
    for(int i=1;i<=r;i++)c[i]+=c[i-1];
    for(int i=n-1;i>=0;i--)sa[--c[x[i]]]=i;
    for(int k=1;k<n;k<<=1)
    {
        int j=0;
        for(int i=n-k;i<n;i++)y[j++]=i;
        for(int i=0;i<n;i++)if(sa[i]>=k)y[j++]=sa[i]-k;
        for(int i=0;i<=r;i++)c[i]=0;
        for(int i=0;i<n;i++)c[x[i]]++;
        for(int i=1;i<=r;i++)c[i]+=c[i-1];
        for(int i=n-1;i>=0;i--)sa[--c[x[y[i]]]]=y[i];
        swap(x,y);
        j=x[sa[0]]=1;
        for(int i=1;i<n;i++)
            x[sa[i]]=y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k]?j:++j;
        r=j+1;
    }
}

int main()
{
    //freopen("lx.in","r",stdin);
    scanf("%s",s);
    n=m=strlen(s);
    for(int i=0;i<n;i++)s[i+n]=s[i];
    n<<=1;
    SA_Init();
    for(int i=0;i<n;i++)
        if(sa[i]<m)ans[len++]=s[sa[i]+m-1];
    printf("%s",ans);
    return 0;
}