后缀数组（SA）学习笔记

后缀数组是一个很强的字符串算法，可以解决众多有关子串的问题。

定义

已知一个字符串 $S$，那么定义 $S_i$ 表示 $i...n$ 形成的后缀。

$sa_i$ 表示把这些后缀按字典序排序后，排名第 $i$ 的串的起始下标。

$rk_i$ 表示把排序后 $S_i$ 的排名，可以发现 $rk_{sa_i}=i$。

$\text{lcp}(S,T)$ 表示 $S,T$ 的最长公共前缀长度。

$h_i$ 表示 $\text{lcp}(S_{sa_i},S_{sa_{i-1}})$，也就是排序后相邻两个串的 $\text{lcp}$。

求法

• $O(n^2\log n)$ 做法

直接暴力快拍，不讲。

• $O(n\log n)$ 做法。

这里要用倍增的思想，考虑枚举一个 $w$，然后把 $S_i$ 重新定义为 $[i,i+w-1]$ 构成的字符串，那么每次 $w$ 乘 $2$，比较两个字符串时只需要拆成两个字符串比较即可。

给个经典图：

直接快排就是 $O(n\log^2n)$ 的了。

考虑优化，这个东西本质上是一个双关键字排序，因此可以直接采用基数排序。

先第二关键词排序，然后第一关键字排序即可，注意需要用稳定排序，复杂度 $O(n\log n)$

const int N=1e6+5;
struct SA{
	void get_sa(int n,int m){//长度为 n，值域为 [1,m]
		m=max(n,m);for(int i=1;i<=n;i++)cnt[rk[i]=a[i]]++;//开始时排名为a[i]，开桶直接统计出现多少次
		for(int i=1;i<=m;i++)cnt[i]+=cnt[i-1];
		for(int i=n;i;i--)sa[cnt[rk[i]]--]=i;//从大到小做，相对顺序不变
		for(int w=1;w<n;w*=2){//倍增枚举长度
			memset(cnt,0,sizeof(cnt));int t=0;
			for(int i=n;i>n-w;i--)id[++t]=i;//这里以第二关键字排序，也就是按 rk[i+w] 排序，排序结果是 id[i]
			for(int i=1;i<=n;i++)if(sa[i]>w)id[++t]=sa[i]-w;
			for(int i=1;i<=n;i++)cnt[rk[i]]++;//然后按第一关键字排序，注意要使用 id[i] 保持顺序
			for(int i=1;i<=m;i++)cnt[i]+=cnt[i-1];
			for(int i=n;i;i--)sa[cnt[rk[id[i]]]--]=id[i];
			t=0;memcpy(tmp,rk,sizeof(rk));
			for(int i=1;i<=n;i++)if(tmp[sa[i]]==tmp[sa[i-1]]&&tmp[sa[i]+w]==tmp[sa[i-1]+w])rk[sa[i]]=t;else rk[sa[i]]=++t;//相同的 rk 取值一样
		}
	}
	int a[N],cnt[N],tmp[N],rk[N],sa[N],id[N];
};

然后考虑 $h$ 怎么求。

首先是一个引理：

$$h[rk[i-1]]-1\le h[rk[i]]$$

这个证明就是考虑 $S_{i-1}$ 和 它匹配的串，假设 $\text{lcp}$ 为 $aA$，这里 $a$ 是一个字母，$A$ 是一个字符串，也就是 $S_{i-1}=aAB$ 这样的形式，与其匹配的串就是 $aAC$ 的形式。

那么考虑 $S_i=AB$，那么与 $S_{i-1}$ 匹配的也至少会有一个 $AC$ 的后缀，显然这个也会排在 $S_i$ 的前面，又由于是按字典序排序，因此至少也会有一个 $A$ 的 $\text{lcp}$，上面的得证。

那么求法也很简单，直接枚举 $i$，转移前面的答案，然后暴力向后看是否能扩展，复杂度容易证明为 $O(n)$。

for(int i=1,t=0;i<=n;i++){
	if(t)t--;
	while(a[i+t]==a[sa[rk[i]-1]+t])t++;
	h[rk[i]]=t;
}

应用

• 查询一个串作为子串出现的位置

