KMP算法

 

看了牛客网进阶学习中的KMP算法，觉得讲的还不错，加深了自己的理解（上学期郑老师讲完基本就忘了）。

目录

kmp要求的问题

next数组的用处

匹配的方法

next数组的求法

最后附上总的代码

---

 

• kmp要求的问题

       从第一个字符串中找到第二个字符串第一次出现的位置，如果没有输出-1。

---

• next数组的用处

      记录最长的公共的前缀和后缀。

      举个例子，abcab，next数组存储的值为2；aaaa，next数组值为3，因为不允许覆盖全部；abc，值就为0了。

---

• 匹配的方法

       由于next数组已经求出了公共的前后缀的长度，如果一个匹配不能完成，那么就直接用str2去匹配str1所对应的后缀部分。如下图所示，就是这么跳跃的。

而用代码描述就是这样的（我自己写的，可能存在问题）：

i和j分别是指向str1和str2的指针，用指针的移动表示字符串开头的移动。

int kmp()
{
	get_next();
	int i = 0;
	int j = 0;
	while(i<s1.length()&&j<s2.length())
	{
		if(s1[i]==s2[j]) //如果两个相等，一起向后 
			i++,j++;
		else if(nxt[j]==-1) //如果不相等，并且nxt数组=-1了，说明没有公共段了，只能i++ 
			i++;
		else 
			j = nxt[j]; //否则直接跳过公共段，从第一个不相同的位置开始匹配 
	}
	return j==s2.length()?i-j:-1;
}

---

• next数组的求法

      next数组是根据数学归纳法求出来的，默认nxt[0]=-1,nxt[1]=0,然后从2开始，如果当前字符-1和上一个字符的公共前缀的后一个字符（记为k）相等，那么结果+1；否则的话要分情况处理，如果nxt[k]！=0的话，那么就让i和字符k的公共前缀的后一个字符去匹配。

      用语言讲太抽象了，还是放张示意图。

这对应的是第一种情况，对于字符d，由于它的前一个字符c和对于c来说的前缀ab的后一个字符c相等，所有nxt数组 = nxt[i-1]+1;

对于第二种情况，就需要如图所示依次匹配，直到next数组结果为-1，即匹配到第一个字符了。

具体的代码实现：

void get_next()
{
	nxt[0] = -1;
	nxt[1] = 0;
	int i = 2;
	int cn = 0; //cn表示当前的最长公共前后缀 
	while(i<s2.length())
	{
		if(s2[i-1]==s2[cn]) //如果i-1和对应的相等，结果+1 
			nxt[i++] = ++cn;
		else if(cn>0) //否则就找cn是否和i-1相等 
			cn = nxt[cn];
		else {
			nxt[i++] = 0;
		}
	}	
}

---

• 最后附上总的代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <vector>
#include <map>
#include <set>
#include <queue>
using namespace std;

#define INIT(x) memset(x,0,sizeof(x))
#define eps 1e-8

typedef long long ll;
const int inf = 0x3f3f3f3f;
const int maxn = 200005;

string s1,s2;
int nxt[maxn];

void get_next()
{
	nxt[0] = -1;
	nxt[1] = 0;
	int i = 2;
	int cn = 0; //cn表示当前的最长公共前后缀 
	while(i<s2.length())
	{
		if(s2[i-1]==s2[cn]) //如果i-1和对应的相等，结果+1 
			nxt[i++] = ++cn;
		else if(cn>0) //否则就看cn是否大于0,能回溯就回溯
			cn = nxt[cn];
		else {
			nxt[i++] = 0;
		}
	}	
}

int kmp()
{
	get_next();
	int i = 0;
	int j = 0;
	while(i<s1.length()&&j<s2.length())
	{
		if(s1[i]==s2[j]) //如果两个相等，一起向后 
			i++,j++;
		else if(nxt[j]==-1) //如果不相等，并且nxt数组=-1了，说明没有公共段了，只能i++ ，这里不可以改成nxt[j]<=0,否则j不能回到s2的开头位置
			i++;
		else 
			j = nxt[j]; //否则直接跳过公共段，从第一个不相同的位置开始匹配 
	}
	return j==s2.length()?i-j:-1; //i位正好匹配到末尾，减去j就是开头
}

int main()
{
	cin>>s1>>s2;
	int ans = kmp();
	cout<<ans<<endl;
	return 0;
}