【模板】后缀数组

最新推荐文章于 2016-05-20 22:02:34 发布

yxr0105

最新推荐文章于 2016-05-20 22:02:34 发布

阅读量295

点赞数

分类专栏：模板后缀数组文章标签：模板后缀数组

本文链接：https://blog.csdn.net/yxr0105/article/details/51286123

版权

模板同时被 2 个专栏收录

17 篇文章 0 订阅

订阅专栏

后缀数组

4 篇文章 0 订阅

订阅专栏

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=100010;
int t1[N],t2[N],sa[N],c[N];
int Rank[N],height[N];
char s[N];
int n;
void buildsa(int m){
    int i,*x=t1,*y=t2;
    for(i=1;i<=m;i++) c[i]=0;
    for(i=1;i<=n;i++) c[x[i]=s[i]]++;
    for(i=1;i<=m;i++) c[i]+=c[i-1];
    for(i=n;i>=1;i--) sa[c[x[i]]--]=i;
    for(int k=1;k<=n;k<<=1){
        int p=0;
        for(i=n-k+1;i<=n;i++) y[++p]=i;
        for(i=1;i<=n;i++) if(sa[i]>k) y[++p]=sa[i]-k;
        for(i=1;i<=m;i++) c[i]=0;
        for(i=1;i<=n;i++) c[x[y[i]]]++;
        for(i=1;i<=m;i++) c[i]+=c[i-1];
        for(i=n;i>=1;i--) sa[c[x[y[i]]]--]=y[i];
        swap(x,y);p=1;x[sa[1]]=1;
        for(i=2;i<=n;i++)
            x[sa[i]]=y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k]?p:++p;
        if(p>=n) break;
        m=p;
    }
}
void getheight(){
    int i,j,k=0;
    for(i=1;i<=n;i++) Rank[sa[i]]=i;
    for(i=1;i<=n;i++){
        if(k) k--;
        j=sa[Rank[i]-1]; // sa[rank[i]]=i;
        while(s[j+k]==s[i+k]) k++;
        height[Rank[i]]=k;
    }
}
int main(){
//  freopen("in.txt","r",stdin);
//  freopen("out.txt","w",stdout);
    scanf("%d\n",&n);
    scanf("%s\n",s+1);
    buildsa(128);getheight();
    for(int i=1;i<=n;i++) printf("%d ",sa[i]-1);puts("");
    for(int i=1;i<=n;i++) printf("%d ",Rank[i]);puts("");
    for(int i=2;i<=n;i++) printf("%d ",height[i]);puts("");
    return 0;
}