既然是子串，那么一定作为一个后缀的前缀出现，又因为已经按照字典序排序了，所以可以直接二分一下暴力判，单次复杂度 $O(n\log n)$，劣于 SAM。

• 查询一个串作为子串出现次数

可以发现在后缀数组上一定是一个区间，所以可以直接二分左右端点即可，同样劣于 SAM。

• 询问两个后缀的 $\text{lcp}$

有一个简单的二分 + hash 做法，复杂度 $O(\log n)$。

这个也比较简单，考虑以前缀方式出现的一定是一个区间，因此就相当于相邻两个做然后取 $\min$ 即可，也就是对区间 $h$ 求 $\min$，用 st 表，复杂度 $O(1)。$

• 不同子串个数，P2408

考虑一个串在那些地方会重复出现，显然是作为前缀出现，那么就是在一个区间中，因此直接考虑每个后缀与前面一个重复的前缀数即可。

所以答案就是 $\frac{n(n+1)}{2}-\sum_{i=2}^n h_i$。

---

P2852 $\color{green}\bigstar$

已知一个字符串 $S$，求所有出现次数至少为 $k$ 的子串的长度最大值。

$n\le 20000$。

ez 题，直接考虑一个子串一定是一个区间的前缀，那么直接对于所有长度为 $k$ 的区间求最小值然后找最大即可。

code

---

P1117 $\color{Gold}\bigstar$

已知一个串 $S$，求有多少个子串可以表示为 AABB 的形式。

$n\le 30000$。

首先有一个很简单的 hash 暴力做法，复杂度 $O(n^2)$，有 $95$，流汗黄豆。

考虑记录 $f_i$ 表示 $i$ 结尾的形如 AA 的串的个数，那么可以只需要倒着再做一遍两边合并就可以得到答案。

但是这个东西不会做。

正解很牛，考虑枚举 AA 串中 A 的长度为 $len$，然后每 $len$ 个点中标记最后一个点，这样可以发现所有选子串方案一定经过两个被标记的点。

画一下图

圆就是标记点，竖线是 AA 的中点，可以发现圆点向左的箭头两个子串相等，向右的也一样。

因此就是求公共前缀和公共后缀，然后算一下合法的方案，那么就是区间加即可。

code

---

P2870 $\color{green}\bigstar$

有一个长度为 $n$ 的串，每次可以开头取一个或者结尾取一个，求拼成的字典序最小的字符串。

$n\le 5\times 10^5$。

考虑如果左右不相等显然直接取小的，但如果相等呢。

考虑对比 $i,j$，那么就是 $[i,n]$ 与 $[1,j]$ 翻转的结果比较字典序，也就是 $S$ 与 $S$ 翻转后的结果一起进行后缀排序。

考虑两个串怎么一起搞，简单的做法是接在后面，中间放一个特殊字符即可。

code

---

P2178 $\color{gray}\bigstar$

题比较长，自己看吧。

简单题，考虑上面 $\text{lcp}$ 是区间的那个东西，因此相当于 $r$ 递减，就是连一些边，答案直接用并查集维护即可，我用了暴力启发式合并。

code

---

P4248 $\color{green}\bigstar$

已知一个串 $S$，求对于所有 $i<j$，求 $\sum len(S_i)+len(S_j)-2\times \text{lcp}(S_i,S_j)$。

$n\le 5\times 10^5$。

相当于只需要算后面那个东西，也就是区间 $\min$ 求和，这个东西可以用单调栈简单维护即可。

code

---

ABC268Ex *2560 $\color{green}\bigstar$

有一个字符串 $T$，求最少把多少个字符变成 *，使得 $S_1,S_2..S_m$ 都没有在 $T$ 中出现过。

简单题，考虑一个 $S_i$ 出现的所有位置，这些位置对应着一个个区间，相当于区间中至少选一个，因此直接考虑设 $f_i$ 表示以 $i$ 为左端点，右端点最小是多少，然后就有 $n$ 个区间直接贪心即可，求这个 $f_i$ 可以直接建 SA，相当于只在一个区间中出现，那么直接线段树维护即可。

更加简单的做法是考虑每个区间一定贪心向后选，因此直接 AC 自动机，匹配上了就把最后一个改掉就可以了。

code

---