Suffix Array

struct SuffixArray
{
    int id;ll rk;
    bool operator<(const SuffixArray &b)const
    {
        return rk
void get_sa(T s[],int n,int rk[])
{
    for(int i=1;i<=n;i++)
        sa[i].id=i,sa[i].rk=s[i];
    for(int i=1;;i<<=1)
    {
        sort(sa+1,sa+1+n);
        for(int j=1;j<=n;j++)
        {
            rk[sa[j].id]=rk[sa[j-1].id];
            if(sa[j].rk!=sa[j-1].rk)
                rk[sa[j].id]++;
        }
        if(rk[sa[n].id]==n)break;
        for(int j=1;j<=n;j++)
        {
            sa[j].id=j;
            sa[j].rk=1ll*rk[j]*__;
            if(j+i<=n)
                sa[j].rk+=rk[j+i];
        }
    }
}

//he[i]: 第i名后缀与第i-1名后缀的LCP
template
void get_he(T s[],int n,int rk[],int he[])
{
    for(int i=1,cp=0;i<=n;i++)
    {
        if(cp)--cp;
        int j=sa[rk[i]-1].id,maxx=max(i,j);
        while(maxx+cp<=n && s[i+cp]==s[j+cp])
            cp++;
        he[rk[i]]=cp;//h[i]=cp;
    }
}

你可能感兴趣的:(Suffix Array)