KMP算法笔记

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;
}