next数组
关于KMP算法已经有很多详解了这里就不赘述了,这里说明一下next数组怎么是来的,求next数组跟模式串匹配子串实际上是一样的,主要就是最长公共前后缀。
void get_next()
{
//ne[1]=0,所以i从2开始
for(int i=2,j=0;i<=n;i++)
{
/*
如果j=0,即第一个字符就不匹配那就是母串移动,也就是i往后移
如果j!=0,且发生不匹配,就是说在第i个字符前面已经有字符匹配成功,就可以利用这个信息移动j
j是最长公共前后缀中后缀的最后一个字符,让j移动到前缀最后一个字符,也就是next[j],
i不动让他们进行比较,这就是KMP算法核心
*/
while(j&&p[i]!=p[j+1]) j=ne[j];
//若是匹配则j移动
if(p[i]==p[j+1]) j++;
//ne[i]是最长公共前后缀的长度
ne[i]=j;
}
}
例题
原题这里->831. KMP字符串
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 1e5+10;
const int M = 1e6+10;
int n,m;
int ne[N];
char s[M],p[N];
void get_next()
{
for(int i=2,j=0;i<=n;i++)
{
while(j&&p[i]!=p[j+1]) j=ne[j];
if(p[i]==p[j+1]) j++;
ne[i]=j;
}
}
int main()
{
cin >> n >> p+1 >> m >> s+1;
get_next();
for(int i=1,j=0;i<=m;i++)
{
while(j&&s[i]!=p[j+1]) j=ne[j];
if(s[i]==p[j+1]) j++;
if(j==n)
{
printf("%d ",i-n+1-1);
//这里可以让while少循环一次
j=ne[j];
}
}
return 0;
}