SA模板

 #include<cstdio>
 #include<algorithm>
 #include<cstring>
 using namespace std;
 const int M=;
 char S[M];
 int n,i,s[M],sa[M],wa[M],wb[M],wc[M],wd[M],height[M],rank[M];
 bool cmp(int *r,int a,int b,int l){return r[a]==r[b]&&r[a+l]==r[b+l];}
 void getsa(int *r,int *sa,int n,int m){
     int *x=wa,*y=wb,j,p;
     for (i=;i<n;i++)wc[x[i]=r[i]]++;
     for (i=;i<m;i++)wc[i]+=wc[i-];
     for (i=n-;i>=;i--)sa[--wc[x[i]]]=i;
     for (j=,p=;p<n;j*=,m=p){
         p=;
         for (i=n-j;i<n;i++)y[p++]=i;
         for (i=;i<n;i++)if (sa[i]>=j)y[p++]=sa[i]-j;
         for (i=;i<n;i++)wd[i]=x[y[i]];
         for (i=;i<m;i++)wc[i]=;
         for (i=;i<n;i++)wc[wd[i]]++;
         for (i=;i<m;i++)wc[i]+=wc[i-];
         for (i=n-;i>=;i--)sa[--wc[wd[i]]]=y[i];
         swap(x,y);p=;x[sa[]]=;
         for (i=;i<n;i++)x[sa[i]]=cmp(y,sa[i-],sa[i],j)?p-:p++;
     }
 }
 void getheight(int *r,int *sa,int n){
     int i,j,k=;
     for (i=;i<=n;i++)rank[sa[i]]=i;
     for (i=;i<n;height[rank[i++]]=k){
         if(k)k--;
         j=sa[rank[i]-];
         while(r[i+k]==r[j+k])k++;
     }
 }
 int main(){
     scanf("%s",S);
     n=strlen(S);
     for (i=;i<n;i++)s[i]=S[i]-'a'+;
     s[n]=;
     getsa(s,sa,n+,);
     getheight(s,sa,n);
     for (i=;i<=n;i++)printf("%d ",sa[i]+);puts("");
     for (i=;i<=n;i++)printf("%d ",height[i]);puts("");
 }