SA后缀数组

需要再多去理解

#include<cstdio> 
#include<cstring>
#include<cmath>
#include<ctime>
#include<iostream>
#include<algorithm>
#include<queue>
#include<set>
#define inf (0x7fffffff)
#define l(a) ((a)<<1)
#define r(a) ((a)<<1|1)
#define b(a) (1<<(a))
#define rep(i,a,b) ;for(int i=a;i<=(b);i++)
#define clr(a) memset(a,0,sizeof(a))
typedef long long ll;
typedef unsigned long long ull;
using namespace std;
int readint(){
    int t=0,f=1;char c=getchar();
    while(!isdigit(c)){
        if(c=='-') f=-1;
        c=getchar();
    }
    while(isdigit(c)){
        t=(t<<3)+(t<<1)+c-'0';
        c=getchar();
    }
    return t*f;
}
const int maxn=2000009;
int n,m=256,mn=maxn,sa1[maxn],a[maxn],sum[maxn],sa[maxn],rank[maxn],height[maxn];//sa与rank意义相反 
char s[maxn];
void cal_SA(){
    clr(sum);
    rep(i,1,n) sum[rank[i]=s[i]]++;//求初始rank
    rep(i,1,m) sum[i]+=sum[i-1];
    for(int i=n;i;i--) sa[sum[rank[i]]--]=i;///基数排序求初始sa 注意这个奇妙的操作! 
    for(int l=1;l<n;l<<=1){
        //printf("rank==");rep(i,1,n) printf("%d ",rank[i]);puts("");
        //printf("sa==");rep(i,1,n) printf("%d ",sa[i]);puts("");
        
        int cnt=0;
        rep(i,n-l+1,n) sa1[++cnt]=i;//后面的第二关键字为最小 
        rep(i,1,n) if(sa[i]>l) sa1[++cnt]=sa[i]-l;//利用sa对第二关键字的排序(sa1即第二关键字的sa) 
        
        //printf("sa1==");rep(i,1,n) printf("%d ",sa1[i]);puts("");
        
        rep(i,1,n) a[i]=rank[sa1[i]];//a为按第二关键字排序后的rank 注意! 
        
        //printf("a==");rep(i,1,n) printf("%d ",a[i]);puts("");
         
        clr(sum);rep(i,1,n) sum[a[i]]++;
        rep(i,1,n) sum[i]+=sum[i-1];
        for(int i=n;i;i--) sa[sum[a[i]]--]=sa1[i];//求出新的sa 
        //综合第一关键字对a排序 和上面类似的奇妙操作(a按第二关键字排序了，所以这样倒着统计即可) 
        
        //printf("sa==");rep(i,1,n) printf("%d ",sa[i]);puts("");
    
        cnt=0;
        rep(i,1,n) a[i]=rank[i];//将rank的值保存出来 
        rep(i,1,n){
            rank[sa[i]]=(a[sa[i-1]]==a[sa[i]]&&a[sa[i-1]+l]==a[sa[i]+l])?cnt:++cnt;
        }//计算出新的rank 此处要利用rank考虑字符串相同的情况 
        
        //printf("rank==");rep(i,1,n) printf("%d ",rank[i]);puts("");
        //puts("");
        
        if(cnt==n) break;//cnt==n即排序完成
    }
    /*rep(i,1,n){
        printf("%d",rank[i]);
        putchar(i==n?'
':' ');
    }*/
    rep(i,1,n){
        printf("%d",sa[i]);
        putchar(i==n?'
':' ');
    }
}
void cal_Height(){//height[i]=suffix(sa[i-1])和suffix(sa[i])的最大公共前缀 
    //suffix(i)和suffix(j)的最大公共前缀为min(height[rank[i]+1],height[rank[i]+2]...height[rank[j]]) 
    int k=0;
    rep(i,1,n){
        if(k) k--;
        int j=sa[rank[i]-1];
        while(s[i+k]==s[j+k]) k++;
        height[rank[i]]=k;
    }//利用h数组(h[i]=height[rank[i]])的性质(h[i]>=h[i-1]-1)(可证)计算height 
    rep(i,2,n){
        printf("%d",height[i]);
        putchar(i==n?'
':' ');
    }
}
void init(){
    scanf("%s",s+1);n=strlen(s+1);
    rep(i,1,n) sum[int(s[i])]=1;
    rep(i,1,m) sum[i]+=sum[i-1];
    rep(i,1,n) s[i]=sum[s[i]-1]+1;//将字符串转变为从1开始的整数 
}
int main(){
    //freopen("#input.txt","r",stdin);
    //freopen("#output.txt","w",stdout);
    init();
    cal_SA();
    cal_Height();
    //fclose(stdin);
    //fclose(stdout);
    return 0;
}