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 * 10^4
• haystack 和 needle 仅由小写英文字符组成

解答：

class Solution {
    //KMP算法
    //时间复杂度O(M+N),空间复杂度O(N)
    public int strStr(String haystack, String needle) {
        if(needle.equals("") || needle == null){
            return 0;
        }        
        int[] next = getNext(needle);
        int hs_len = haystack.length();
        int nl_len = needle.length();
        int i = 0, j = 0;
        while(i < hs_len && j < nl_len){
            if(j == -1 || haystack.charAt(i) == needle.charAt(j)){
                i++;
                j++;
            }else{
                j = next[j];
            }
        }
        if(j == nl_len){
            return i - nl_len;
        }else{
            return -1;  
        }              
    }

    //求字符串的最长匹配前缀和后缀的next数组
    public int[] getNext(String needle){
        int len = needle.length();
        int[] next = new int[len];
        next[0] = -1;
        int i = 0, j = -1;
        while(i < len - 1){
            if(j == -1 || needle.charAt(i) == needle.charAt(j)){
                i++;
                j++;
                //对传统求next进行优化，减少跳跃
                if(needle.charAt(i) == needle.charAt(j)){
                    next[i] = next[j];
                }else{
                    next[i] = j;
                }
            }else{
                j = next[j];
            }
        }
        return next;
    }
}

参考自：

作者：LeetCode-Solution
链接：https://leetcode-cn.com/problems/implement-strstr/solution/shi-xian-strstr-by-leetcode/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。