codevs1500 后缀排序

题目描述 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。

正解：后缀数组

解题报告：

　　后缀数组模板题

  

//It is made by jump~
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <ctime>
#include <vector>
#include <queue>
#include <map>
#include <set>
#ifdef WIN32   
#define OT "%I64d"
#else
#define OT "%lld"
#endif
using namespace std;
typedef long long LL;
const int MAXN = 300011;
char ch[MAXN];
int n,m,tot;
int wa[MAXN],wb[MAXN],wv[MAXN],c[MAXN];
int rank[MAXN],height[MAXN];
int sa[MAXN];

inline int getint()
{
       int w=0,q=0;
       char c=getchar();
       while((c<'0' || c>'9') && c!='-') c=getchar();
       if (c=='-')  q=1, c=getchar();
       while (c>='0' && c<='9') w=w*10+c-'0', c=getchar();
       return q ? -w : w;
}

inline void da(int m){
    int i,*x=wa,*y=wb;
    for(i=1;i<=m;i++) c[i]=0;
    for(i=1;i<=n;i++) c[x[i]=(ch[i]-'a'+1)]++;
    for(i=1;i<=m;i++) c[i]+=c[i-1];
    for(i=n;i>=1;i--) sa[c[x[i]]--]=i;
    for(int k=1,p;k<=n;k=k*2) {
    p=0;
    for(i=n-k+1;i<=n;i++) y[++p]=i;
    for(i=1;i<=n;i++) if(sa[i]>k) y[++p]=sa[i]-k;
    for(i=1;i<=m;i++) c[i]=0;
    for(i=1;i<=n;i++) c[x[y[i]]]++;     
    for(i=1;i<=m;i++) c[i]+=c[i-1];
    for(i=n;i>=1;i--) sa[c[x[y[i]]]--]=y[i];
    swap(x,y);  x[sa[1]]=1; p=1;
    for(i=2;i<=n;i++) x[sa[i]]=(y[sa[i-1]]==y[sa[i]] && y[sa[i-1]+k]==y[sa[i]+k])?p:++p;
    if(p==n) break; m=p;
    }
}

inline void calheight(){
    int i,j,k=0;
    for(i=1;i<=n;height[rank[i++]]=k)
    for(k?k--:0 , j=sa[rank[i]-1];ch[i+k]==ch[j+k];k++) ;
}

inline void work(){
    scanf("%s",ch+1);   n=strlen(ch+1);
    da(256);
    for(int i=1;i<=n;i++) rank[sa[i]]=i;
    calheight();
    for(int i=1;i<=n;i++) printf("%d ",sa[i]);
    printf("
");
    for(int i=2;i<=n;i++) printf("%d ",height[i]);
}

int main()
{
  work();
  return 0;
}