字符串匹配 Kmp

给定一个主串 t t t 和模式串 p p p,求 p p p t t t 中出现的下标。

我会暴力!

char t[N], p[N]; scanf("%s%s", t + 1, p + 1);
	for (int i = 1; i <= strlen(t + 1); i++)  { 
		bool f = 1; int k = i;
		for (int j = 1; j <= strlen(p + 1); i++, j++) 
			if (t[i] != p[j]) {
				f = 0; break;
			}
		if (f) printf("%d\n", i);
        else i = k; // 如果匹配失败,指针 i 回溯
	}

我们用指针 i i i t t t,指针 j j j p p p。在暴力中,如果匹配失败那么 i i i j j j 都会回溯到 1 1 1

比如 t = ABBABBABABAAABABAAA , p = ABBABAABABAA t = \text{ABBABBABABAAABABAAA}, p =\text{ABBABAABABAA} t=ABBABBABABAAABABAAA,p=ABBABAABABAA。让 p p p 去匹配 t t t

ABBABBABABAAABABAAA 	i from 1 to 6
ABBABAABABAA 			j from 1 to 6
     !

第一次匹配中, j j j 在第六位失配,但是 p p p 的前五位 ABBAB 匹配成功了。暴力会让 i i i 回溯到 2 2 2 j j j 回溯到 1 1 1,重新进行匹配。而我们要想办法减少回溯。

那么我们要利用已经匹配成功的前五位。可以发现 ABBAB最长公共前后缀AB,这意味着 p p p 的前缀 AB,可以与 t t t 的后缀 AB 匹配,所以把 j j j 回溯到前缀 AB 的下一位也就是 ∣ AB ∣ + 1 = 3 |\text{AB}| + 1 = 3 AB+1=3。而 t t t 中后缀 AB 的下一位为 i i i 所以 i i i 原地不动。

AB B AB BABABAAABABAAA i from 1 to 6
AB B AB AABABAA		   j from 1 to 6
        !
ABBABBABABAAABABAAA	   i from 6 to 10 
   ABBABAABABAA 	   j from 2 + 1 = 3 to 7
         !  

为了实现刚才的算法。我们需要预处理一个 n x t nxt nxt 数组, n x t i nxt_i nxti t 1 ∼ i t_{1 \sim i} t1i 的最长公共前后缀的长度 + 1 +1 +1我会暴力!

考虑 t t t 匹配到了第 i i i 位,这意味着求出了 n x t i − 1 nxt_{i-1} nxti1。如果 t 1 , i − 1 t_{1,i-1} t1,i1 的前缀后面的字符等于 t i t_i ti,那么 n x t i = n x t i − 1 + 1 nxt_i = nxt_{i-1} + 1 nxti=nxti1+1

前 i - 1 位 ABBABBA
最长公共前后缀为 A
前 i 位     ABBABBAB
前缀的下一位 = s 的第 i 位 = B
所以 nxt[i] = nxt[i - 1] + 1 = 2;

否则,考虑前缀的前缀和后缀的后缀,如果前缀的前缀的下一个字符等于 t i t_i ti,那么 n x t i = n x t n x t i − 1 + 1 nxt_{i} = nxt_{nxt_{i - 1}} + 1 nxti=nxtnxti1+1

前 i - 1 位 ABAAABA
最长公共前后缀为 ABA
前 i 位     ABAAABAB
前缀的下一位 ≠ s 的第 i 位
前缀的前缀为 A
前缀的前缀的下一位 = s 的第 i 位 = B
所以 nxt[i] = nxt[nxt[i - 1]] + 1 = 2
#include <bits/stdc++.h>
using namespace std;
const int N = 1e6 + 5;
int read() {
	int x = 0, f = 0; char ch = 0;
	while (!isdigit(ch)) f |= ch == '-', ch = getchar();
	while (isdigit(ch)) x = (x << 3) + (x << 1) + (ch ^ 48), ch = getchar();
	return f ? -x : x;
}
int nxt[N];
char t[N], p[N];
int nt, np;
void getnxt() {
	int i = 0, j = -1;
	nxt[0] = -1;
	while (i < np) {
		if (j == -1 || p[i] == p[j]) nxt[++i] = ++j;
		else j = nxt[j];
	}
}
void kmp() {
	int i = 0, j = 0;
	while (i < nt) {
		if (j == -1 || t[i] == p[j]) i++, j++; 
		else j = nxt[j];
		if (j == np) printf("%d\n", i - j + 1), j = nxt[j]; 
	}
}
int main(){
	scanf("%s%s", t, p);
	nt = strlen(t), np = strlen(p);
	getnxt(); kmp();
	for (int i = 1; i <= np; i++) printf("%d ", nxt[i]); 
	return 0;
}

update:这里给出一个更好的理解的 kmp,假设我们在 s 1 s_1 s1 处发现 s 1 ∼ s n s_1 \sim s_n s1sn t 1 ∼ t n t_1 \sim t_n t1tn 匹配。现在要匹配 s 2 ∼ s n s_2 \sim s_n s2sn t 1 ∼ t n t_1 \sim t_n t1tn。因为 s 1 ∼ s n = t 1 ∼ t n s_1 \sim s_n = t_1 \sim t_n s1sn=t1tn 所以 s 2 ∼ s n = t 2 ∼ t n s_2\sim s_n = t_2 \sim t_n s2sn=t2tn,所以是匹配的是 t 2 ∼ t n t_2 \sim t_n t2tn t 1 ∼ t n t_1 \sim t_n t1tn。实际上就是求一个 n x t nxt nxt 其中 n x t i nxt_i nxti 表示 t 1 ∼ t i t_1 \sim t_i t1ti t t t 的最长公共前缀。

posted @ 2020-02-21 13:27  ylxmf2005  阅读(42)  评论(0)    收藏  举报