题目描述 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。
正解:后缀数组
解题报告:
后缀数组模板题
1 //It is made by jump~ 2 #include <iostream> 3 #include <cstdlib> 4 #include <cstring> 5 #include <cstdio> 6 #include <cmath> 7 #include <algorithm> 8 #include <ctime> 9 #include <vector> 10 #include <queue> 11 #include <map> 12 #include <set> 13 #ifdef WIN32 14 #define OT "%I64d" 15 #else 16 #define OT "%lld" 17 #endif 18 using namespace std; 19 typedef long long LL; 20 const int MAXN = 300011; 21 char ch[MAXN]; 22 int n,m; 23 int wa[MAXN],wb[MAXN],wv[MAXN],sa[MAXN]; 24 int tong[MAXN]; 25 int rank[MAXN],height[MAXN]; 26 27 inline int getint() 28 { 29 int w=0,q=0; 30 char c=getchar(); 31 while((c<‘0‘ || c>‘9‘) && c!=‘-‘) c=getchar(); 32 if (c==‘-‘) q=1, c=getchar(); 33 while (c>=‘0‘ && c<=‘9‘) w=w*10+c-‘0‘, c=getchar(); 34 return q ? -w : w; 35 } 36 37 inline bool cmp(int *r,int a,int b,int l){ return r[a]==r[b] && r[a+l]==r[b+l]; } 38 39 inline void da(){ 40 int *x=wa,*y=wb; m=256; 41 for(int i=0;i<m;i++) tong[i]=0; 42 for(int i=0;i<n;i++) tong[ x[i] = (ch[i]-‘a‘) ]++; 43 for(int i=1;i<m;i++) tong[i]+=tong[i-1]; 44 for(int i=n-1;i>=0;i--) sa[--tong[x[i]]]=i; 45 int p=0; int j=1; int i; 46 for(;p<n && j<n;j=j*2,m=p) { 47 for(p=0,i=n-j;i<n;i++) y[p++]=i; 48 for(i=0;i<n;i++) if(sa[i]>=j) y[p++]=sa[i]-j; 49 for(int i=0;i<n;i++) wv[i]=x[y[i]]; 50 for(int i=0;i<m;i++) tong[i]=0; 51 for(int i=0;i<n;i++) tong[wv[i]]++; 52 for(int i=1;i<m;i++) tong[i]+=tong[i-1]; 53 for(int i=n-1;i>=0;i--) sa[--tong[wv[i]]]=y[i]; 54 swap(x,y); p=1; 55 for(x[sa[0]]=0,i=1;i<n;i++) { 56 x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++; 57 } 58 } 59 } 60 61 inline void calheight(){ 62 int i,j,k=0; 63 for(i=0;i<n;height[rank[i++]]=k) 64 for(k?k--:0 , j=sa[rank[i]-1];ch[i+k]==ch[j+k];k++) ; 65 } 66 67 inline void work(){ 68 n=getint(); 69 scanf("%s",ch); 70 da(); 71 //for(int i=0;i<n;i++) rank[sa[i]]=i; 72 //calheight(); 73 for(int i=0;i<n;i++) printf("%d\n",sa[i]+1); 74 } 75 76 int main() 77 { 78 work(); 79 return 0; 80 }
时间: 2024-10-12 09:06:58