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

题目背景

这是一道模板题。

题目描述

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

输入输出格式

输入格式：

一行一个长度为 nn 的仅包含大小写英文字母或数字的字符串。

输出格式：

一行，共n个整数，表示答案。

输入输出样例

输入样例#1： 复制

ababa

输出样例#1： 复制

5 3 1 4 2

说明

n <= 10^6n<=106

后缀数组知识点详解：

https://www.cnblogs.com/wyboooo/p/9854468.html

 #include <iostream>
 #include <set>
 #include <cmath>
 #include <stdio.h>
 #include <cstring>
 #include <algorithm>
 #include <vector>
 #include <queue>
 #include <map>
 using namespace std;
 typedef long long LL;
 #define inf 0x7f7f7f7f

 const int maxn = 1e6 + ;
 char s[maxn];
 int n;
 int sa[maxn];
 int t1[maxn], t2[maxn], c[maxn];
 int rnk[maxn], height[maxn];

 void build_sa(char s[], int n, int m)
 {
     int i, j, p, *x = t1, *y = t2;
     for(i = ; i < m; i++)c[i] = ;
     for(i = ; i < n; i++)c[x[i] = s[i] - '']++;
     for(i = ; i < m; i++)c[i] += c[i - ];
     for(i = n - ; i >= ; i--)sa[--c[x[i]]] = i;
     for(j = ; j <= n; j <<= ){
         p = ;
         for(i = n - j; i < n; i++)y[p++] = i;
         for(i = ; i < n; i++)if(sa[i] >= j) y[p++] = sa[i] - j;
         for(i = ; i < m; i++)c[i] = ;
         for(i = ; i < n; i++)c[x[y[i]]]++;
         for(i = ; i < m; i++)c[i] += c[i - ];
         for(i = n - ; i >= ; i--)sa[--c[x[y[i]]]] = y[i];
         swap(x, y);
         p = ;
         x[sa[]] = ;
         for(i = ; i < n; i++)
             x[sa[i]] = y[sa[i - ]] == y[sa[i]] && y[sa[i - ] + j] == y[sa[i] + j] ? p - :p++;
         if(p >= n)break;
         m = p;
     }
 }

 int main()
 {
     scanf("%s", s);
     n = strlen(s);
     build_sa(s, n, );
     printf("%d", sa[] + );
     for(int i = ; i < n; i++){
         printf(" %d", sa[i] + );
     }
     printf("\n");
     return ;
 }