後綴數組自學

sa[i],表示在所有後綴中,排在第i位的後綴是哪個後綴。
rank[i],表示以第i個字母爲結尾的後綴在所有後綴中排第rank[i]個。
height 數組,表示height[i]=(suffix(SA[i-1])和suffix(SA[i])的最長公共前綴),也就是排名相鄰的兩個後綴的最長公共前綴的長度 。

#include<iostream>
#include <cstdio>
#include <iostream>
#include <cstdlib>
#include <algorithm>
#include <cstring>
using namespace std;
const int maxn=200010;

int wa[maxn],wb[maxn],wsf[maxn],wv[maxn];
int rank[maxn],height[maxn],s[maxn],sa[maxn];
void make_sa(int n,int m){
    int i,j,p,*x=wa,*y=wb;
    for(i=0;i<m;i++)  wsf[i]=0;
    for(i=0;i<n;i++)  wsf[x[i]=s[i]]++;
    for(i=1;i<m;i++)  wsf[i]+=wsf[i-1];
    for(i=n-1;i>=0;i--)  sa[--wsf[x[i]]]=i;
    for(p=1,j=1;p<n;j<<=1,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++)  wv[i]=x[y[i]];
        for(i=0;i<m;i++)  wsf[i]=0;
        for(i=0;i<n;i++)  wsf[wv[i]]++;
        for(i=1;i<m;i++)  wsf[i]+=wsf[i-1];
        for(i=n-1;i>=0;i--)  sa[--wsf[wv[i]]]=y[i];
        swap (x,y);
        x[sa[0]]=0;
        for(p=1,i=1;i<n;i++){
            x[sa[i]]=(y[sa[i-1]]==y[sa[i]]&&y[sa[i-1]+j]==y[sa[i]+j]) ? p-1 : p++;
        }
    }
}
void make_height(int n){
    int i,j,k=0;
    for(i=1;i<=n;i++) rank[sa[i]]=i;
    for(i=0;i<n;i++){
        if(k)k--;
        else k=0;
        j=sa[rank[i]-1];
        while(s[i+k]==s[j+k]) k++;
        height[rank[i]]=k;
    }
}
int main (){
    make_sa (srelen (s)+1,s中不同字符種類);
    make_height (strlen (s));
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章