后缀数组(dc3) 时间 O(n) 空间 O(3n)
模板
1 #define F(x) ((x)/3+((x)%3==1?0:tb)) 2 #define G(x) ((x)<tb?(x)*3+1:((x)-tb)*3+2) 3 4 struct SuffixArray{ 5 int sa[maxn*3];//排名为i的后缀的起始下标 6 int Rank[maxn*3];//起始下标为i的排名 7 int height[maxn*3]; 8 int wa[maxn*3],wb[maxn*3],wv[maxn*3],ws[maxn*3]; 9 10 int c0(int *r,int a,int b){ 11 return r[a]==r[b]&&r[a+1]==r[b+1]&&r[a+2]==r[b+2]; 12 } 13 int c12(int k,int *r,int a,int b){ 14 if(k==2) return r[a]<r[b]||r[a]==r[b]&&c12(1,r,a+1,b+1); 15 else return r[a]<r[b]||r[a]==r[b]&&wv[a+1]<wv[b+1]; 16 } 17 18 void Sort(int *r,int *a,int *b,int n,int m){ 19 int i; 20 for(i=0;i<n;i++) wv[i]=r[a[i]]; 21 for(i=0;i<m;i++) ws[i]=0; 22 for(i=0;i<n;i++) ws[wv[i]]++; 23 for(i=1;i<m;i++) ws[i]+=ws[i-1]; 24 for(i=n-1;i>=0;i--) b[--ws[wv[i]]]=a[i]; 25 return; 26 } 27 28 void dc3(int *r,int *sa,int n,int m){ 29 int i,j,*rn=r+n,*san=sa+n,ta=0,tb=(n+1)/3,tbc=0,p; 30 r[n]=r[n+1]=0; 31 for(i=0;i<n;i++) if(i%3!=0) wa[tbc++]=i; 32 Sort(r+2,wa,wb,tbc,m); 33 Sort(r+1,wb,wa,tbc,m); 34 Sort(r,wa,wb,tbc,m); 35 for(p=1,rn[F(wb[0])]=0,i=1;i<tbc;i++) 36 rn[F(wb[i])]=c0(r,wb[i-1],wb[i])?p-1:p++; 37 if(p<tbc) dc3(rn,san,tbc,p); 38 else for(i=0;i<tbc;i++) san[rn[i]]=i; 39 for(i=0;i<tbc;i++) if(san[i]<tb) wb[ta++]=san[i]*3; 40 if(n%3==1) wb[ta++]=n-1; 41 Sort(r,wb,wa,ta,m); 42 for(i=0;i<tbc;i++) wv[wb[i]=G(san[i])]=i; 43 for(i=0,j=0,p=0;i<ta && j<tbc;p++) 44 sa[p]=c12(wb[j]%3,r,wa[i],wb[j])?wa[i++]:wb[j++]; 45 for(;i<ta;p++) sa[p]=wa[i++]; 46 for(;j<tbc;p++) sa[p]=wb[j++]; 47 } 48 49 void cal(int *r,int n){ 50 int i,j,k=0; 51 for(i=1;i<=n;i++) Rank[sa[i]] = i; 52 for(i=0;i<n;i++){ 53 if(k > 0) k--; 54 j=sa[Rank[i]-1]; 55 while(r[i+k] == r[j+k]) k++; 56 height[Rank[i]] = k; 57 } 58 59 } 60 61 /*使用方法 62 cd3(r,SA.sa,n+1,m); 63 cal(r,n);*/ 64 }SA;