LeetCode:28. 找出字符串中第一个匹配项的下标(KMP算法)

发布于:2024-12-09 ⋅ 阅读:(173) ⋅ 点赞:(0)

跟着carl学算法,本系列博客仅做个人记录,建议大家都去看carl本人的博客,写的真的很好的!
代码随想录

LeetCode:28. 找出字符串中第一个匹配项的下标(KMP算法)
给你两个字符串 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 。

注意next数组的求法,注意都是先找不相等的时候然后再去判断是否相等,多注意代码注释

	public int strStr(String haystack, String needle) {
        // 求next数组 最长相等前后缀的数组
        int[] next = getNext(needle);

        int j = 0;
        for(int i = 0; i < haystack.length(); i++){
            while(j > 0 && needle.charAt(j) != haystack.charAt(i)){
                j = next[j - 1];
            }
            if(needle.charAt(j) == haystack.charAt(i)){
                j++;
            }
            // 如果j已经遍历完了 则说明在haystack中存在needle
            if(j == needle.length()){
                // 返回第一个匹配项的下标
                return i - needle.length() + 1;
            }
        }
        return -1;
    }

    private int[] getNext(String needle){
        int[] next = new int[needle.length()];
        // i之前的字符的最长相等前后缀的长度
        int j = 0;
        for(int i = 1; i < needle.length(); i++){
            // 注意这里面的 必须是先判断不等 然后再去判断相等,如果反过来就错了,比如 aabaaac;
            // 这个while最终的结果是要么 j回到0 要么找到等于i位置的字符
            while(j > 0 && needle.charAt(j) != needle.charAt(i)){
                j = next[j - 1];
            }
            // 如果找到了和i位置字符相等的字符 则最长相等前后缀需要加1,如果没找到则是0 
            if(needle.charAt(j) == needle.charAt(i)){
                j++;
            }
            next[i] = j;
        }
        return next;
    }

网站公告

今日签到

点亮在社区的每一天
去签到