KMP匹配

28. 实现 strStr()

难度简单796

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

通过次数335,627

提交次数841,797

class Solution {
    public int strStr(String haystack, String needle) {
        if (needle.length() == 0) return 0;
        int res = match(haystack, needle);
        return res;
    }

    public int match(String s, String p) {
        int[] next = buildNext(p);
        int m = s.length(), i = 0;
        int n = p.length(), j = 0;
        while (j < n && i < m) {
            if (0 > j || s.charAt(i) == p.charAt(j)) {
                i++;
                j++;
            } else {
                j = next[j];
            }
        }
        if (j != p.length()) return -1;
        return i - j;
    }

    public int[] buildNext(String p) {
        int m = p.length(), j = 0;
        int[] next = new int[m];
        int t = next[0] = -1;

        while (j < m - 1) {
            if (0 > t || p.charAt(j) == p.charAt(t)) {
                j++;
                t++;
                next[j] = t;
            } else {
                t = next[t];
            }
        }

        return next;
    }
}

赏

谢谢你请我吃糖果

微信