AcWing 算法基础课 KMP

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

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

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

推荐数组从1开始计数

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

自写：

 1 bool KMP(string &str1, string &str2)
 2 {
 3     int n1 = str1.size();
 4     int n2 = str2.size();
 5     vector<int> match_self(n2 + 1, 0);
 6     for (int i = 2; i <= n2; i++)
 7     {
 8         int prev = match_self[i - 1];
 9         while (prev&&str2[i - 1] != str2[prev]) prev = match_self[prev];
10         if (str2[i - 1] == str2[prev])
11             match_self[i] = prev + 1;
12     }
13     vector<int> match(n1 + 1, 0);
14     for (int i = 1; i <= n1; i++)
15     {
16         int prev = match[i - 1];
17         while (prev&&str1[i - 1] != str2[prev]) prev = match_self[prev];
18         if (str1[i - 1] == str2[prev])
19             match[i] = prev + 1;
20         if (match[i] == n2)
21             return true;
22 
23     }
24     return false;
25 }

View Code

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