【UOJ #35】后缀排序 后缀数组模板

http://uoj.ac/problem/35
以前做后缀数组的题直接粘模板。。。现在重新写一下模板
注意用来基数排序的数组一定要开到N。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 100003;

int t1[N], t2[N], c[N];

void st(int *x, int *y, int *sa, int n, int m) {
    for (int i = 0; i <= m; ++i) c[i] = 0;
    for (int i = 0; i < n; ++i) ++c[x[y[i]]];
    for (int i = 1; i <= m; ++i) c[i] += c[i - 1];
    for (int i = n - 1; i >= 0; --i) sa[--c[x[y[i]]]] = y[i];
}

void mkhz(int *r, int *sa, int n, int m) {
    int *x, *y, *t, i, j, p;
    x = t1; y = t2;
    for (i = 0; i < n; ++i) x[i] = r[i], y[i] = i;
    st(x, y, sa, n, m);
    for (p = 1, j = 1; j < n && p < n; j <<= 1, m = p - 1) {
        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;
        st(x, y, sa, n, m);
        for (t = x, x = y, y = t, p = 1, x[sa[0]] = 0, i = 1; i < n; ++i)
            x[sa[i]] = y[sa[i - 1]] == y[sa[i]] && y[sa[i - 1] + j] == y[sa[i] + j] ? p - 1 : p++;
    }
}

void mkh(int *r, int *sa, int *rank, int *h, int n) {
    h[1] = 0; int k = 0, j;
    for (int i = 1; i < n; ++i) rank[sa[i]] = i;
    for (int i = 1; i < n; h[rank[i++]] = k)
        for (k ? --k : k = 0, j = sa[rank[i] - 1]; r[j + k] == r[i + k]; ++k);
}

char s[N];
int r[N], sa[N], h[N], rank[N];

int main() {
    scanf("%s", s + 1);
    int len = strlen(s + 1);
    for (int i = 1; i <= len; ++i) r[i] = s[i] - 'a' + 1;
    mkhz(r, sa, len + 1, 26);
    for (int i = 1; i <= len; ++i) printf("%d ", sa[i]);
    puts("");
    
    mkh(r, sa, rank, h, len + 1);
    for (int i = 2; i <= len; ++i) printf("%d ", h[i]);
    puts("");
    return 0;
}