Leetcode 28. 实现 strStr()

28. 实现 strStr()
实现 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 仅由小写英文字符组成

/*
第一个：KMP算法（字符串匹配）
重点：由于j有可能为-1，而needle.length()返回的是一个无符号整数，所以在比较时需要将无符号整数强制转为有符号数
而且因为有-1，所以在if(k == -1 || needle[j] == needle[k])
if(j == -1 || haystack[i] == needle[j])
中要注意把k==-1和j == -1 放在前面，否则就会出现数组下标为负数的错误
*/
class Solution {
public:
    void getnext(int next[],string needle){
        int j = 0;
        int k = -1;
        next[0] = -1;
        while(j < needle.length() - 1){
            if(k == -1 || needle[j] == needle[k]){
                if(needle[++k] == needle[++j]){
                    next[j] = next[k];
                }
                else next[j] = k;
            }
            else k = next[k];
        }
    } 

    int strStr(string haystack, string needle) {
        if(needle.empty()){
            return 0;
        }
        int next[needle.length()];
        getnext(next,needle);
        int i = 0,j = 0;
        while(i < (int)haystack.length() && j < (int)needle.length()){
            if(j == -1 || haystack[i] == needle[j]){
                ++i;
                ++j;
            }
            else j = next[j];
        }
        if(j == needle.length()){
            return i - j;
        }
        else return -1;
    }
};

/*
第二个：string.find()
*/
class Solution {
public:
    int strStr(string haystack, string needle) {
        return haystack.find(needle);
    }
};