后缀数组模板

读入一个长度为 n 的由大小写英文字母或数字组成的字符串，请把这个字符串的所有非空后缀按字典序从小到大排序，然后按顺序输出后缀的第一个字符在原串中的位置。

位置编号为 1 到 n。

n <= 1e6

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 1e6 + 10;

int sa[N], rk[N], tmp[N], cntA[N], cntB[N], a[N], b[N], n;

char str[N];

int main() {
    scanf("%s", str + 1), n = strlen(str + 1);
    for(int i = 1 ; i <= n ; ++ i) ++ cntA[str[i]];
    for(int i = 1 ; i <= 256 ; ++ i) cntA[i] += cntA[i - 1];
    for(int i = n ; i >= 1 ; -- i) sa[cntA[str[i]] --] = i;
    rk[sa[1]] = 1;
    for(int i = 2 ; i <= n ; ++ i) rk[sa[i]] = rk[sa[i - 1]] + (str[sa[i]] != str[sa[i - 1]]);
    for(int i = 0 ; i <= 256 ; ++ i) cntA[i] = 0;
    for(int k = 1 ; rk[sa[n]] < n ; k <<= 1) {
        for(int i = 1 ; i <= n ; ++ i) cntA[i] = cntB[i] = 0;
        for(int i = 1 ; i <= n ; ++ i)
            ++ cntA[a[i] = rk[i]], ++ cntB[b[i] = i + k > n ? 0 : rk[i + k]];
        for(int i = 1 ; i <= n ; ++ i) cntA[i] += cntA[i - 1], cntB[i] += cntB[i - 1];
        for(int i = n ; i >= 1 ; -- i) tmp[cntB[b[i]] --] = i;
        for(int i = n ; i >= 1 ; -- i) sa[cntA[a[tmp[i]]] --] = tmp[i];
        rk[sa[1]] = 1;
        for(int i = 2 ; i <= n ; ++ i) rk[sa[i]] = rk[sa[i - 1]] + (a[sa[i]] != a[sa[i - 1]] || b[sa[i]] != b[sa[i - 1]]);
    }
    for(int i = 1 ; i <= n ; ++ i) printf("%d ", sa[i]);
}

 

转载于:https://www.cnblogs.com/KingSann/articles/7891610.html