Codevs 1500 后缀排序(后缀数组)

时间：2023-03-09 04:20:19
Codevs 1500 后缀排序(后缀数组)

1500 后缀排序

时间限制: 1 s

空间限制: 128000 KB

题目等级 : 大师 Master

题目描述 Description

天凯是MIT的新生。Prof. HandsomeG给了他一个长度为n的由小写字母构成的字符串，要求他把该字符串的n个后缀(suffix)从小到大排序。

何谓后缀？假设字符串是S=S1S2……Sn，定义Ti=SiSi+1……Sn。T1, T2, …, Tn就叫做S的n个后缀。

关于字符串大小的比较定义如下（比较规则和PASCAL中的定义完全相同，熟悉PASCAL的同学可以跳过此段）：

若A是B的前缀，则A

/*

第一道后缀数组(现在才学。。。

求sa裸题。。

*/

#include<iostream>

#include<cstdio>

#define MAXN 15001

using namespace std;

int n,m=128,s[MAXN],c[MAXN],t1[MAXN],t2[MAXN],sa[MAXN],rank[MAXN];

char ch[MAXN];

bool cmp(int *y,int a,int b,int k)

{

    int a1=y[a],b1=y[b];

    int a2=a+k>=n?-1:y[a+k];

    int b2=b+k>=n?-1:y[b+k];

    return a1==b1&&a2==b2;

}

void slove()

{

    int *x=t1,*y=t2;

    for(int i=0;i<m;i++) c[i]=0;

    for(int i=0;i<n;i++) c[x[i]=s[i]]++;

    for(int i=1;i<m;i++) c[i]+=c[i-1];

    for(int i=n-1;i>=0;i--) sa[--c[x[i]]]=i;

    for(int k=1,p=0;k<=n;k<<=1,m=p,p=0)

    {

        for(int i=n-k;i<n;i++) y[p++]=i;//无第二关键字的优先.

        for(int i=0;i<n;i++) if(sa[i]>=k) y[p++]=sa[i]-k;//第二关键字排序

        for(int i=0;i<m;i++) c[i]=0;

        for(int i=0;i<n;i++) c[x[y[i]]]++;

        for(int i=1;i<m;i++) c[i]+=c[i-1];

        for(int i=n-1;~i;i--) sa[--c[x[y[i]]]]=y[i];//rank与sa互逆.

        swap(x,y);p=1;x[sa[0]]=0;

        for(int i=1;i<n;i++)

        {

            if(cmp(y,sa[i-1],sa[i],k)) x[sa[i]]=p-1;//更新排名.

            else x[sa[i]]=p++;

        }

        if(p>=n) break;

    }

}

int main()

{

    scanf("%d",&n);

    scanf("%s",ch);

    for(int i=0;i<n;i++) s[i]=ch[i];

    slove();

    for(int i=0;i<n;i++) printf("%d\n",sa[i]+1);

    return 0;

}

相关文章