前言
kmp算法是一个字符串匹配算法,由于做过好几次,每次都会忘,因此做一个记录。
leetcode题目链接实现strStr()。
KMP
关键在于计算待匹配字符串的前缀数组。
个人感觉这篇文章写的比较好【宫水三叶】简单题学 KMP 算法。
class Solution {
public:
void getNext(vector<int> &next, const string& s) {
int j = -1;
next[0] = j;
for (int i = 1; i < s.size(); ++i) {
while (j >= 0 && s[i] != s[j+1]) {
j = next[j];
}
if (s[i] == s[j+1]) {
j++;
}
next[i] = j;
}
}
int strStr(string haystack, string needle) {
if (needle.size() == 0) {
return 0;
}
vector<int> next(needle.size(), 0);
getNext(next, needle);
int j = -1;
for (int i = 0; i < haystack.size(); ++i) {
while(j >= 0 && haystack[i] != needle[j+1]) {
j = next[j];
}
if (haystack[i] == needle[j+1]) {
j++;
}
if (j == (needle.size()-1)) {
return i - needle.size() + 1;
}
}
return -1;
}
};