参考
练习地址:leetcode28题 https://leetcode-cn.com/problems/implement-strstr/
KMP算法是一个快速查找匹配串的算法,主要用于快速从原字符串找到匹配字符串。
1. KMP匹配过程:
- 匹配串检查之前已经匹配成功的部分是否存在相同的前缀和后缀,如果存在,则跳转到前缀的下一个位置继续匹配
- 跳转到下一个匹配位置后,尝试匹配,发现两个指针的字符对不上,并且此时匹配串指针前面不存在相同的前缀和后缀,这时候只能回到匹配串的起始位置重新开始:
2. 分析实现
显然扫描完成原串的这一操作是不可避免的,我们可以优化的只能是检查已匹配部分的相同前缀和后缀这一过程。
再进一步,我们检查匹配串的任意一个位置而言,由该位置发起的下一个匹配点位置其实与原串无关。
对于匹配串 abcabd 的字符 d 而言,由它发起的下一个匹配点跳转必然是字符 c 的位置。因为字符 d 位置的相同「前缀」和「后缀」字符 ab 的下一位置就是字符 c。
可见从匹配串某个位置跳转到下一个匹配位置这一过程是与原串无关的,我们将这一过程称为找next点
我们可以预处理next数组,数组中每个位置的值就是该下标应该跳转的目标位置(next点)
3. next数组的构建
- 假设有匹配串aaabbab,我们来看看对应的next是如何被构建的。
4. 代码实现
Q: 给你两个字符串 haystack 和 needle ,请你在 haystack 字符串中找出 needle 字符串的第一个匹配项的下标(下标从 0 开始)。如果 needle 不是 haystack 的一部分,则返回 -1 。
class Solution {
public:
int strStr(string haystack, string needle) {
int n = haystack.size(), m = needle.size();
if (m == 0) {
return 0;
}
// 构建 next 数组,数组长度为匹配串的长度(next 数组是和匹配串相关的)
vector<int> next(m);
// 构造过程 i = 1,j = 0 开始,i 小于匹配串长度
for (int i = 1, j = 0; i < m; i++) {
// 匹配不成功的话,j = next(j-1),j指针指向前一个位置的next数组所对应的位置,
// 这个过程循环进行,直到haystack[i]==haystack[j],或者j == 0
while (j > 0 && needle[i] != needle[j]) {
j = next[j - 1];
}
// 匹配成功的话,next[i] = j + 1 ,(这里先让j++)
if (needle[i] == needle[j]) {
j++;
}
// 更新 next[i],结束本次循环,i++
next[i] = j;
}
// 匹配过程,i = 0,j = 0 开始,i 小于原串长度
for (int i = 0, j = 0; i < n; i++) {
// 匹配不成功 j = next(j-1)
while (j > 0 && haystack[i] != needle[j]) {
j = next[j - 1];
}
// 匹配成功的话,先让 j++,结束本次循环后 i++
if (haystack[i] == needle[j]) {
j++;
}
// 整一段匹配成功,直接返回下标
if (j == m) {
return i - m + 1;
}
}
return -1;
}
};