KMP算法
介绍
用处:用于串的模式匹配,即找出模式串在主串中的出现位置
朴素想法,直接遍历两个串,失配回到主串开始比较位置的下一位继续匹配,复杂度(O(nm))
KMP算法即在(O(n+m))复杂度内匹配的算法
算法实现
通过一个叫(next)数组的东西,使指向主串的(i)指针不回溯,只改变指向模式串的(j)指针
所以重点就是在于失配后(j)移动到哪里,也就是关于(next)数组的部分
(Next[])的基本意义:
在串ss中寻找串s,
当前已经得到了(s[1…j-1]=ss[i-j+1…i-1]),
正在检查(s[j]与ss[i])是否相等,发现不相同,失配,
(如果相同,i和j自增1,继续向后检查)
如果按照暴力算法,ss上的“指针”需要回到i-j+1,s上的“指针”j需要回到0;
我们试图让i往回不移动,仅仅是让j往回移动,移动到一个合适的、不会丢失任何匹配可能的位置,这个位置就是我们的Next[j]。
(next)数组的定义是
即最大的(P[1 , k-1] = P[j-k+1 ,j-1])
求出next之后在失配时让(j=next[j])就是KMP算法了
如何求出(next)
已知 Next[j]=kn。
即存在 (s[1,kn-1]=s[j-kn+1,j-1])
(最长后缀等于字符串前缀)
分情况讨论,对于(j+1)(现在需要求的Next数组位)有两种情况
- 若$ s[kn]=s[j]$,那么可以得到 (s[1,kn]=s[j-kn+1,j]) ,根据定义,不难推断出(Next[j+1]=kn+1)
- 若不相等,则转化为匹配自身的问题,不断地令(j=Next[j])直至出现 (s[kn]=s[j])(转化为1情况,(Next[j+1]=kn+1));或者(j=1)(找不到后缀等于前缀)匹配到头,(Next[j+1]=1)。
7月22日更新:
找到了更好理解的kmp板子,这个模板中,nxt[i]表示最大的x,满足s[1 : x] 是s[1 : i] 的后缀,匹配的时候用前一个nxt来匹配。如果字符串从0开始,那么将j+1改为j,nxt[j]改为nxt[j-1]即可,j的初值不用改变
模版
#include <bits/stdc++.h>
#define N 1000005
using namespace std;
int n, m;
int nxt[N];
int s[N], t[N];
void getNext(int s[]) {
nxt[1] = 0;
for (int i = 2, j = 0; i <= m; i++) {
while (j && s[i] != s[j + 1]) j = nxt[j];
if (s[i] == s[j + 1]) j++;
nxt[i] = j;
}
}
int kmp(int s[], int t[], int pos) {
for (int i = pos, j = 0; i <= n; i++) {
while (j && t[j + 1] != s[i]) j = nxt[j];
if (t[j + 1] == s[i]) j++;
if (j >= m) return i - m + 1;
}
return -1;
}
int main() {
int t1;
scanf("%d", &t1);
while (t1--) {
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; i++) {
scanf("%d", &s[i]);
}
for (int i = 1; i <= m; i++) {
scanf("%d", &t[i]);
}
memset(nxt, 0, sizeof(nxt));
getNext(t);
printf("%d
", kmp(s, t, 1));
}
return 0;
}