SA模板

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int M=100010;
char S[M];
int n,i,s[M],sa[M],wa[M],wb[M],wc[M],wd[M],height[M],rank[M];
bool cmp(int *r,int a,int b,int l){return r[a]==r[b]&&r[a+l]==r[b+l];}
void getsa(int *r,int *sa,int n,int m){
    int *x=wa,*y=wb,j,p;
    for (i=0;i<n;i++)wc[x[i]=r[i]]++;
    for (i=1;i<m;i++)wc[i]+=wc[i-1];
    for (i=n-1;i>=0;i--)sa[--wc[x[i]]]=i;
    for (j=1,p=1;p<n;j*=2,m=p){
        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<n;i++)wd[i]=x[y[i]];
        for (i=0;i<m;i++)wc[i]=0;
        for (i=0;i<n;i++)wc[wd[i]]++;
        for (i=1;i<m;i++)wc[i]+=wc[i-1];
        for (i=n-1;i>=0;i--)sa[--wc[wd[i]]]=y[i];
        swap(x,y);p=1;x[sa[0]]=0;
        for (i=1;i<n;i++)x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
    }
}
void getheight(int *r,int *sa,int n){
    int i,j,k=0;
    for (i=1;i<=n;i++)rank[sa[i]]=i;
    for (i=0;i<n;height[rank[i++]]=k){
        if(k)k--;
        j=sa[rank[i]-1];
        while(r[i+k]==r[j+k])k++;
    }
}
int main(){
    scanf("%s",S);
    n=strlen(S);
    for (i=0;i<n;i++)s[i]=S[i]-'a'+1;
    s[n]=0;
    getsa(s,sa,n+1,100);
    getheight(s,sa,n);
    for (i=1;i<=n;i++)printf("%d ",sa[i]+1);puts("");
    for (i=2;i<=n;i++)printf("%d ",height[i]);puts("");
}