实现 strStr() 函数。
给定一个 haystack 字符串和一个 needle 字符串,在 haystack 字符串中找出 needle 字符串出现的第一个位置 (从0开始)。如果不存在,则返回 -1。
示例 1:
输入: haystack = "hello", needle = "ll" 输出: 2
示例 2:
输入: haystack = "aaaaa", needle = "bba" 输出: -1
说明:
当 needle
是空字符串时,我们应当返回什么值呢?这是一个在面试中很好的问题。
对于本题而言,当 needle
是空字符串时我们应当返回 0 。这与C语言的 strstr() 以及 Java的 indexOf() 定义相符。
class Solution {
public:
vector<int> calcNext(string needle){
vector<int> Next;
Next.push_back(-1);
int k = -1;
for(int i = 1;i<needle.size();i++){
while(k>-1 && needle[k+1] != needle[i]){
k = Next[k];
}
if(needle[k+1] == needle[i]){
k = k+1;
}
Next.push_back(k);
}
return Next;
}
int strStr(string haystack, string needle) {
int j = 0;
if(haystack.size() < needle.size()) return -1;
if(needle =="" ){
return 0;
}else if(haystack == ""){
return -1;
}
vector<int> Next = calcNext(needle);
int k = -1;
for(int i = 0; i < haystack.size();i++){
while(k>-1 && needle[k+1] != haystack[i]){
k = Next[k];
}
if(needle[k+1] == haystack[i]) k = k+1;
if(k == needle.size()-1){
return i - needle.size()+1;
}
}
return -1;
}
};