题目描述
给你两个字符串 haystack 和 needle ,请你在 haystack 字符串中找出 needle 字符串的第一个匹配项的下标(下标从 0 开始)。如果 needle 不是 haystack 的一部分,则返回 -1 。
示例 1:
输入:haystack = “sadbutsad”, needle = “sad”
输出:0
解释:“sad” 在下标 0 和 6 处匹配。
第一个匹配项的下标是 0 ,所以返回 0 。
示例 2:
输入:haystack = “leetcode”, needle = “leeto”
输出:-1
解释:“leeto” 没有在 “leetcode” 中出现,所以返回 -1
解题思想
使用KMP:使用子串计算一个next数组,根据next数组在主串中进行匹配。(公共前后缀的思想)
求next数组
案例1
案例2
获取next代码
void getNext(string& needle, vector<int>& next){
int j = 0;
next[0] = 0;
for(int i = 1; i < needle.size(); i++){
while(j > 0 && s[i] != s[j]){
j = next[j-1];
}
if(s[i] == s[j])
++j;
next[i] = j;
}
}
代码
class Solution {
public:
//获取next数组
void getNext(const string &s, vector<int> &next){
int j=0;
next[0]=0;
for(int i=1;i<s.size();i++){
while(j>0 && s[i]!=s[j])
j=next[j-1];
if(s[i]==s[j])
j++;
next[i]=j;
}
}
int strStr(string haystack, string needle) {
if(needle.size()==0) return 0;
int j=0;
vector<int> next(needle.size());
getNext(needle,next);
//求子串和主串匹配
for(int i=0;i<haystack.size();i++){
while(j>0&& haystack[i]!=needle[j])
j=next[j-1];
if(haystack[i]==needle[j])
j++;
if(j==needle.size())
return i-needle.size()+1;
}
return -1;
}
};