后缀数组,最长公共祖先

大白书上的代码有点问题，主要是重复的后缀无法正常排序（如aaaaaaa，排序后是乱序的3456120），因此不利于用二分查找输出所有结果。然后最长公共祖先LCP部分的代码越界访问，导致height[0]不是0。所以自己打了一份代码供自己参考。

代码

#include<bits/stdc++.h>
using namespace std;
const int maxn = 100010;

char s[maxn],ss[maxn];
int sa[maxn],t1[maxn],t2[maxn],c[maxn];
int l;

void get_sa(int m)
{
    int *x=t1,*y=t2;
    l=strlen(s);
    for(int i=0;i<l;i++)
        x[sa[i]=i]=s[i]-'a';
    for(int i=0;i<=l;i=(i?i<<1:i+1))
    {
        int p=0;
        for(int j=l-i;j<l;j++) y[p++]=j;
        for(int j=0;j<l;j++) if(sa[j]>=i) y[p++]=sa[j]-i;
        for(int j=0;j<m;j++) c[j]=0;
        for(int j=l-1;j>=0;j--) c[x[y[j]]]++;
        for(int j=1;j<m;j++) c[j]+=c[j-1];
        for(int j=l-1;j>=0;j--) sa[--c[x[y[j]]]]=y[j];
        swap(x,y);
        p=1,x[sa[0]]=0;
        for(int j=1;j<l;j++)
            x[sa[j]]=y[sa[j-1]]==y[sa[j]]&&(sa[j-1]+i>=l?-1:y[sa[j-1]+i])==(sa[j]+i>=l?-1:y[sa[j]+i])?p-1:p++;
        if(p>=l) return;
        m=p;
    }
}

int m;

int cmp_str(int p,char* ss)
{
    return strncmp(ss,s+sa[p],m);
}

int find(char* ss)
{
    m=strlen(ss);
    int L=0,R=l-1;
    while(L<=R)
    {
        int M=L+(R-L)/2;
        int ans=cmp_str(M,ss);
        if(!ans) return M;
        else if(ans>0) L=M+1;
        else R=M-1;
    }
    return -1;
}

void find_all(char* ss)
{
    m=strlen(ss);
    int L1=0,R1=l;
    while(L1<R1)
    {
        int M=L1+(R1-L1)/2;
        if(cmp_str(M,ss)<=0) R1=M;
        else L1=M+1;
    }
    int L2=0,R2=l;
    while(L2<R2)
    {
        int M=L2+(R2-L2)/2;
        if(cmp_str(M,ss)>=0) L2=M+1;
        else R2=M;
    }
    printf("{");
    for(int i=L1;i<L2;i++)
        printf("%d",sa[i]);
    puts("}");
}

int rank[maxn],height[maxn];

void get_hight()
{
    for(int i=0;i<l;i++) rank[sa[i]]=i;
    int k=0;
    height[0]=0;
    for(int i=0;i<l;i++)
    {
        if(!rank[i]) continue;
        if(k) k--;
        int j=sa[rank[i]-1];
        while(s[i+k]==s[j+k]) k++;
        height[rank[i]]=k;
    }
    printf("height:");
    for(int i=0;i<l;i++)
        printf("%d ",height[i]);
    puts("");
}

const int POW=20;
int st[maxn][POW];

void get_st()
{
    for(int i=0;i<l;i++)
        st[i][0]=height[i];
    for(int i=1;(1<<i)<=l;i++)
        for(int j=0;j+(1<<i)-1<l;j++)
            st[j][i]=min(st[j][i-1],st[j+(1<<(i-1))][i-1]);
}

int rmq(int i,int j)
{
    i=rank[i],j=rank[j];
    if(i>j) swap(i,j);
    i++;
    int k=0;
    while((1<<(k+1))<=j-i+1) k++;
    return min(st[i][k],st[j-(1<<k)+1][k]);
}

int main()
{
    scanf("%s",s);
    l=strlen(s);
    get_sa(26);
    get_hight();
    get_st();


    //输出sa数组。
    /*
    for(int i=0;i<l;i++) printf("%d ",sa[i]);
    puts("");
    */


    //输入模式串，找到一个匹配的后缀
    /*
    while(~scanf("%s",ss)) printf("找到后缀%d\n",find(ss));
    */


    //输入模式串，找出所有匹配的后缀。
    /*
    while(~scanf("%s",ss)) printf("后缀%d\n集合",sa[find(ss)]),find_all(ss);
    */


    //输入后缀i,j,输出LCP长度。
    /*
    int i,j;
    while(~scanf("%d %d",&i,&j)) printf("后缀%d和后缀%d的LCP的长度为:%d\n",i,j,rmq(i,j));
    */


    return 0;
}