后缀数组

//rk[0~n-1]:1~n sa[1~n]=0~n-1 he[1~n]
#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
const int N=1e5;
int na[N],nb[N],ns[N],nv[N];
int sa[N],he[N],rk[N];

int cmp(int *r,int a,int b,int l){
    return r[a]==r[b]&&r[a+l]==r[b+l];
}
void da(int *r,int n,int m){
    int *x=na,*y=nb;
    for(int i=0;i<m;i++)ns[i]=0;
    for(int i=0;i<n;i++)ns[x[i]=r[i]]++;
    for(int i=1;i<m;i++)ns[i]+=ns[i-1];
    for(int i=n-1;i>=0;i--)sa[--ns[x[i]]]=i;

    for(int j=1,p=1,i; p<n; j*=2,m=p){
        for(p=0,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++)nv[i]=x[y[i]];
        for(i=0;i<m;i++)ns[i]=0;
        for(i=0;i<n;i++)ns[nv[i]]++;
        for(i=1;i<m;i++)ns[i]+=ns[i-1];
        for(i=n-1;i>=0;i--)sa[--ns[nv[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 gethe(int *r,int n){
    for(int i=1;i<=n;i++)rk[sa[i]]=i;
    for(int i=0,k=0,j=0; i<n; he[rk[i++]]=k){
        for(k?k--:0,j=sa[rk[i]-1]; r[i+k]==r[j+k]; k++);
    }
}
void debug(int n){
    for(int i=0;i<=n;i++)printf("%d ",rk[i]);printf("rk\n");
    for(int i=0;i<=n;i++)printf("%d ",sa[i]);printf("sa\n");
    for(int i=0;i<=n;i++)printf("%d ",he[i]);printf("he\n");
}
void solve(int *r,int n){
    r[n]=0;
    da(r,n+1,30);
    gethe(r,n);
    debug(n);
}
char ss[N]="abaa";
int aa[N];
int main(){
    int n=strlen(ss);
    for(int i=0;i<n;i++){
        aa[i]=ss[i]-'a'+1;
    }
    solve(aa,n);
    return 0;
}