扩展 KMP

扩展 KMP

简述

例题

给你两个字符串 $a$，$b$，长度分别为 $n$，$m$。

请输出 $b$ 的每一个后缀与 $b$ 的最长公共前缀以及 $a$ 的每一个后缀的最长公共前缀。

扩展 KMP：求出字符串 $a$ 的所有后缀与 $b$ 的最长公共前缀长度，时间复杂度 $\mathcal{O}(|S|+|T|)$。

该解法思想与 KMP 类似，所以称作扩展 KMP。

求法

定义 $nex{t}_i$ 为 $b$ 由 $i$ 开始的后缀与 $b$ 的最长公共前缀长度。（即为自配）

定义 $ex{t}_i$ 为 $a$ 由 $i$ 开始的后缀与 $b$ 的最长公共前缀长度。

这是本题的第一问，也是第二问的辅助数组。

易得，此时 $nex{t}_0=|b|$。

思考，如果已知 $nex{t}_0\sim nex{t}_{x-1}$，如何求 $nex{t}_x?$

有一个较具体的例子，如果 $k=121$，那么 $nex{t}_0$ 到 $nex{t}_{120}$ 都已经计算完毕，且 $l_{120}=100$，$r_{120}=130$。

这意味着 $b[100..130]=b[0..3]$，那么 $b[121..130]=b[21..30]$，这样 $nex{t}_{21}$ 对于计算 $nex{t}_{121}$ 就非常有帮助，如果 $nex{t}_{21}=3$，那么 $nex{t}_{121}=3$。

设 $nex{t}_0\sim nex{t}_k$ 已经算好，记 $p$ 为在以前的匹配过程中在 $b$ 串中的最远位置，即 $p=max(i+next[i]-1)$，其中 $i=1...k$。

设取到这个最大值 $p$ 的位置是 $p0$。

则 $a[p0...p]=b[0...p-p0]$。

假设 $i+next[i-k]\le p$，则 $nex{t}_x=next[x-k]$。

否则，暴力枚举 $nex{t}_x$。

求 $ext$ 同理，若 $i+next[i-k]\le p$，则 $ex{t}_x=next[x-k]$。

否则，暴力枚举 $ex{t}_x$。

由于此时的 $k$ 不降，所以时间复杂度为 $\mathcal{O}(|S|)$。

$tobecontinue$

代码实现

#include <bits/stdc++.h>
using namespace std;

#define _ (int) 3e7 + 5

char a[_], b[_];

int n, m, nxt[_], ext[_];

long long Ans1, Ans2;

void get_nxt()
{
	nxt[0] = m;
	int j = 0;
	while(j + 1 < m && b[j] == b[j + 1]) ++j;
	nxt[1] = j;
	int k = 1;
	for(int i = 2; i < m; ++i)
	{
		int p = k + nxt[k] - 1;
		if(i + nxt[i - k] <= p) nxt[i] = nxt[i - k];
		else
		{
			j = max(p - i + 1, 0);
			while(i + j < m && b[i + j] == b[j]) ++j;
			nxt[i] = j;
			k = i;
		}
	}
}

void get_ext()
{
	int j = 0;
	while(j < n && j < m && a[j] == b[j]) ++j;
	ext[0] = j;
	int k = 0;
	for(int i = 1; i < n; ++i)
	{
		int p = k + ext[k] - 1;
		if(i + nxt[i - k] <= p) ext[i] = nxt[i - k];
		else
		{
			j = max(p - i + 1, 0);
			while(i + j < n && j < m && a[i + j] == b[j]) ++j;
			ext[i] = j;
			k = i;
		}
	}
}

signed main()
{
	scanf("%s%s", a, b);
	n = strlen(a);
	m = strlen(b);
	get_nxt();
	get_ext();
	for(int i = 0; i < m; ++i) Ans1 ^= 1ll * (i + 1) * (nxt[i] + 1);
	for(int i = 0; i < n; ++i) Ans2 ^= 1ll * (i + 1) * (ext[i] + 1);
	printf("%lld\n%lld\n", Ans1, Ans2);
	return 0;
}