[后缀数组] Luogu P3809 后缀排序

题目描述

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

题解

• 板题

代码

#include <cstdio>
#include <cstring>
#include <iostream>
#define N 1000010 
using namespace std;
int n,m=122,t[N],x[N],c[N],sa[N];
char s[N];
void get_SA()
{
    for (int i=1;i<=n;i++) c[x[i]=s[i]]++;
    for (int i=2;i<=m;i++) c[i]+=c[i-1];
    for (int i=n;i>=1;i--) sa[c[x[i]]--]=i;
    for (int k=1;k<=n;k<<=1)
    {
        int p=0;
        for (int i=n-k+1;i<=n;i++) t[++p]=i;
        for (int i=1;i<=n;i++) if (sa[i]>k) t[++p]=sa[i]-k;
        for (int i=1;i<=m;i++) c[i]=0;
        for (int i=1;i<=n;i++) c[x[i]]++;
        for (int i=2;i<=m;i++) c[i]+=c[i-1];
        for (int i=n;i>=1;i--) sa[c[x[t[i]]]--]=t[i],t[i]=0;
        swap(x,t);
        p=1,x[sa[1]]=1;
        for (int i=2;i<=n;i++) x[sa[i]]=(t[sa[i-1]]==t[sa[i]]&&t[sa[i-1]+k]==t[sa[i]+k])?p:++p; 
        if (p==n) break;
        m=p;
    }
    for (int i=1;i<=n;i++) printf("%d ",sa[i]);
}
int main()
{
    scanf("%s",s+1),n=strlen(s+1),get_SA();    
}