【LeetCode】找出字符串中第一个匹配项的下标 [M]（KMP算法）

28. 找出字符串中第一个匹配项的下标 - 力扣（LeetCode）

一、题目

给你两个字符串 haystack 和 needle ，请你在 haystack 字符串中找出 needle 字符串的第一个匹配项的下标（下标从 0 开始）。如果 needle 不是 haystack 的一部分，则返回  -1 。

示例 1：
输入：haystack = "sadbutsad", needle = "sad"
输出：0
解释："sad" 在下标 0 和 6 处匹配。
第一个匹配项的下标是 0 ，所以返回 0 。

示例 2：
输入：haystack = "leetcode", needle = "leeto"
输出：-1
解释："leeto" 没有在 "leetcode" 中出现，所以返回 -1 。

提示：

• 1 <= haystack.length, needle.length <= 104
• haystack 和 needle 仅由小写英文字符组成

二、代码

class Solution {
    public int strStr(String haystack, String needle) {
        // 过滤无效参数
        if (haystack == null || haystack.length() == 0 || needle == null || needle.length() == 0) {
            return -1;
        }
        // 将字符串转换为字符数组
        char[] s1 = haystack.toCharArray();
        char[] s2 = needle.toCharArray();
        // x表示此时s1已经遍历比较到哪个位置了
        int x = 0;
        // y表示此时s2已经遍历比较到哪个位置了
        int y = 0;
        // 快速计算s2的next[]数组
        // O(M) m <= n
        int[] next = getNextArray(s2);

        /**
         * 开始比较匹配s1和s2，循环结束条件就是只要有一个字符串遍历完了就结束
         * y越界：s1的某一个开头撸过了所有s2，从x出发往左推y的长度。这就此时在s1中找到了一个和s2相等的子串。
         * x越界：说明尝试所有s1的开头，都没有搞出s2，说明s1中就不存在和s2相等的子串。
         */
        // O(N)
        while (x < s1.length && y < s2.length) {
            // 如果两个位置相等，则两个字符串都向后移动一格位置，去比较下一个位置的字符相不相等。
            if (s1[x] == s2[y]) {
                x++;
                y++;
            // 如果此时s2已经右推到最开头了，无法再右推了，但是此时还是没找到能够匹配上的子串，就说明此时以s1的x位置的字符不可能匹配出s2子串了，所以直接将s1向后右移一个位置，
            // 去检查以下一个位置为开头能不能匹配出s2。此时y也正好回到了下标0位置，也就是s2也从头开始匹配了。
            } else if (next[y] == -1){
                x++;
            // 发现x和y位置的字符不相等，并且s2还可以右推，通过将y = next[y]与s1的x对齐来实现s2的右推
            } else {
                y = next[y];
            }
        }

        // 如果是y越界，说明在s1中找到了一个和s2相等的子串。字串的第一个位置的下标就是x - y。x此时的含义是s1中能匹配到的s2子串的最后一个字符的下标，y表示s2子串的长度，所以x - y得到的就是s1中能匹配到的s2子串的第一个位置下标。
        // 如果是x越界，说明没有找到相等子串，直接返回-1
        return y == s2.length ? x - y : -1;
    }

    // 快速计算s2的next[]数组
    // 这个是利用前面位置已经求得的next信息来加速求得后面位置的next信息
    public int[] getNextArray(char[] s2) {
        // s2长度为1，直接返回-1
        if (s2.length == 1) {
            return new int[] {-1};
        }
        // 创建next数组
        int[] next = new int[s2.length];
        // next数组的第一个和第二个位置的值都是固定的。
        next[0] = -1;
        next[1] = 0;
        // 目前在哪个位置上求next数组的值
        int i = 2;
        // 当前是哪个位置的值在和i-1位置的字符比较
        int cn = 0;
        // 开始求next数组信息
        while (i < next.length) {
            // 配成功的时候，此时cn的值也是i - 1的最长相等前缀的长度，也就是next[i - 1] = cn
            // 匹配成功之后next[i] = next[i - 1] + 1
            if (s2[i - 1] == s2[cn]) {
                next[i++] = ++cn;
            // 如果没有匹配上，就将cn继续前跳，跳到以前cn前面最长相等前缀的后面一个位置
            } else if (cn > 0) {
                cn = next[cn];
            // 如果cn已经等于-1了，说明此时cn已经跳到s2的第一个位置了，并且此时str2[i - 1]和str2[cn]还是没有匹配上字符
            // 说明此时i位置前面就没有相等的前缀和后缀了，直接将该位置的next信息设置为0
            } else {
                next[i++] = 0;
            }
        }

        return next;
    }
}

三、解题思路 

这个题就是一个最直接的KMP算法。时间复杂度为O(N)。