28、实现strStr()

实现 strStr() 函数。

给你两个字符串 haystack 和 needle ，请你在 haystack 字符串中找出 needle 字符串出现的第一个位置（下标从 0 开始）。如果不存在，则返回 -1 。

说明：
当 needle 是空字符串时，我们应当返回什么值呢？这是一个在面试中很好的问题。
对于本题而言，当 needle 是空字符串时我们应当返回 0 。这与 C 语言的 strstr() 以及 Java 的 indexOf() 定义相符。

来源：力扣（LeetCode）
题目链接：https://leetcode-cn.com/problems/implement-strstr
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题方法一：暴力枚举

class Solution {
public:
    int strStr(string haystack, string needle) {
        int n = haystack.size();
        int m = needle.size();
        if(m == 0) return 0;
        int i = 0, j = 0;
        while(i < n){
            if(haystack[i] == needle[j]){
                if(j == m - 1) return i - j;
                ++i;
                ++j;
            }
            else{
                i = i - j + 1;
                j = 0;  
            }
        }
        return -1;
    }
};

方法二：KMP算法

class Solution {
public:
    int strStr(string haystack, string needle) {
        int n = haystack.size();
        int m = needle.size();
        if(m == 0) return 0;
        
        vector<int> needleNext(m);
        for(int i = 1, j = 0; i < m; ++i){
            while(j > 0 && needle[i] != needle[j]){
                j = needleNext[j - 1];
            }
            if(needle[i] == needle[j]) ++j;
            needleNext[i] = j;
        }

        for(int i = 0, j = 0; i < n; ++i){
            while(j > 0 && haystack[i] != needle[j]){
                j = needleNext[j - 1];
            }
            if(haystack[i] == needle[j]){
                if(j == m - 1) return i - j;
                ++j;
            }
        }
        return -1;
    }
};