ACM - 数据结构 | 后缀数组

最新推荐文章于 2024-12-29 22:45:43 发布

R-bin

最新推荐文章于 2024-12-29 22:45:43 发布

阅读量222

点赞数

分类专栏： ACM - 数据结构文章标签：数据结构

本文链接：https://blog.csdn.net/weixin_43400995/article/details/105028760

版权

ACM - 数据结构专栏收录该内容

3 篇文章 0 订阅

订阅专栏

后缀数组

定义

$s u f f (i)$ 从第 $i$ 个字符开始的后缀

$s a [i]$ 保存排名为 $i$ 的后缀的首字符下标

$r k [i]$ $s u f f (i)$ 的排名

$L C P (i, j)$ $s u f f (s a [i])$ 和 $s u f f (s a [j])$ 的最长公共前缀

$h e i g h t (i)$ $L C P (i, i - 1)$

代码

若数组的长度为 $n$ ，则 $s [0]$ 到 $s [n - 1]$ 保存其ASCII码值，并且将 $s [n]$ 赋值为 $0$ ，最大值小于 $m$

$r k [0 : n - 1]$ 为有效值， $r k [n]$ 必定为 $0$ 无效值

$s a [1 ： n]$ 为有效值， $s a [0]$ 必定为 $n$ 是无效值

$h e i g h t [2 : n]$ 为有效值

const int MAXN = 2e5 + 10;
int wa[MAXN], wb[MAXN], wsf[MAXN], wv[MAXN];
int rk[MAXN], height[MAXN], sa[MAXN];
int RMQ[MAXN];
int mm[MAXN];
int best[20][MAXN];
int s[MAXN];
char str[MAXN]

int cmp(int *r, int a, int b, int k){
	return r[a] == r[b] && r[a + k] == r[b + k];
}
void getsa(int *s, int *sa, int n, int m)
{
	int i, j, p, *x = wa, *y = wb, *t;
	for (i = 0; i<m; i++)  wsf[i] = 0;
	for (i = 0; i <= n; i++)  wsf[x[i] = s[i]]++;
	for (i = 1; i<m; i++)  wsf[i] += wsf[i - 1];
	for (i = n; i >= 0; i--)  sa[--wsf[x[i]]] = i;
	p = 1;
	j = 1;
	for (; p <= n; j *= 2, m = p)
	{
		for (p = 0, i = n + 1 - j; i <= n; i++)  y[p++] = i;
		for (i = 0; i <= n; i++)  if (sa[i] >= j)  y[p++] = sa[i] - j;
		for (i = 0; i <= n; i++)  wv[i] = x[y[i]];
		for (i = 0; i<m; i++)  wsf[i] = 0;
		for (i = 0; i <= n; i++)  wsf[wv[i]]++;
		for (i = 1; i<m; i++)  wsf[i] += wsf[i - 1];
		for (i = n; i >= 0; i--)  sa[--wsf[wv[i]]] = y[i];
		t = x;
		x = y;
		y = t;
		x[sa[0]] = 0;
		for (p = 1, i = 1; i <= n; i++)
			x[sa[i]] = cmp(y, sa[i - 1], sa[i], j) ? p - 1 : p++;
	}
}

void getheight(int *s, int n)
{
	int i, j, k = 0;
	for (i = 1; i <= n; i++)  rk[sa[i]] = i;
	for (i = 0; i<n; i++)
	{
		if (k)
			k--;
		else
			k = 0;
		j = sa[rk[i] - 1];
		while (s[i + k] == s[j + k])
			k++;
		height[rk[i]] = k;
	}
}

void initRMQ(int n) {
	mm[0] = -1;
	for (int i = 1; i <= n; i++)
		mm[i] = ((i&(i - 1)) == 0) ? mm[i - 1] + 1 : mm[i - 1];
	for (int i = 1; i <= n; i++) best[0][i] = i;
	for (int i = 1; i <= mm[n]; i++)
		for (int j = 1; j + (1 << i) - 1 <= n; j++) {
			int a = best[i - 1][j];
			int b = best[i - 1][j + (1 << (i - 1))];
			if (RMQ[a] < RMQ[b]) best[i][j] = a;
			else best[i][j] = b;
		}
}

int askRMQ(int a, int b) {
	int t;
	t = mm[b - a + 1];
	b -= (1 << t) - 1;
	a = best[t][a]; b = best[t][b];
	return RMQ[a] < RMQ[b] ? a : b;
}

int lcp(int a, int b) {
	a = rk[a]; b = rk[b];
	if (a > b) swap(a, b);
	return height[askRMQ(a + 1, b)];
}

void work() {
	while (scanf("%s", str) == 1) {
		int n = 0;
		int len = strlen(str1);
		for (int i = 0; i < len; i++) s[n++] = str[i] - 'a' + 1;
		s[n] = 0;	//keypoint,n没有增加
		getsa(s, sa, n, 128);
		getheight(s, n);
		initRMQ(n);
	}
}