题目描述
给你两个字符串 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
代码
class Solution {
public:
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;
}
};