LUOGU 3089 后缀排序（模板）

传送门

解题思路

这是一个神奇的算法，sa[i]表示排名第i为的元素是啥，rk[i]表示第i个元素排名是啥。然后使用基数排序+倍增的思想去处理。主要是参考的这位大佬的博客(https://www.cnblogs.com/victorique/p/8480093.html#autoid-1-3-4)

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>

using namespace std;
const int MAXN = 1e6+5;

int sa[MAXN],x[MAXN],y[MAXN];
int num,c[MAXN],n,m;
int rk[MAXN],height[MAXN];
char s[MAXN];

inline void Get_SA(){
    for(register int i=2;i<=m;i++) c[i]+=c[i-1];
    for(register int i=n;i;i--) sa[c[x[i]]--]=i;
    for(register int k=1;k<=n;k<<=1){
        num=0;
        for(register int i=n-k+1;i<=n;i++) y[++num]=i;
        for(register int i=1;i<=n;i++)
            if(sa[i]>k) y[++num]=sa[i]-k;
        for(register int i=2;i<=m;i++) c[i]=0;
        for(register int i=1;i<=n;i++) c[x[i]]++;
        for(register int i=2;i<=m;i++) c[i]+=c[i-1];
        for(register int i=n;i;i--)
            sa[c[x[y[i]]]--]=y[i],y[i]=0;
//      for(register int i=1;i<=n;i++)
//          cout<<sa[i]<<" ";
//      cout<<endl;
        swap(x,y);num=1;x[sa[1]]=1;
        for(register int i=2;i<=n;i++){
            if(y[sa[i]]==y[sa[i-1]] && y[sa[i]+k]==y[sa[i-1]+k])
                x[sa[i]]=num;
            else x[sa[i]]=++num;
        }
        if(num==n) return;
        m=num;
    }
}

inline void Get_height(){
    for(register int i=1;i<=n;i++) rk[sa[i]]=i;
//  for(register int i=1;i<=n;i++) cout<<rk[i]<<" ";
    int k=0,j=0;
    for(register int i=1;i<=n;i++){
        if(rk[i]==1) continue;
        if(k) k--;
        j=sa[rk[i]-1];
        while(j+k<=n && i+k<=n && s[i+k]==s[j+k]) k++;
        height[rk[i]]=k;
    }
    for(register int i=1;i<=n;i++)
        cout<<height[i]<<" ";
}

int main(){
    scanf("%s",s+1);
    n=strlen(s+1);
    m=(int)'z';
    for(register int i=1;i<=n;i++){
        x[i]=s[i];
        c[x[i]]++;
    }
    Get_SA();
    for(register int i=1;i<=n;i++)
        printf("%d ",sa[i]);
//  Get_height();
    return 0;
}