真-后缀数组模板

#include<cstdio>

#include<cstring>

#include<algorithm>

#define maxn 10000000//往大的开也不会死的

using namespace std;

int wa[maxn],wb[maxn],wv[maxn],ws[maxn];

int rank[maxn],height[maxn];

int r[maxn],sa[maxn],n;

int cmp(int *r,int a,int b,int l)

{return r[a]==r[b]&&r[a+l]==r[b+l];} 

void da(int *r,int *sa,int n,int m)

{

    int i,j,p,*x=wa,*y=wb,*t;

    for(i=0;i<m;i++) ws[i]=0;

    for(i=0;i<n;i++) ws[x[i]=r[i]]++; 

    for(i=1;i<m;i++) ws[i]+=ws[i-1];

    for(i=n-1;i>=0;i--) sa[--ws[x[i]]]=i;  

    for(j=1,p=1;p<n;j*=2,m=p)

    {

        for(p=0,i=n-j;i<n;i++) y[p++]=i;  

        for(i=0;i<n;i++) if(sa[i]>=j) y[p++]=sa[i]-j;  

        for(i=0;i<n;i++) wv[i]=x[y[i]];  //这里已经不知道写错了多少遍了

        for(i=0;i<m;i++) ws[i]=0;

        for(i=0;i<n;i++) ws[wv[i]]++;

        for(i=1;i<m;i++) ws[i]+=ws[i-1];

        for(i=n-1;i>=0;i--) sa[--ws[wv[i]]]=y[i];

        for(t=x,x=y,y=t,p=1,x[sa[0]]=0,i=1;i<n;i++)

        x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++; 

    }

    return;

 }

void calheight(int *r,int *sa,int n)

{

    int i,j,k=0;

    for(i=1;i<=n;i++) rank[sa[i]]=i; 

    for(i=0;i<n;height[rank[i++]]=k) 

    for(k?k--:0,j=sa[rank[i]-1];r[i+k]==r[j+k];k++); 

    return;

}

int main()

{

char rr[1000000];

scanf("%s",rr);

n=strlen(rr);

for(int i=0;i<n;i++)r[i]=rr[i]-'a'+1;

da(r,sa,n+1,128);//相当于传了strlen()+1过去，这里非常迷，不知道为什么要传strlen（）+1过去？？！

calheight(r,sa,n);//这里有一个非常坑的地方这里是n而不是n+1啊，我已经调了一个下午了QAQ，

for(int i=1;i<=n;i++)printf("%d ",sa[i]+1);

printf("\n");

for(int i=2;i<=n;i++)printf("%d ",height[i]);

}