题目:
实现 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() 定义相符。
扫描二维码关注公众号,回复:
2469768 查看本文章
解答:本人用的KMP算法实现,关于KMP,这里有一篇文章写得不错:
https://blog.csdn.net/v_july_v/article/details/7041827
代码如下:
class Solution {
public:
int strStr(string haystack, string needle) {
if (needle.empty()) {
return 0;
}
int i = 0, j = 0;
int tlen = haystack.length(); //text length
int plen = needle.length(); //pattern length
int *next = new int[plen];
getNext(needle, next);
while (i < tlen && j < plen) {
if (j == -1 || haystack[i] == needle[j]) {
i++;
j++;
}
else {
j = next[j];
}
}
delete next;
next = nullptr;
if (j == plen) {
return i - j;
}
else {
return -1;
}
}
void getNext(string pattern, int* next) {
//known next[j], get next[j + 1]
int k = -1, j = 0;
next[0] = -1;
while (j < pattern.length() - 1) {
if (k == -1 || pattern[k] == pattern[j]) {
k++;
j++;
if (pattern[j] != pattern[k]) {
next[j] = k;
}
else {
next[j] = next[k];
}
}
else {
k = next[k];
}
}
}
};