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