思路:
- 写一个next数组求模式串的最长相等前后缀。
- 主函数的写法跟next数组里面的写法相似。
代码:
class Solution {
public:
//求next数组最长相等前后缀
void getNext(int* next, const string& s) {//形参是next指针(数组不能用引用传参)和模式串(能加引用的一定要加,可以提高传送速度)
//1.初始化
int j = 0;//1.j是指向前缀末尾位置
next[0] = 0;//只有一个元素的字符串,最长相等前后缀就是0,初始化直接设为0即可
for(int i = 1; i < s.size(); i++) {//数组用for循环遍历,j从1开始,因为一个元素的字符串,最长相等前后缀就是0,不需要从0开始了
//2.前后缀不相等的时候需要不断的向后回退,所以用while不断的回退,而不是用if,并且j不能回退成负的,所以需要条件j>0
while (j > 0 && s[i] != s[j]) {
j--;//回退一次
j = next[j];//kmp算法的核心就是这段,j更新为回退后next数组的值
}
if (s[i] == s[j]) {
j++;//相等的话j就累加
}
next[i] = j;//j是指向前缀末尾的指针也是最长相等前后缀,用来更新next数组
}
}
int strStr(string haystack, string needle) {
//过关斩将,先排除模式串长度为0的情况
if (needle.size() == 0) {
return 0;
}
//在主函数中声明next数组,长度与模式串一样,在next方法里求next数组
int next[needle.size()];
getNext(next, needle);//获取next数组
int j = 0;//j是指向模式串的
//下面开始匹配,跟求next数组相似
for (int i = 0; i < haystack.size(); i++) {
while(j > 0 && haystack[i] != needle[j]) {
j--;
j = next[j];
}
if (haystack[i] == needle[j]) {
j++;
}
//匹配完时,j指向模式串的最后
if (j == needle.size() ) {
return (i - j + 1);//开始匹配的长度i-j+1,要加上1才行,这个高中做数学的时候就学过
}
}
return -1;
}
};