后缀数组【模板】

这真是一个神奇的算法,并且实在是太难理解了,弄了好长时间都没有找到合适的模板,今天终于遇到了,具体算法流程网上有很多,这里就只贴一下模板吧

#include <cstring>
#include <iostream>
#include <cstdio>
using namespace std;
const int maxn=1e6+10;
int sa[maxn];// sa[i] 表示排名为i的后缀的位置
int rak[maxn];// rak[i]表示第i个位置开始的后缀的排名
int tp[maxn]; //tp[i]表示 基数排序的第二关键字
int tax[maxn];//表示 i号元素出现了多少次,用于辅助基数排序
char s[maxn];
int a[maxn];
int N,M;
void Qsort()
{
    for(int i=0;i<=M;i++) tax[i]=0;
    for(int i=1;i<=N;i++) tax[rak[i]]++;
    for(int i=1;i<=M;i++) tax[i]+=tax[i-1];
    for(int i=N;i>=1;i--) sa[ tax[rak[tp[i]]]--]=tp[i];
}
void Ssort()
{
    M=127;
    for(int i=1;i<=N;i++)
    {
        rak[i]=a[i],tp[i]=i;
    }
    Qsort();
    for(int w=1,p=1;p<N; M = p ,w<<=1)
    {
        p=0;
        for(int i=1;i<=w;i++) tp[++p]=N-w+i;
        for(int i=1;i<=N;i++) if(sa[i]>w) tp[++p]= sa[i]-w;
        Qsort();
        swap(rak,tp);
        rak[sa[1]]=p=1;
        for(int i=2;i<=N;i++)
        {
            rak[sa[i]] =( tp[sa[i-1]] == tp[sa[i]]&&tp[sa[i-1]+w]== tp[sa[i]+w])?p:++p;
        }
    }
}
int main()
{
    scanf("%s",s);
    N=strlen(s);
    for(int i=0;i<N;i++)
    {
        a[i+1]=s[i]-48;
    }
    Ssort();
    for(int i=1;i<=N;i++)
    {
        printf("%d ",sa[i]);
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/becky_w/article/details/80161665