1. 程式人生 > >【洛谷 P4051】 [JSOI2007]字符加密(後綴數組)

【洛谷 P4051】 [JSOI2007]字符加密(後綴數組)

num ons 鏈接 ring tdi href [1] show 。。

題目鏈接
兩眼題。。
第一眼裸SA
第二眼要復制一倍再跑SA。
一遍過。。

#include <cstdio>
#include <cstring> 
#include <algorithm>
using namespace std;
const int MAXN = 1000010;
int sa[MAXN], x[MAXN], c[MAXN], y[MAXN], n, m = 122, q[MAXN], out[MAXN];
char s[MAXN];
int cmp(const int a, const int b){
    return x[a] < x[b];
}
int main(){
    scanf("%s", s + 1);
    n = strlen(s + 1);
    for(int i = 1; i <= n; ++i) s[i + n] = s[i];
    n <<= 1;
    for(int i = 1; i <= n; ++i) ++c[x[i] = s[i]];
    for(int i = 2; i <= m; ++i) c[i] += c[i - 1];
    for(int i = 1; i <= n; ++i)     sa[c[x[i]]--] = i;
    for(int k = 1; k <= n; k <<= 1){
       int num = 0;
       for(int i = n; i >= n - k + 1; --i) y[++num] = i;
       for(int i = 1; i <= n; ++i) if(sa[i] > k) y[++num] = sa[i] - k;
       for(int i = 1; i <= m; ++i) c[i] = 0;
       for(int i = 1; i <= n; ++i) ++c[x[i]];
       for(int i = 2; i <= m; ++i) c[i] += c[i - 1];
       for(int i = n; i; --i) sa[c[x[y[i]]]--] = y[i]; 
       memcpy(y, x, sizeof x);
       x[sa[1]] = 1; num = 1;
       for(int i = 2; i <= n; ++i)
          x[sa[i]] = (y[sa[i]] == y[sa[i - 1]] && y[sa[i] + k] == y[sa[i - 1] + k]) ? num : ++num;
        if(num == n) break;
        m = num;
    }
    n >>= 1;
    for(int i = 1; i <= n; ++i) q[i] = i;
    sort(q + 1, q + n + 1, cmp);
    for(int i = 1; i <= n; ++i)
       printf("%c", s[q[i] + n - 1]);
    return 0;
}

【洛谷 P4051】 [JSOI2007]字符加密(後綴數組)