后缀数组(SA)总结

后缀数组(SA)总结

这个东西鸽了好久了,今天补一下

概念

后缀数组\(SA\)是什么东西?

它是记录一个字符串每个后缀的字典序的数组

\(sa[i]\):表示排名为\(i\)的后缀是哪一个。

\(rnk[i]\):可以理解为\(SA\)数组的逆,记录后缀\(i\)的排名是多少,\(rnk[SA[i]]=i\)

\(lcp[i]\):别人一般叫\(height\),表示后缀\(SA[i]\)\(SA[i-1]\)的最长公共前缀的长度。

后缀排序

求出后缀数组的算法,模板题

代码

先上代码,便于理解

#define cmp(i, j, k) (y[i] == y[j] && y[i + k] == y[j + k])
void Get_SA() {
    static int x[MAX_N], y[MAX_N], bln[MAX_N];
	int M = 122; 
	for (int i = 1; i <= N; i++) bln[x[i] = a[i]]++; 
	for (int i = 1; i <= M; i++) bln[i] += bln[i - 1]; 
	for (int i = N; i >= 1; i--) sa[bln[x[i]]--] = i; 
	for (int k = 1; k <= N; k <<= 1) { 
		int p = 0; 
		for (int i = 0; i <= M; i++) y[i] = 0; 
		for (int i = N - k + 1; i <= N; i++) y[++p] = i; 
		for (int i = 1; i <= N; i++) if (sa[i] > k) y[++p] = sa[i] - k;
		for (int i = 0; i <= M; i++) bln[i] = 0; 
		for (int i = 1; i <= N; i++) bln[x[y[i]]]++; 
		for (int i = 1; i <= M; i++) bln[i] += bln[i - 1]; 
		for (int i = N; i >= 1; i--) sa[bln[x[y[i]]]--] = y[i]; 
	    swap(x, y); x[sa[1]] = p = 1;
		for (int i = 2; i <= N; i++) x[sa[i]] = cmp(sa[i], sa[i - 1], k) ? p : ++p;
		if (p >= N) break;
		M = p; 
	} 
} 

算法流程

\(sa\)的算法有倍增法和\(DC3\),因为后者有码量大、常数大、我不会等种种缺点,

这里只介绍倍增算法。

我们如果对于每个倍增完的二元组,每个都\(sort\)一下,复杂度是\(O(nlog^2)\)的。

那么将基数排序应用到其中去,就可以做到\(O(nlogn)\),具体做法:

我们考虑一下普通的基数排序是怎么排二元组

先将第二位丢进桶里,然后按照第一维的次序取出。

那么这个字符串怎么排呢?

首先当\(k=0\)时,我们直接桶排一下就行了。

但是我们还要接着排啊,

还记得吧,基排序是先按照第二维从小往大排

那么,我们就先把第二维的顺序搞出来

首先最小的一定就是没有第二维的东西

所以我们先把这些数直接丢进数组里面

接下来就是有第二维的东西啦

\(i\)位的第二维是啥?\(rnk[i+k]\)

所以,从小到达枚举\(sa\),这样保证第二维从小往大

那么,只要\(sa[i]>k\)

就证明它是一个东西的第二维

所以,把\(sa[i]−k\)

丢到数组里面去就好啦

这样的话,按照第二维就排好啦

再来依次按照第一维丢到桶里面去

做一遍基数排序就好啦

这样就能够求出\(sa\)

看起来很简单诶。。

只是数组不要搞混了

一定搞清楚每个数组是干啥的

比如我的代码

\(sa\)是后缀数组,\(sa[i]\)表示排名为i的串是哪一个

\(rnk[i]\)相当于排名,\(rnk[i]\)表示第i个串的排名

\(x,y\)两个数组是记录顺序的

分别记录第一维和第二维的排序的顺序

\(bln\)是桶。

如果实在理解不了,就背吧,反正也没有多长

那么\(lcp\)数组怎么求呢?

\(\forall i<j\),不妨设\(rnk[j]<rnk[k]\),那么以\(j\)开头的后缀和\(k\)开头的后缀的最长公共前缀就是\(\min _{i=rnk[j]+1}^{rnk[k]} lcp[i]\)

有一个引理:

定义\(h[i]=lcp[rnk[i]]\),那么,\(h[i]\geq h[i-1]-1\)

证明:设\(s[k...]\)为排在\(s[i-1...]\)的前一名的后缀,其最长公共前缀为\(h[i-1]\),则\(s[k+1...]\)\(s[i...]\)的最长公共前缀显然大于等于\(h[i-1]-1\),原结论得证。

然后这样求就可以了:

    for (int i = 1; i <= N; i++) rnk[sa[i]] = i; 
    for (int i = 1, j = 0; i <= n; i++) { 
        if (j) j--; 
        while (a[i + j] == a[sa[rnk[i] - 1] + j]) ++j; 
        lcp[rnk[i]] = j; 
    }

一些trick

总结了一些食用SA时的\(trick\)

一、对于可重复的最长重复子串问题(若子串\(s\)重复出现次数大于等于二,则称重复子串)\(Ans=\max_{i=1}^nlcp_i\)

二、对于不可重叠的最长重复子串问题,二分,将问题转化为是否有两个长度为\(k\)的子串是相同的,且不重叠。将\(lcp\)数组分组,最长公共前缀不小于\(k\)的为一组其中如果有一组\(sa[i]\)之差大于\(k\)时,则成
立。

三、对于可重叠的重复\(k\)次最长重复子串,与上一种方法思路相似,二分,问题转化为判断是否存在\(k\)个长度为\(l\)的子串是相同的,将最长公共子串大于\(l\)的后缀分为一组,查看每一组内后缀个数是否大于\(k\)

四、对于多个字符串的问题,通常用一个原串中不会出现的字符将两个字符串连接为一个。对于最长公共子串问题,首先将两个字符串用一个未出现过的字符连接起来,然后求出它们的最长公共前缀,解时注意判断是否在间隔符两边。

五、求取长度不小于\(k\)的公共子串个数时,将两个字符串按照上述方法连接,中间用一个未曾出现过的字符隔开,计算所有后缀之间最长公共前缀的长度,用单调栈维护最长公共前缀的长度。

六、对于在多个字符串中,出现不小于\(k\)个字符串的最长公共子串。按照上述方法连接多个字符串后,使用二分法。对于给定的长度,先分组,判断每组字符串后缀是否出现在不同的\(k\)个字符串中。

七、对于在每个字符串中至少出现两次且不重叠的最长公共子串时,按照上述方法连接多个字符串,使用二分法。对于给定的长度,先分组,判断是否有一组包含每个字符串中的两个不重叠答案。

一些后话

我还不太熟悉,题目暂未整理出来。

以后会提供每个\(trick\)的例题及一些题单。

如有错漏之处,请联系作者。

参考文章:
yyb的博客 https://www.cnblogs.com/cjyyb/p/8335194.html
清华大学出版社《ACM/ICPC算法基础训练教程》第8章

posted @ 2019-01-21 19:38  heyujun  阅读(1475)  评论(8编辑  收藏  举报