28.实现strStr()

问题描述

实现 strStr() 函数。

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

示例 1:

输入: haystack = “hello”, needle = “ll”
输出: 2
示例 2:

输入: haystack = “aaaaa”, needle = “bba”
输出: -1
说明:

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

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

解决思路

使用KMP算法求解
算法讲解视频：https://www.bilibili.com/video/av3246487?from=search&seid=10466135597243491209

代码实现

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

        if(needle.isEmpty()) return 0;
        
        int[] next = prefix_table(needle);
        //字符转数组
        char[] h_arr = haystack.toCharArray();
        char[] n_arr = needle.toCharArray();

        for(int i = 0, j = 0; i < h_arr.length; i++){
            while(j > 0 && h_arr[i] != n_arr[j]){
                j = next[j - 1];
            }
            if(h_arr[i] == n_arr[j]){
                j++;
            }
            if(j == n_arr.length){
                return i-j+1;
            }
        }
        return -1;
    }
}