【洛谷】P3809 【模板】后缀排序

题目地址：

https://www.luogu.com.cn/problem/P3809

题目背景：
这是一道模板题。

题目描述：
读入一个长度为$n$的由大小写英文字母或数字组成的字符串，请把这个字符串的所有非空后缀按字典序（用ASCII数值比较）从小到大排序，然后按顺序输出后缀的第一个字符在原串中的位置。位置编号为$1$到$n$。

输入格式：
一行一个长度为$n$的仅包含大小写英文字母或数字的字符串。

输出格式：
一行，共$n$个整数，表示答案。

数据范围：
$1\le n\le 10^6$。

参考https://blog.csdn.net/qq_46105170/article/details/119878109。代码如下：

#include <iostream>
#include <cstring>
using namespace std;

const int N = 1e6 + 10;
int n, m;
char s[N];
int sa[N], rk[N], y[N], c[N];

void get_sa() {
  for (int i = 1; i <= n; i++) c[rk[i] = s[i]]++;
  for (int i = 2; i <= m; i++) c[i] += c[i - 1];
  for (int i = n; i; i--) sa[c[rk[i]]--] = i;

  for (int k = 1;; k <<= 1) {
    int num = 0;
    for (int i = n - k + 1; i <= n; 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[rk[i]]++;
    for (int i = 2; i <= m; i++) c[i] += c[i - 1];
    for (int i = n; i; i--) sa[c[rk[y[i]]]--] = y[i];

    swap(rk, y);
    rk[sa[1]] = num = 1;
    for (int i = 2; i <= n; i++)
      rk[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;
  }
}

int main() {
  scanf("%s", s + 1);
  n = strlen(s + 1), m = 'z';
  
  get_sa();
  for (int i = 1; i <= n; i++) printf("%d ", sa[i]);
}

时间复杂度$O(n\log n)$，空间$O(n)$。