AcWing 算法基础课 KMP

KMP用于o(n)的字符串匹配。

使用next数组记录以当前阶段为后缀的，和以开始位置为前缀匹配的最长长度，即匹配过程中进行后移后保留的长度。

next数组的计算和匹配过程类似。

推荐数组从1开始计数

注意，自身match时，是不包含当前位置的前缀，而互相match时，是包含当前位置的前缀，所以两次循环的起始不同。

自写：

bool KMP(string &str1, string &str2)
{
    int n1 = str1.size();
    int n2 = str2.size();
    vector<int> match_self(n2 + 1, 0);
    for (int i = 2; i <= n2; i++)
    {
        int prev = match_self[i - 1];
        while (prev&&str2[i - 1] != str2[prev]) prev = match_self[prev];
        if (str2[i - 1] == str2[prev])
            match_self[i] = prev + 1;
    }
    vector<int> match(n1 + 1, 0);
    for (int i = 1; i <= n1; i++)
    {
        int prev = match[i - 1];
        while (prev&&str1[i - 1] != str2[prev]) prev = match_self[prev];
        if (str1[i - 1] == str2[prev])
            match[i] = prev + 1;
        if (match[i] == n2)
            return true;

    }
    return false;
}

 

　　

AcWing 831 KMP字符串

#include<iostream>
using namespace std;
const int N = 10010, M = 100010;
char p[N], s[M];
int ne[N];
int main()
{
int n, m;
cin >> n >> p + 1 >> m >> s+1;
for (int i = 2,j=0; i <= n; i++)//j代表上次匹配到的长度
{

while (j&&p[i] != p[j + 1]) j = ne[j];
if (p[i] == p[j + 1])
j++;
ne[i] = j;

}
for (int i = 1, j = 0; i <= m; i++)//j代表上次匹配到的长度
{
while (j&&s[i] != p[j + 1]) j = ne[j];
if (s[i] == p[j + 1]) j++;
if (j == n)
{
cout << i - n << ' ';
j = ne[j];
}
}
return 0;
}