Codevs_P1500 后缀排序(后缀数组+基数排序)

Codevs传送门

时间限制: 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< B；否则令p满足：A1A2…Ap-1=B1B2…Bp-1，Ap<>Bp。如果Ap< Bp，则A< B；否则A>B。

输入描述 Input Description
第一行一个整数n(n<=15000)
第二行是一个长度为n字串。

输出描述 Output Description
输出文件包含n行，第i行是一个整数pi。表示所有的后缀从小到大排序后是Tp1, Tp2, …, Tpn。

样例输入 Sample Input

4
abab

样例输出 Sample Output

3
1
4
2

数据范围及提示 Data Size & Hint
说明：后缀排序后的顺序是T3=”ab”, T1=”abab”, T4=”b”, T2=”bab”。所以输出是3, 1, 4, 2。

Sol:
裸题就不说了…倍增算法…

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define N 15005
int m=26,n,l;int s[N];
int sa[N],t1[N],t2[N],c[N],rk[N];
void DA(){
    register int *x=t1,*y=t2,i,k,p;
    for(i=0;i<m;i++) c[i]=0;
    for(i=0;i<n;i++) c[x[i]=s[i]]++;
    for(i=1;i<m;i++) c[i]+=c[i-1];
    for(i=n-1;~i;i--) sa[--c[x[i]]]=i;
    for(k=1,p=0;k<=n;k<<=1,m=p,p=0){
        for(i=n-k;i<n;i++) y[p++]=i;
        for(i=0;i<n;i++) if(sa[i]>=k) y[p++]=sa[i]-k;
        for(i=0;i<m;i++) c[i]=0;
        for(i=0;i<n;i++) c[x[y[i]]]++;
        for(i=1;i<m;i++) c[i]+=c[i-1];
        for(i=n-1;~i;i--) sa[--c[x[y[i]]]]=y[i];
        swap(x,y),p=1,x[sa[0]]=0;
        for(i=1;i<n;i++) x[sa[i]]=y[sa[i-1]]==y[sa[i]]
            &&y[sa[i-1]+k]==y[sa[i]+k]?p-1:p++;
        if(p>=n) break;
    }
}
int main(){
    scanf("%d",&n);char ch=getchar();
    while(ch>'z'||ch<'a') ch=getchar();
    while(ch>='a'&&ch<='z') s[l++]=ch-'a',ch=getchar();
    DA();for(int i=0;i<n;i++) printf("%d\n",sa[i]+1);
    return 0;
}