[模板] 后缀数组 C++

 1 #include<cstdio>
 2 #include<cstring>
 3 #include<iostream>
 4 #define FOR(a,b,c) for(int a=(b);a<=(c);a++)
 5 using namespace std;
 6 
 7 const int maxn =             //长度范围 
 8 const int maxm =             //字符范围 
 9 
10 int s[maxn];                //字符串对应Ascii码 使用前可提前构造 
11 int sa[maxn],c[maxm],t[maxn],t2[maxn];
12 
13 //m代表字符最大值 n代表字符串长
14 //常常需要在字符串末尾加0以防止RE 令s[n]=0 调用build_sa(m,n+1)即可 
15 void build_sa(int m,int n) {
16     int i,*x=t,*y=t2;
17     for(i=0;i<m;i++) c[i]=0;
18     for(i=0;i<n;i++) c[x[i]=s[i]]++;
19     for(i=1;i<m;i++) c[i]+=c[i-1];
20     for(i=n-1;i>=0;i--) sa[--c[x[i]]]=i;
21     
22     for(int k=1;k<=n;k<<=1) {
23         int p=0;
24         for(i=n-k;i<n;i++) y[p++]=i;
25         for(i=0;i<n;i++) if(sa[i]>=k) y[p++]=sa[i]-k;
26         
27         for(i=0;i<m;i++) c[i]=0;
28         for(i=0;i<n;i++) c[x[y[i]]]++;
29         for(i=0;i<m;i++) c[i]+=c[i-1];
30         for(i=n-1;i>=0;i--) sa[--c[x[y[i]]]]=y[i];
31         
32         swap(x,y);
33         p=1; x[sa[0]]=0;
34         for(i=1;i<n;i++) 
35             x[sa[i]]=y[sa[i]]==y[sa[i-1]] && y[sa[i]+k]==y[sa[i-1]+k]?p-1:p++;
36         if(p>=n) break;
37         m=p;
38     }
39 }
40 //n代表字符串长度
41 //调用getHeight(n)即可 
42 int rank[maxn],height[maxn];
43 void getHeight(int n) {
44     int i,j,k=0;
45     for(i=0;i<=n;i++) rank[sa[i]]=i;  //[0,n]
46     for(i=0;i<n;i++) {                  //不计算height[n] 即末尾0 
47         if(k) k--;
48         j=sa[rank[i]-1];
49         while(s[j+k]==s[i+k]) k++;
50         height[rank[i]]=k;
51     }
52 }