sa模板

#include<cstdio>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
#define maxn 1000006
using namespace std;
int n,m,tax[maxn],rk[maxn],sa[maxn],tp[maxn],H[maxn],hei[maxn];
int st[maxn][22];
char ch[maxn];
void Sort(){
    for(int i=1;i<=m;i++)tax[i]=0;
    for(int i=1;i<=n;i++)tax[rk[tp[i]]]++;
    for(int i=1;i<=m;i++)tax[i]+=tax[i-1];
    for(int i=n;i>=1;i--)sa[tax[rk[tp[i]]]--]=tp[i];
}
void SA_Sort(){
    m=26;
    for(int i=1;i<=n;i++)rk[i]=ch[i]-'a'+1,tp[i]=i;
    Sort();
    for(int w=1,p;p<n;m=p,w<<=1){
        p=0;
        for(int i=1;i<=w;i++)tp[++p]=n-w+i;
        for(int i=1;i<=n;i++)if(sa[i]>w)tp[++p]=sa[i]-w;
        Sort();swap(rk,tp);
        rk[sa[1]]=p=1;
        for(int i=2;i<=n;i++){
            rk[sa[i]]=(tp[sa[i-1]]==tp[sa[i]]&&tp[sa[i-1]+w]==tp[sa[i]+w])?p:++p;
        }
    }
    int k=0;
    for(int i=1;i<=n;i++){
        if(k)k--;
        int j=sa[rk[i]-1];
        for(;ch[j+k]==ch[i+k];k++);
        hei[rk[i]]=k;
        st[rk[i]][0]=k;
    }
    for(int j=1;j<=20;j++){
        for(int i=1;i<=n;i++)st[i][j]=min(st[i][j-1],st[i+(1<<(j-1))][j-1]);
    }
}
int main(){
    scanf("%s",ch+1);n=strlen(ch+1);
    SA_Sort();
    return 0;
}

 

posted @ 2020-01-29 10:12  liankewei123456  阅读(173)  评论(0编辑  收藏  举报