后缀数组模板 by hzwer

这是来自hzwerorzorz的模板

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <cmath>
using namespace std;
namespace Suffix_Array{//a数组为字符串s离散之后的int类型，以1开头,n为字符串长度 
    #define Suffix_Maxn 300005
    char s[Suffix_Maxn];
    int a[Suffix_Maxn]={0},v[Suffix_Maxn]={0},rk[2][Suffix_Maxn]={0},sa[2][Suffix_Maxn]={0},height[Suffix_Maxn]={0};
    int Sa[Suffix_Maxn],Rank[Suffix_Maxn];
    int n,p=0,q=1,k;
    void Mul(int *rk,int *sa,int *RK,int *SA){
        for(int i=1;i<=n;i++)v[rk[sa[i]]]=i;
        for(int i=n;i>=1;i--)if(sa[i]>k)SA[v[rk[sa[i]-k]]--]=sa[i]-k;
        for(int i=n-k+1;i<=n;i++)SA[v[rk[i]]--]=i;
        for(int i=1;i<=n;i++)RK[SA[i]]=RK[SA[i-1]]+(rk[SA[i]]!=rk[SA[i-1]]||rk[SA[i]+k]!=rk[SA[i-1]+k]);
    }
    void Suffix_Init(int m){
        for(int i=1;i<=n;i++)v[a[i]]++;
        for(int i=1;i<=m;i++)v[i]+=v[i-1];
        for(int i=n;i>=1;i--)sa[p][v[a[i]]--]=i;
        for(int i=1;i<=n;i++)rk[p][sa[p][i]]=rk[p][sa[p][i-1]]+(a[sa[p][i-1]]!=a[sa[p][i]]);
        for(k=1;k<n;k<<=1,swap(p,q))Mul(rk[p],sa[p],rk[q],sa[q]);
        for(int i=1,k=0;i<=n;i++){
            int j=sa[p][rk[p][i]-1];
            while(a[i+k]==a[j+k])k++;
            height[rk[p][i]]=k;
            if(k>0)k--;
        }
        memcpy(Sa,sa[p],sizeof(sa[p]));
        memcpy(Rank,rk[p],sizeof(rk[p]));
    }
}
using namespace Suffix_Array;
int main(){
    return 0;
}

转载于:https://www.cnblogs.com/Cedric341561/p/6811035.html