【算法】KMP模板（C++）

---

前言

本文主要记录基础KMP算法的原理及实现，方便查找和使用

---

一、KMP作用

KMP算法是一种改进的字符串匹配算法，由D.E.Knuth，J.H.Morris和V.R.Pratt提出的，因此人们称它为克努特—莫里斯—普拉特操作（简称KMP算法）。KMP算法的核心是利用匹配失败后的信息，尽量减少模式串与主串的匹配次数以达到快速匹配的目的。具体实现就是通过一个next()函数实现，函数本身包含了模式串的局部匹配信息。KMP算法的时间复杂度$O(m+n)$。

二、代码实现

1.定义数据结构

char a[1000010]; // 文本串
char b[1000010]; // 模板串(将被匹配的子串)
int kmp_next[1000010]; // next数组

2.初始化next数组

void getNext(int m){
	int j = 0;
	// 初始化next[0]的值
	kmp_next[0] = 0;
	for(int i=1; i<m; ++i){
		// 当这一位不匹配时，将j指向此位之前最大公共前后缀的位置
		while(j>0 && b[i]!=b[j]) j=kmp_next[j-1];
		// 如果这一位匹配，那么将j+1，继续判断下一位
		if(b[i]==b[j]) ++j;
		// 更新next[i]的值
		kmp_next[i] = j;
	}
}

三、例题简述

1.使用KMP寻找匹配位置

例：给定两个字符串 $a$ , $b$，求 $b$ 是否为 $a$ 的子串，并输出 $b$ 在 $a$ 中第一次出现的位置。

int kmp(int n,int m){
	int i, j = 0;
	// 初始化位置p = -1
	int p = -1;
	// 初始化next数组
	getNext(m);
	for(i=0; i<n; ++i){
		// 当这一位不匹配时，将j指向此位之前最大公共前后缀的位置
		while(j>0 && b[j]!=a[i]) j=kmp_next[j-1];
		// 如果这一位匹配，那么将j+1，继续判断下一位
		if(b[j]==a[i]) ++j;
		// 如果是子串(m位完全匹配)，则更新位置p的值，并中断程序
		if(j==m){
			p = i - m + 1;
			break;
		}
	}
	// 返回位置p的值
	return p;
}

2.使用KMP计算匹配次数

例：给定两个字符串 $a$ , $b$，并输出 $b$ 在 $a$ 中出现的次数。

int kmp(int n,int m){
    // 初始化答案数量res = 0
	int i, j = 0, res = 0;
	// 初始化next数组
	getNext(m);
	for(i=0; i<n; ++i){
		// 当这一位不匹配时，将j指向此位之前最大公共前后缀的位置
		while(j>0 && b[j]!=a[i]) j=kmp_next[j-1];
		// 如果这一位匹配，那么将j+1，继续判断下一位
		if(b[j]==a[i]) ++j;
		// 如果是子串(m位完全匹配)，则答案数量增加，res = res + 1
		if(j==m) ++res;
	}
	// 返回答案数量
	return res;
}

---

总结

以上为作者总结的基础KMP匹配算法的模板，有需要的读者可以收藏本篇
例题：洛谷P3375