[JSOI2007]字符加密 后缀数组

题目描述

喜欢钻研问题的JS 同学，最近又迷上了对加密方法的思考。一天，他突然想出了一种他认为是终极的加密办法：把需要加密的信息排成一圈，显然，它们有很多种不同的读法。

例如‘JSOI07’，可以读作： JSOI07 SOI07J OI07JS I07JSO 07JSOI 7JSOI0 把它们按照字符串的大小排序： 07JSOI 7JSOI0 I07JSO JSOI07 OI07JS SOI07J 读出最后一列字符：I0O7SJ，就是加密后的字符串（其实这个加密手段实在很容易破解，鉴于这是突然想出来的，那就^^）。 但是，如果想加密的字符串实在太长，你能写一个程序完成这个任务吗？

输入输出格式

输入格式：

 

输入文件包含一行，欲加密的字符串。注意字符串的内容不一定是字母、数字，也可以是符号等。

 

输出格式：

 

输出一行，为加密后的字符串。

 

 

题解： 

后缀数组模板题，直接套上即可. 

Code:

#include <cstdio>
#include <algorithm>
#include <cstring>
#define setIO(s) freopen(s".in","r",stdin)
#define maxn 1000000
using namespace std;
char str[maxn];
int n,m,rk[maxn],sa[maxn],tp[maxn],c[maxn],arr[maxn];
void qsort(){
    for(int i=0;i<=m;++i) c[i]=0;
    for(int i=1;i<=n;++i) ++c[rk[tp[i]]];
    for(int i=1;i<=m;++i) c[i]+=c[i-1];
    for(int i=n;i>=1;--i) sa[c[rk[tp[i]]]--]=tp[i]; 
}
void suffix(){
    for(int i=1;i<=n;++i) rk[i]=arr[i],tp[i]=i;
    qsort();
    for(int k=1;k<=n;k<<=1){
        int num=0;
        for(int i=n-k+1;i<=n;++i) tp[++num]=i;
        for(int i=1;i<=n;++i) if(sa[i]>k)tp[++num]=sa[i]-k;
        qsort();
        swap(rk,tp);
        rk[sa[1]]=1;
        num=1;
        for(int i=2;i<=n;++i)
            rk[sa[i]]=(tp[sa[i]]==tp[sa[i-1]]&&tp[sa[i-1]+k]==tp[sa[i]+k]) ? num : ++num; 
        if(num==n) break;
        m=num; 
    }
}
int main(){
    //setIO("input"); 
    int cnt;
    scanf("%s",str);
    cnt=n=strlen(str);
    m=200; 
    for(int i=n;i<=2*n-1;++i) str[i]=str[i-n];
    n*=2;
    for(int i=1;i<=n;++i) arr[i]=str[i-1]; 
    suffix();
    //printf("%d\n",cnt); 
    for(int i=1;i<=n;++i){
        if(sa[i]>cnt) continue;
      //  printf("%d ",sa[i]+cnt-2); 
        putchar(str[sa[i]+cnt-2]); 
    }
    return 0;
}