LeetCode刷题007 找出字符串中第一个匹配项的下标

找出字符串中第一个匹配项的下标

题目

给你两个字符串 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 <= 10^4
haystack 和 needle 仅由小写英文字符组成

题解

KMP算法

Code

class Solution {

    public static final int N = 10010;
    public static int[] ne = new int[N];

    public int strStr(String haystack, String needle) {
        ne[0] = -1;
        for(int i = 1, j = -1; i < needle.length(); i++){
            while(j >= 0 && needle.charAt(i) != needle.charAt(j + 1)) j = ne[j];
            if(needle.charAt(i) == needle.charAt(j + 1)) j++;
            ne[i] = j;
        }
        for(int i = 0, j = -1; i < haystack.length(); i++){
            while(j >= 0 && haystack.charAt(i) != needle.charAt(j + 1)) j = ne[j];
            if(haystack.charAt(i) == needle.charAt(j + 1)) j++;
            if(j == needle.length() - 1){
                return i - needle.length() + 1;
            }
        }
        return -1;
    }
}