[Acwing算法基础] 2.5 KMP算法

kmp思想

这个视频讲的很清楚。
程序员吴师兄的视频讲解

一个例子

核心代码

1. 求next数组：

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

2. kmp 匹配过程

// kmp 匹配过程
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);
        j = ne[j];
    }
}

完整代码

/*
关键在于找出最大的后缀和前缀相等

kmp的匹配过程和
求next数组的过程代码几乎一样
*/

#include <iostream>

using namespace std;

const int N = 10010, M = 100010;

int n, m;
char p[N], s[M];
int ne[N];

int main()
{
    cin >> n >> p + 1 >> m >> s + 1;
    
    // 求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;
    }
    
    
    // kmp 匹配过程
    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);
            j = ne[j];
        }
    }
    
    return 0;
}