28. 实现 strStr()（KMP算法实现）

Posted 2020-12-19 phun19

实现 strStr() 函数。

给定一个 haystack 字符串和一个 needle 字符串，在 haystack 字符串中找出 needle 字符串出现的第一个位置 (从0开始)。如果不存在，则返回 -1。

示例 1:

输入: haystack = "hello", needle = "ll"
输出: 2

示例 2:

输入: haystack = "aaaaa", needle = "bba"
输出: -1

KMP实现（速度非常慢）

class Solution {
    public int strStr(String haystack, String needle) {
        if(needle.length() == 0)
            return 0;
        int[] next = buildNext(needle);
        int m = next.length, i = 0;
        int n = haystack.length(), j = 0;
        while(j < n && i < m) {
            if(0 > i || haystack.charAt(j) == needle.charAt(i)) {
                i++;
                j++;
            } else {
                i = next[i];
            }
        }
        return i == m ? j - i : -1;

    }
    
    public int[] buildNext(String needle) {
        int len = needle.length(), j = 0;
        int[] next = new int[len];
        int t = next[j] = -1;
        while(j < len - 1) {
            if(0 > t || needle.charAt(j) == needle.charAt(t)) {
                j++;
                t++;
                next[j] = t;
            } else {
                t = next[t];
            }
        }
        return next;
    }
}

　　

class Solution {
    public int strStr(String haystack, String needle) {
        if(needle == null || needle.length() == 0)
            return 0;
        int len_hs  = haystack.length();
        int len_n   = needle.length();
        int index = 0;
        int begin = 0;
        while(index < len_hs) {
           if(haystack.charAt(index) == needle.charAt(begin)) {
               index++;
               begin++;
           } else {
               if(begin > 0) {
                   index = index - begin;
                   begin = 0;
               }
               index++;
           }
           if(begin == len_n)
               return index - len_n;
        }
        return -1;
    }
}

　　

说明:

当 needle 是空字符串时，我们应当返回什么值呢？这是一个在面试中很好的问题。

对于本题而言，当 needle 是空字符串时我们应当返回 0 。这与C语言的 strstr() 以及 Java的 indexOf() 定义相符。

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