KMP算法学习
一、算法介绍
KMP是三位大牛:D.E.Knuth、J.H.Morris和V.R.Pratt同时发现的。
KMP算法其实就是一种改进的字符串匹配算法,关键是利用匹配后失败的信息,尽量减少模式串(W)与主串(T)的匹配次数以达到快速匹配的目的。具体实现就是实现一个next() 函数,函数本身包含了模式串的局部匹配信息。时间复杂度 O(m+n)。
在实质上,KMP算法就是利用了贪心的思想去加速了字符串的匹配。
二、KMP算法分析
假设我们给定了两个字符串S,T。其中S是模式串,T是主串。
#string S = "abcabca" T = "aaaaaabababababcabcabababcabcaaa"
// 我们首先来构造一个nxt[]数组,来记录跳转的指针
nxt[] = {-1,0,0,0,1,2,3}
假设,我们通过O(n)的算法来遍历T字符串的时候。
当我们发现匹配到S串的第五个位置的时候,如果我们发现T字符串当前位置与S串不同,那我们可以认为T字符串的前一个字符与S的第一个字符相同。所以,在当前位置,我们可以从S的第二个开始继续匹配。
三、完整代码
#include<bits/stdc++.h>
using namespace std;
#define LL long long
const int maxn = 1e5+10;
const LL mod = 998244353;
int nxt[maxn];
void getNext(string s)
{
int len=s.length();nxt[0]=0;
nxt[1]=0;
for(int i=1;i<len;i++)
{
int j=nxt[i];
while(j&&s[i]!=s[j]) j=nxt[j];
if(s[i]==s[j]) nxt[i+1]=j+1;
else nxt[i+1]=0;
}
}
LL KMP(string s,string ss)
{
LL len1=s.length(),len2=ss.length();
if(len2>len1) return 0;
LL j=0,ans=0;
for(LL i=0;i<len1;i++)
{
while(j&&s[i]!=ss[j]) j=nxt[j];
if(s[i]==ss[j]) j++;
if(j==len2)
{
return i-j + 1;
}
}
return -1;
}
int main()
{
string s,sub;
cin >> s >> sub;
getNext(sub);
for(int i = 0;i < s.length();++i)
cout << nxt[i] << ' ';
cout << endl;
int ans = KMP(s,sub);
cout << ans << endl;
return 0;
}
总结
KMP算法是一种通过贪心来实现加快字符串匹配的算法。
算法的思想是值得深思的,加深对于其中每一个变量的理解会获益良多。
在一般的情况下可以通过str类中的find函数,实现KMP匹配。
参考链接:https://blog.csdn.net/crazy_scott/article/details/79211961