【LeetCode每日一题】实现 strStr (KMP算法)

实现 strStr (KMP算法)

1、题目描述

实现 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

2、算法描述

核心思想：
	用一个数组来记录一下子串在匹配的时候需要回溯的地方，这样才进行模式匹配的时候可以很容易找到回溯的位置
具体实现：
	1、定义一个next数组，记录子串回溯的位置
	2、先遍历要比较的子字符串，记录一下它的next数组
	3、再对主字符串进行遍历，看具体位置然后再进行回溯
	4、考虑特殊情况，子串为空，则返回0

3、具体实现

package com.java;

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