【模板】后缀数组模板

这几天在学后缀数组（补题补到绝望决定来学点新东西），等我再学的明白一点，并且有时间心情好的时候（遥遥无期，而且估计写了也没人看），就来写一篇后缀数组的知识点博客~

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>

using namespace std;
const int maxn=10000;
char s[maxn];
int sa[maxn],t[maxn],t1[maxn],c[maxn];
int n;
void build_sa(int m){
    int *x=t,*y=t1;
    for(int i=0;i<m;i++)c[i]=0;
    for(int i=0;i<n;i++)c[x[i]=s[i]]++;
    for(int i=1;i<m;i++)c[i]+=c[i-1];
    for(int i=n-1;i>=0;i--)sa[--c[x[i]]]=i;

    for(int k=1;k<=n;k<<=1){
        int p=0;
        for(int i=n-k;i<n;i++)y[p++]=i;
        for(int i=0;i<n;i++)if(sa[i]>=k)y[p++]=sa[i]-k;
        for(int i=0;i<m;i++)c[i]=0;
        for(int i=0;i<n;i++)c[x[y[i]]]++;//这里为啥是这样？一会结合基数排序想一下
        for(int i=1;i<m;i++)c[i]+=c[i-1];
        for(int i=n-1;i>=0;i--)sa[--c[x[y[i]]]]=y[i];
        swap(x,y);
        p=1,x[sa[0]]=0;
        for(int i=1;i<n;i++)
          x[sa[i]]=y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k]?p-1:p++;
        if(p>=n)break;
        m=p;
    }
}

int m;
int cmp_suffix(char* pattern,int p){
    return strncmp(pattern,s+sa[p],m);
}
int find(char *P){
    m=strlen(P);
    if(cmp_suffix(P,0)<0)return -1;
    if(cmp_suffix(P,n-1)>0)return -1;
    int L=0,R=n-1;
    while(R>=L){
        int M=L+(R-L)/2;
        int res=cmp_suffix(P,M);
        if(!res)return M;
        if(res<0)R=M-1;
        else L=M+1;
    }
}
int rank[maxn],height[maxn];//rank数组是后缀i的排名，height数组是sa[i-1]和sa[i]的最长公共前缀
//后缀j和k的LCP长度等于RMQ（height,rank[j]+1,rank[k]）。
void getHeight(){
    int k=0;
    for(int i=0;i<n;i++)rank[sa[i]]=i;
    for(int i=0;i<n;i++){
        if(k)k--;
        int j=sa[rank[i]-1];
        while(s[i+k]==s[j+k])k++;
        height[rank[i]]=k;
    }
}

int d[maxn][30];
void RMQ_init(){
    for(int i=0;i<n;i++){
        d[i][0]=height[i];
    }
    for(int i=1;(1<<i)<=n;i++){
        for(int j=0;(1<<i)+j<n;j++){
            d[j][i]=min(d[j][i-1],d[j+(1<<(i-1))][i-1]);
        }
    }
}

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

int main(){
    scanf("%s",s);
    n=strlen(s);
    build_sa('a'+27);
    getHeight();
    RMQ_init();
    int j,k;
    while(scanf("%d%d",&j,&k)!=EOF&&(j||k)){
        printf("%d
",LCP(j,k));
    }
return 0;
}