后缀数组

http://uoj.ac/problem/35

#include<cstdio>
#include<cstring>
#include<algorithm>
#define maxn 100005
using namespace std;
char s[maxn];
int sa[maxn],t[maxn],t2[maxn],c[maxn],Rank[maxn],height[maxn],n;
void build_sa(int m){
    int i,*x=t,*y=t2,*T,p ;
    n++;
    for(i=0;i<m;++i)c[i]=0;
    for(i=0;i<n;++i)++c[x[i]=s[i]];
    for(i=1;i<m;++i)c[i]+=c[i-1];
    for(i=n-1;i>=0;--i)sa[--c[x[i]]]=i;
    for(int k=1;k<=n;k<<=1)
    {
       p=0;
       for(i=n-1;i>=n-k;--i)y[p++]=i;
       for(i=0;i<n;++i)if(sa[i]>=k)y[p++]=sa[i]-k;
       for(i=0;i<m;++i)c[i]=0;
       for(i=0;i<n;++i)++c[x[y[i]]];
       for(i=1;i<m;++i)c[i]+=c[i-1];
       for(i=n-1;i>=0;--i)sa[--c[x[y[i]]]]=y[i];
       swap(x,y);
       x[sa[0]]=0;p=1;
       for(i=1;i<n;++i)
         x[sa[i]]=y[sa[i-1]]==y[sa[i]]&&y[sa[i-1]+k]==y[sa[i]+k]?p-1:p++;
       if(p>=n)break;
       m=p;
    }
    n--;
    for(int i=1;i<=n;++i)printf("%d ",sa[i]+1);
    printf("\n");

}
void LCP(){
    int i,j,k=0;
    for(i=1;i<=n;i++)Rank[sa[i]]=i;
    for(int i=0;i<n;++i)
    {
       j=sa[Rank[i]-1];//h[i-1]
       if(k)k--;
       while(s[i+k]==s[j+k])k++;
       height[Rank[i]]=k;//h[i]
    }
    for(int i=2;i<=n;++i)printf("%d ",height[i]);
}
int main(){
    scanf("%s",s);
    n=strlen(s);
    build_sa(256);
    LCP();
    return 0;
}

推荐阅读更多精彩内容