首页 > 代码库 > 后缀数组模板

后缀数组模板

后缀有序化:

inline bool equ(int x,int y,int l){return rank[x]==rank[y]&&rank[x+l]==rank[y+l];}void suffix_sort(){    for(int i=1;i<=N;i++){sa[i]=i;rank[i]=s[i];}    for(int i,pos=0,sig=255,l=0;pos<N;sig=pos){        for(pos=0,i=N-l+1;i<=N;i++)p[++pos]=i;        for(i=1;i<=N;i++)if(sa[i]>l)p[++pos]=sa[i]-l;        for(i=1;i<=sig;i++)cnt[i]=0;        for(i=1;i<=N;i++)cnt[rank[i]]++;        for(i=1;i<=sig;i++)cnt[i]+=cnt[i-1];        for(i=N;i;i--)sa[cnt[rank[p[i]]]--]=p[i];        for(pos=0,i=1;i<=N;i++)tmp[sa[i]]=equ(sa[i],sa[i-1],l)?pos:++pos;        for(i=1;i<=N;i++)rank[i]=tmp[i];        l=l==0?1:l<<1;    }}

O(N)求height

1 void get_height(){2     for(int i=1,j=0,k;i<=N;i++){3         if(!(k=sa[rank[i]-1])){j=0;continue;}4         if(j)j--;5         while(s[i+j]==s[k+j])j++;6         height[rank[i]]=j;7     }8 }

 

后缀数组模板