2015年5月初次学习后缀数组, 当时是 error202 讲的. 代码一直不熟练, 导致运用得不熟练, 2016年 ACM-ICPC China Finals 就有一道后缀数组的裸题, 我们队并没有过. 现在回想起来, 真是荒废了大把时间, 悔之晚矣.

Implementation

后缀数组原理比较好理解, 实现起来也有比较简单的写法. 下面的实现参考了 2009 年国家集训队队员罗穗骞的论文《后缀数组--处理字符串的有力工具》中的实现. 论文中给出的实现比较巧妙, 但比较难理解, 需要读者懂得计数排序（counting sort）和基数排序（radix sort）作为预备.

//比较两个串是否相同
//分别比较两个关键字
bool same(int *rank, int idx1, int idx2, int len){return rank[idx1]==rank[idx2] && rank[idx1+len]==rank[idx2+len];
}// 输入字符串的末尾要补一个 '0' (C-style string 默认如此), n是字符串的实际长度+1.
void SA(int *s, int *sa, int *sa2, int *rk, int *cnt, int *hgt, int n, int m){//counting sortfor(int i=0; i<m; i++) cnt[i]=0;for(int i=0; i<n; i++) cnt[rk[i]=s[i]]++;for(int i=1; i<m; i++) cnt[i]+=cnt[i-1];for(int i=n-1; i>=0; i--) sa[--cnt[rk[i]]]=i;　　//stable sortfor(int len=1; len<n; len*=2){//step-1: 计算（填充） sa2[]//sa2[]: 按第二关键字对后缀排序的结果//对后缀按第二关键字排序这一步可直接用上一次的sa[]数组, 这个过程可形象地理解为将后缀按顺序从一个数组中拿出来放到另一个数组中int p=0;for(int i=n-len; i<n; i++) sa2[p++]=i;   //第二关键字为空的后缀放在最开头//接着放第二关键字非空的//形象化地理解: 将后缀从一个数组里按顺序拿出来放到另一个数组里for(int i=0; i<n; i++) if(sa[i]>=len) sa2[p++]=sa[i]-len;//step-2 fill sa[], countig sort//计数排序: 每个后缀的第一关键字 (first key) 为上一次针对长度减半的 (部分) 后缀求出来的rankfor(int i=0; i<m; i++) cnt[i]=0;for(int i=0; i<n; i++) cnt[rk[i]]++;for(int i=1; i<m; i++) cnt[i]+=cnt[i-1];for(int i=n-1; i>=0; i--) sa[--cnt[rk[sa2[i]]]]=sa2[i];//step-1 and step-2 together constitute a radix sort//fill rk[]swap(rk, sa2);rk[sa[0]]=0;for(int i=1; i<n; i++)//这里, sa2指向的是旧rank数组rk[sa[i]]=rk[sa[i-1]]+!same(sa2, sa[i-1], sa[i], len);m=rk[sa[n-1]]+1;if(m==n) break;}//CALCULATE hgt[]for(int i=0, j, lcp=0; i<n-1; i++){lcp?--lcp:0;// rk[i]>0j=sa[rk[i]-1];    for(; s[j+lcp]==s[i+lcp]; lcp++);hgt[rk[i]]=lcp;}
}

实现要点

void SA(int *s, int *sa, int *sa2, int *rk, int *cnt, int *hgt, int n, int m)

s: 输入字符串, 末尾补一个特殊字符 \0.
sa: 后缀数组
sa2: 以第二关键字对后缀排序所得数组, 辅助数组
cnt: 用于计数排序的辅助数组
rk: rank数组, 辅助数组
n: 输入字符串 s 的长度, 为其实际长度+1
m: 单字符rank的范围, 辅助变量.

在输入字符串 s 末尾补 0 的好处

罗穗骞论文中概括为

为了函数操作的方便

具体为

bool same(int *rank, int idx1, int idx2, int len){return rank[idx1]==rank[idx2] && rank[idx1+len]==rank[idx2+len];
}

从same()的实现上, 可以看出规定 s[n-1]=0 的好处, 如果 s[idx1]==s[idx2] (注意 idx1 != idx2), 说明以 idx1 或 idx2 开头的长度为 len 的字符串肯定不包括字符 s[n-1] , 所以调用变量 rank[idx1+len] 和 rank[idx2+len] 不会导致数组越界, 这样就不需要做特殊判断.

另外，在求 hgt[] 时,

    for(int i=0, j, lcp=0; i<n-1; i++){lcp?--lcp:0;// rk[i]>0j=sa[rk[i]-1];    for(; s[j+lcp]==s[i+lcp]; lcp++);hgt[rk[i]]=lcp;}

由于输入数组末尾补零的缘故, 有 $\mathrm{rk}[n-1]=0$ , $\mathrm{rk}[i]>0, (0\le i < n-1)$ .
从而 j=sa[rk[i]-1]; 不会越界.

hight 数组的求法

height[i]: sa[i] 和 sa[i-1] 两后缀的最长公共前缀 (LCP) 的长度.
height[i] 满足如下性质:
$$\mathrm{height}[\mathrm{rank}[i]] \ge \mathrm{height}[\mathrm{rank}[i-1]]-1$$
依据这个性质, 我们可以按 $\mathrm{height}[\mathrm{rank}[0]], \mathrm{height}[\mathrm{rank}[1]], \dots, \mathrm{height}[\mathrm{rank}[n-1]]$ 的顺序计算 $\mathrm{height}[]$ 数组, 复杂度为 $O(n)$ .

注意: 上面的实现是在 SA() 内部计算的 hgt[] 的. 计算 hgt[] 需借助 rank[] , 若要在 SA() 外面计算 hgt[] , 则须根据求出的 sa[] 再算一遍 rk[] . rk 是作为指针传入 SA() 的, 然而 SA() 中有 swap(rk, sa2); 这一条语句, 因此在 SA() 外部并不能断定 rk[] 里存的是什么.