后缀数组(模板)

#include <bits/stdc++.h>
using namespace std;
const int N = 1e6+10;
char s[N];
int n,c[N],x[N],y[N],sa[N],m=122;
void getsa(){
    for(int i=1;i<=n;i++) c[x[i]=s[i]]++;//桶排序
    for(int i=2;i<=m;i++) c[i]+=c[i-1];//求小于i的个数 or 排名
    for(int i=n;i>=1;i--) sa[c[x[i]]--]=i;//sa[i]=j,排名为i的位置在j
    for(int k=1;k<=n;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[x[i]]++;
        for(int i=2;i<=m;i++) c[i]+=c[i-1];
        for(int i=n;i>=1;i--) sa[c[x[y[i]]]--]=y[i],y[i]=0;
        //走遍桶排序更新sa
        swap(x,y);
        //仅为了把memset(x,0,sizeof(x)),y更新下
        x[sa[1]]=1;num=1;
        for(int i=2;i<=n;i++)
            x[sa[i]]=(y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k])?num:++num;
        //lcm(i,j)=suff(sa[i]),suff(sa[j])的公共前缀
        //lcm(i,i)=suff(sa[i])=n-sa[i]+1
        //lcm(i,j)=min(lcm(i,i-1),lcm(i-1,j))...可dp
        if(num==n)break;
        m=num;
    }
    for(int i=1;i<=n;i++)
        cout<<sa[i]<<' ';
}
int main()
{
    cin>>s+1;
    n=strlen(s+1);
    getsa();
}