洛谷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 + 5;
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 = 0; i < m; i++)c[i] = 0;
    for(i = 0; i < n; i++)c[x[i] = s[i] - '0']++;
    for(i = 1; i < m; i++)c[i] += c[i - 1];
    for(i = n - 1; i >= 0; i--)sa[--c[x[i]]] = i;
    for(j = 1; j <= n; j <<= 1){
        p = 0;
        for(i = n - j; i < n; i++)y[p++] = i;
        for(i = 0; i < n; i++)if(sa[i] >= j) y[p++] = sa[i] - j;
        for(i = 0; i < m; i++)c[i] = 0;
        for(i = 0; i < n; i++)c[x[y[i]]]++;
        for(i = 1; i < m; i++)c[i] += c[i - 1];
        for(i = n - 1; i >= 0; i--)sa[--c[x[y[i]]]] = y[i];
        swap(x, y);
        p = 1;
        x[sa[0]] = 0;
        for(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++;
        if(p >= n)break;
        m = p;
    }
}

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