后缀排序

Posted 2021-01-07 liguanlin1124

后缀数组板子，给出一个字符串，要求输出sa数组。

首先声明一下各数组的作用。

rank:序列上的点->点权;

sa:排名->位置;

tmp:辅助数组;

#include<cstdio>
#include<cstring>
#define N 1000050
int rank[N],tmp[N],sa[N],hs[N],num[N];
int n;
char s[N];
bool cmp(int i,int j,int k)
{
    if(i+k>n||j+k>n)return 0;
    return rank[i]==rank[j]&&rank[i+k]==rank[j+k];
}
int main()
{
    scanf("%s",s+1);
    n = strlen(s+1);
    int i,cnt=0;
    for(i=1;i<=n;i++)num[i]=s[i];
    for(i=1;i<=n;i++)hs[num[i]]++;
    for(i=1;i<=200;i++)if(hs[i])tmp[i]=++cnt;
    for(i=1;i<=200;i++)hs[i]+=hs[i-1];
    for(i=1;i<=n;i++)rank[i]=tmp[num[i]],sa[hs[num[i]]--]=i;
    for(int k=1;cnt!=n;k<<=1)
    {
        for(i=1;i<=n;i++)hs[i]=0;
        
        for(i=1;i<=n;i++)hs[rank[i]]++;
        for(i=1;i<=n;i++)hs[i]+=hs[i-1];
        
        for(i=n;i>=1;i--)if(sa[i]>k)tmp[sa[i]-k]=hs[rank[sa[i]-k]]--;
        for(i=1;i<=k;i++)tmp[n-i+1]=hs[rank[n-i+1]]--;
        
        for(i=1;i<=n;i++)sa[tmp[i]]=i;
        
        for(i=1,cnt=0;i<=n;i++)tmp[sa[i]]=cmp(sa[i],sa[i-1],k)?cnt:++cnt;
        for(i=1;i<=n;i++)rank[i]=tmp[i];
    }
    for(i=1;i<=n;i++)printf("%d ",sa[i]);
    printf("
");
    return 0;
}