[UOJ#35] [UOJ后缀数组模板题] 后缀排序 [后缀数组模板]

后缀数组，解决字符串问题的有利工具，本题代码为倍增SA算法

具体解释详见2009年国家集训队论文

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <ctime>

using namespace std;

int    str[110000];
int    Barrel[2][110000],U[110000],Tmp[110000],SA[110000],H[110000];
char    r[110000];

void    Calc_H(const int & n,const int * Rank)
{
    int    i,j,k=0;
    for(i=0;i<n;H[Rank[i++]]=k)
        for(k?k--:0,j=SA[Rank[i]-1];str[i+k]==str[j+k];++k);
    return ;
}

bool    cmp(const int * s,const int a,const int b,const int l)
{
    return s[a]==s[b] && s[a+l]==s[b+l];
}

int*    Get_SA(const int & n,int  m)
{
    int    i,j,p,*x=Barrel[0],*y=Barrel[1];
    memset(U,0,sizeof(U));
    for(i=0;i<n;++i)U[x[i]=str[i]]++;
    for(i=1;i<m;++i)U[i]+=U[i-1];
    for(i=n-1;i>=0;--i)SA[--U[x[i]]]=i;

    for(j=1,p=1;p<n;j<<=1,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)Tmp[i]=x[y[i]];
        for(i=0;i<m;++i)U[i]=0;
        for(i=0;i<n;++i)U[Tmp[i]]++;
        for(i=1;i<m;++i)U[i]+=U[i-1];
        for(i=n-1;i>=0;--i)SA[--U[Tmp[i]]]=y[i];
        for(swap(x,y),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++;
    }
    Calc_H(n,x);
    return x;
}

int main()
{
    int    i,n;
    scanf("%s",r);
    n=strlen(r);
    for(i=0;i<n;++i)str[i]=r[i];

    Get_SA(n+1,256);

    for(i=1;i<=n;++i)printf("%d ",SA[i]+1);
    printf("
");
    for(i=2;i<=n;++i)printf("%d ",H[i]);
    
    return 0;
}

在计算H数组的时候因为在计算SA数组的程序中x所代表的数组中就是Rank数组，所以不需要重新计算