BZOJ 1031: [JSOI2007]字符加密Cipher( 后缀数组 )

为什么我的后缀数组跑得这么慢...

把字符串复制一遍放在最后, 然后跑sa, 扫一遍就行了... 

---------------------------------------------------------------------

#include<bits/stdc++.h>

 

using namespace std;

 

const int maxn = 200009;

 

struct HASH {

int id[maxn], N;

HASH() {

N = 0;

}

inline void add(int v) {

id[N++] = v;

}

inline void work(){

sort(id, id + N);

N = unique(id, id + N) - id;

}

inline int hash(int c) {

return lower_bound(id, id + N, c) - id;

}

inline int _hash(int c) {

return id[c];

}

} h;

 

int sa[maxn], rank[maxn], height[maxn], cnt[maxn], N;

char S[maxn];

 

void build_sa() {

int m = h.N, *x = height, *y = rank;

for(int i = 0; i < m; i++) cnt[i] = 0;

for(int i = 0; i < N; i++) cnt[x[i] = S[i]]++;

for(int i = 1; i < m; i++) cnt[i] += cnt[i - 1];

for(int i = N - 1; ~i; i--) sa[--cnt[x[i]]] = i;

for(int k = 1; k <= N; k <<= 1) {

int p = 0;

for(int i = N - k; i < N; i++) y[p++] = i;

for(int i = 0; i < N; i++) if(sa[i] >= k) y[p++] = sa[i] - k;

for(int i = 0; i < m; i++) cnt[i] = 0;

for(int i = 0; i < N; i++) cnt[x[y[i]]]++;

for(int i = 1; i < N; i++) cnt[i] += cnt[i - 1];

for(int i = N - 1; ~i; i--) sa[--cnt[x[y[i]]]] = y[i];

swap(x, y); x[sa[0]] = 0; p = 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] ? p - 1 : p++;

if(p >= N) break;

m = p;

}

}

 

int main() {

scanf("%s", S);

N = strlen(S);

for(int i = 0; i < N; i++)

   h.add(S[i + N] = S[i]);

N <<= 1;

h.add(S[N++] = -1);

h.work();

for(int i = 0; i < N; i++)

   S[i] = h.hash(S[i]);

build_sa();

for(int i = 0; i < N; i++) if(sa[i] < N / 2)

   putchar(h._hash(S[sa[i] + N / 2 - 1]));

return 0;

}

--------------------------------------------------------------------- 

1031: [JSOI2007]字符加密Cipher

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 4199  Solved: 1704
[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

HINT

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

Source