【模板】后缀数组

 注意x,y数组的意义

x[i]表示编号为i的后缀的第一关键字

y[i]表示第二关键字排名为i的后缀编号

h[i]表示编号为i的后缀与其前一名的最大公共前缀

height[i]表示排名为i的后缀与其前一名的最大公共前缀

h[i]>=h[i-1]-1

代码打的比较丑，有时间重新打一遍

#include<bits/stdc++.h>
using namespace std;
const int N=2000010;
int n,s[N];
int m,x[N],y[N],sa[N],cnt[N],rank[N];
int h[N],height[N];
char ss[N];
void build_sa(){
    for(int i=0;i<n;i++) cnt[x[i]]++;
    for(int i=1;i<m;i++) cnt[i]+=cnt[i-1];
    for(int i=n-1;~i;i--) sa[--cnt[x[i]]]=i;
    for(int i=0;i<m;i++) cnt[i]=0;
    for(int k=1;k<=n;k<<=1){
        int num=0;
        for(int i=n-1;i>=n-k;i--) y[num++]=i;
        for(int i=0;i<n;i++) if(sa[i]>=k) y[num++]=sa[i]-k;
        for(int i=0;i<n;i++) cnt[x[i]]++;
        for(int i=1;i<m;i++) cnt[i]+=cnt[i-1];
        for(int i=n-1;~i;i--) sa[--cnt[x[y[i]]]]=y[i];
        for(int i=0;i<m;i++) cnt[i]=0;
        swap(x,y);
        num=0;
        x[sa[0]]=num;
        for(int i=1;i<n;i++){
            if(y[sa[i]]!=y[sa[i-1]]||y[sa[i-1]+k]!=y[sa[i]+k]) num++;
            x[sa[i]]=num;
        }
        num++;
        if(num>=n) break;
        m=num;
    }
    for(int i=0;i<n;i++) rank[sa[i]]=i;
    return;
}
void build_height(){
    int t=0;
    for(int i=0;i<n;i++){
        if(t) t--;
        if(!rank[i]) h[i]=0;
        else{
            while(s[i+t]==s[sa[rank[i]-1]+t]) t++;
            h[i]=t;
        }
        height[rank[i]]=h[i];
    }
    return;
}
int main(){
    scanf("%s",ss);
    n=strlen(ss);
    for(int i=0;i<n;i++){
        if(ss[i]>='0'&&ss[i]<='9') s[i]=ss[i]-'0';
        else if(ss[i]>='A'&&ss[i]<='Z') s[i]=ss[i]-'A'+10;
        else s[i]=ss[i]-'a'+36;
        x[i]=s[i];
    }
    m=62;
    build_sa();build_height();
    for(int i=0;i<n;i++) printf("%d ",sa[i]+1);
    printf("
");
    for(int i=1;i<n;i++) printf("%d ",height[i]);
    return 0;
}