[BZOJ1031][JSOI2007]字符加密Cipher 做题笔记

题目来源：http://www.lydsy.com/JudgeOnline/problem.php?id=1031

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。

后缀数组
- 把原串复制一遍放到原串后面，破环成链。
- 这样每一个后缀都代表了旋转的一种情况。

#include <cstdio>
#include <cstring>
#include <algorithm>
const int maxn=220000;
using namespace std;

char s[maxn];
int p[maxn],r[maxn];
int sa[maxn],rank[maxn];
int wa[maxn],ws[maxn],wv[maxn],wb[maxn];
inline bool 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 *sa,int n,int m) {
    int *x=wa, *y=wb;
    memset(ws,0,sizeof(ws));
    for (int i=0;i<n;i++) ws[x[i]=r[i]]++;
    for (int i=1;i<m;i++) ws[i]+=ws[i-1];
    for (int i=n-1;i>=0;i--) sa[--ws[x[i]]]=i;
    int p=1;
    for (int j=1;p<n;j<<=1,m=p) {
        p=0;
        for (int i=n-j;i<n;i++) y[p++]=i;
        for (int i=0;i<n;i++) if (sa[i]>=j) y[p++]=sa[i]-j;
        memset(ws,0,sizeof(ws));
        for (int i=0;i<n;i++) ws[wv[i]=x[y[i]]]++;
        for (int i=1;i<m;i++) ws[i]+=ws[i-1];
        for (int i=n-1;i>=0;i--) sa[--ws[wv[i]]]=y[i];
        swap(x,y); x[sa[0]]=0; p=1;
        for (int i=1;i<n;i++) 
            x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
    }
}
int main () {
    scanf("%s",s);
    int slen=strlen(s);
    int n=slen*2;
    for (int i=0;i<slen;i++) s[slen+i]=s[i];
    s[n-1]=0;
    for (int i=0;i<n;i++) r[i]=s[i];
    da(r,sa,n,300);
    for (int i=0;i<n;i++) rank[sa[i]]=i;
    for (int i=0;i<slen;i++) p[rank[i]]=s[i+slen-1];
    for (int i=0;i<n;i++) 
        if (p[i]) printf("%c",p[i]);
    return 0;
}