BZOJ 1031: [JSOI2007]字符加密Cipher

1031: [JSOI2007]字符加密Cipher

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 1379  Solved: 529
[Submit][Status][Discuss]

Description

喜欢钻研问题的JS 同学，最近又迷上了对加密方法的思考。一天，他突然想出了一种他认为是终极的加密办法：把需要加密的信息排成一圈，显然，它们有很多种不同的读法。例如下图，可以读作： JSOI07 SOI07J OI07JS I07JSO 07JSOI 7JSOI0 把它们按照字符串的大小排序： 07JSOI 7JSOI0 I07JSO JSOI07 OI07JS SOI07J 读出最后一列字符：I0O7SJ，就是加密后的字符串（其实这个加密手段实在很容易破解，鉴于这是突然想出来的，那就^^）。 但是，如果想加密的字符串实在太长，你能写一个程序完成这个任务吗？

Input

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

Output

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

Sample Input

JSOI07

Sample Output

I0O7SJ

数据规模
对于40%的数据字符串的长度不超过10000。
对于100%的数据字符串的长度不超过100000。

思路：后缀数组后缀数组后缀数组……

只用求sa数组就可以了，其实难点不再后面，就是看后缀数组写的来不。（http://wenku.baidu.com/view/87116e4d2e3f5727a5e9622d.html）

http://wenku.baidu.com/view/ed1be61e10a6f524ccbf85fd.html

先把字符串复制一遍，至于求出sa数组后，我们知道只用顺序扫一遍，那么符合条件的顺序就是字典序了。（输出即可！）

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <math.h>
#include <map>
#include <vector>
#include <cstdlib>
#define MAX 200010
using namespace std;

int wa[MAX],wb[MAX],ws[MAX],wv[MAX];
int r[MAX],sa[MAX];
char str[MAX],s[MAX];
int n,m=0,l,i,ans=0;

int cmp(int *r,int a,int b,int l)
{
        return r[a]==r[b]&&r[a+l]==r[b+l];
}
void Da(int *r,int n,int m)
{
        int i,j,p,*x=wa,*y=wb,*t;
        for(i=0;i<m;i++) ws[i]=0;
        for(i=0;i<n;i++) ws[x[i]=r[i]]++;
        for(i=1;i<m;i++) ws[i]+=ws[i-1];
        for(i=n-1;i>=0;i--) sa[--ws[x[i]]]=i;
        for(j=1,p=1;p<n;j*=2,m=p)
        {
                for(p=0,i=n-j;i<n;i++) y[p++]=i;
                for(i=0;i<n;i++) if(sa[i]>=j) y[p++]=sa[i]-j;
                for(i=0;i<n;i++) wv[i]=x[y[i]];
                for(i=0;i<m;i++) ws[i]=0;
                for(i=0;i<n;i++) ws[wv[i]]++;
                for(i=1;i<m;i++) ws[i]+=ws[i-1];
                for(i=n-1;i>=0;i--) 
                         sa[--ws[wv[i]]]=y[i];
                for(t=x,x=y,y=t,p=1,x[sa[0]]=0,i=1;i<n;i++)
                        x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
        }
        return ;
}
int height[MAX],rank[MAX];
void Callheight(int *r,int *sa,int n)
{
        int i,j,k=0;
        for(i=1;i<=n;i++) rank[sa[i]]=i;
        for(i=0;i<n;height[rank[i++]]=k)
                for(k?k--:0,j=sa[rank[i]-1];r[i+k]==r[j+k];k++);
                return ;
}
int main()
{
    gets(str);
     strcat(s,str);
     strcat(s,str);
     n=strlen(s);
     s[n]=1;
     for (int i=0;i<=n;i++)
     {
         r[i]=s[i];
         if (r[i]>m)
             m=r[i];
     }
     int l=strlen(str);
     n++;
     Da(r,n,m+1);
     for (int i=0;i<n;i++)
         if (sa[i]<l)
             printf("%c",s[sa[i]+l-1]);
     printf("\n");
        return 0;
}