28. 实现 strStr()
难度简单832
实现 strStr() 函数。
给你两个字符串 haystack和needle,请你在haystack字符串中找出needle字符串出现的第一个位置(下标从 0 开始)。如果不存在,则返回 -1。
说明:
当 needle 是空字符串时,我们应当返回什么值呢?这是一个在面试中很好的问题。
对于本题而言,当 needle 是空字符串时我们应当返回 0 。这与 C 语言的 strstr() 以及 Java 的 indexOf()) 定义相符。
示例 1:
输入:haystack = “hello”, needle = “ll”
输出:2
示例 2:
输入:haystack = “aaaaa”, needle = “bba”
输出:-1
示例 3:
输入:haystack = “”, needle = “”
输出:0
提示:
- 0 <= haystack.length, needle.length <= 5 * 104
- haystack和needle仅由小写英文字符组成
class Solution {
public:
int strStr(string haystack, string needle) {
int len1 = haystack.size(), len2 = needle.size();
if(!len2) return 0;
vector<int> next(len2);
//预处理next
for(int i = 1, j = 0; i < len2; ++ i){
while(j > 0 && needle[i] != needle[j]){
j = next[j - 1];
}
if(needle[i] == needle[j]){
j ++;
}
next[i] = j;
}
//进行匹配
for(int i = 0, j = 0; i < len1; ++ i){
while(j > 0 && haystack[i] != needle[j]){
j = next[j - 1];
}
if(haystack[i] == needle[j]){
j ++;
}
if(j == len2){
return i - len2 + 1;
}
}
return -1;
}
};