题目描述
字符串匹配, 子串在母串中哪些位置上匹配成功, 输出母串中每个成功匹配的开始位置.
核心
- 构造ne数组, ne数组是子串中当前位置的后缀与前缀的最短匹配距离
- 每次匹配失败后, 找到子串当前位置的前缀继续匹配(相比暴力的从下一个位置开始匹配优化)
样例
输入样例:
3
aba
5
ababa
输出样例:
0 2
算法1
(kmp) $O(n+m)$
C++
y总代码(理解背下就行)
#include <iostream>
#include <algorithm>
using namespace std;
int n, m;
const int N = 100010, M = 1000010;
char p[N], s[M];
int ne[N];
int main()
{
cin >> n >> p + 1 >> m >> s + 1;
//构造ne数组的过程
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)
{
cout << i - n << " ";
j = ne[j];
}
}
return 0;
}