美文网首页
利用KMP算法解决LeetCode第28题:实现strStr()

利用KMP算法解决LeetCode第28题:实现strStr()

作者: MrFengZH | 来源:发表于2019-05-06 14:28 被阅读0次

    简介

    • KMP算法是一种字符串匹配算法,由D.E.Knuth,J.H.Morris和V.R.Pratt同时发现,因此人们称它为克努特——莫里斯——普拉特操作(简称KMP算法)。KMP算法的关键是利用匹配失败后的信息,尽量减少模式串与主串的匹配次数以达到快速匹配的目的。具体实现就是实现一个next()函数,函数本身包含了模式串的局部匹配信息。时间复杂度O(m+n)。

    算法分析

    假设主串T用i指针遍历,而模式串P用j指针遍历。

    和暴力法的区别

    情况1:如果P[0]就不匹配了,那么i指针指向下一位,而j指针不变,这在暴力和KMP都一样,不用多说。
    情况2:如果P的前几个字符匹配的话,情况就有所不同。在暴力算法中,如果T[i] != P[j],i需要回溯到i-j+1的位置,而j变为0。而在KMP算法中,则是利用已经部分匹配的有效信息,i指针不回溯,通过修改j指针,让模式串移动到有效的位置

    重点和难点

    KMP算法的重点和难点是如何求j指针在不匹配时的下一位置,这里将该位置设为k。这里引入一个next数组,next[j]表示P[j]匹配失败后,j指针应指向的位置,即next[j] = k。

    分析

    为什么要求k值呢?k值是怎么来的?

    当T[i] != P[j]时(j>0),已知P[0 ... j-1] == T[i-j ... i-1],若P[0 ... k-1] == P[j-k ... j-1],
    则P[0 ... k-1] == T[i-k ... i-1],此时只需将j指针移动到k,而i指针不需要移动,即可继续进行比较。
    所以此时问题转化为求令P[0 ... k-1] == P[j-k ... j-1]的k的最大值,即P的前j个字符组成的子串的最长相同前缀后缀的长度(k)

    next数组如何求

    • 特殊情况:next[0] = -1(此时j不用动,i向后移一位);next[1] = 0(此时i不用动,j回溯到第0位);如果前j个字符都相同,则next[j] = j-1。
    • 其他情况:next[j] = k,其中k为P[0 ... j-1]中最长相同前缀后缀的长度

    实战:实现strStr()(LeetCode第28题)

    题目描述

    给定一个 haystack 字符串和一个 needle 字符串,在 haystack 字符串中找出 needle 字符串出现的第一个位置 (从0开始)。如果不存在,则返回 -1。当 needle 是空字符串时我们应当返回 0 。这与C语言的 strstr() 以及 Java的 indexOf() 定义相符。

    示例

    示例1:

    输入: haystack = "hello", needle = "ll"
    输出: 2
    

    示例2:

    输入: haystack = "aaaaa", needle = "bba"
    输出: -1
    

    代码

        /**
         * 利用KMP算法求解
         *
         * 需要进行匹配的字符(即haystack字符串),称为主串,简称T
         * 模式(Pattern)字符串(即needle字符串),简称P
         *
         */
        private static int strStr_KMP(String haystack, String needle) {
            if (needle.equals("")) {
                return 0;
            }
    
            char [] T = haystack.toCharArray();
            char [] P = needle.toCharArray();
            int [] next = getNext(needle);
            int i = 0;
            int j = 0;
    
            while (i < haystack.length() && j < needle.length()) {
                if (j == -1 || T[i] == P[j]) {  //当j == -1时,i向后移动一位,j也要归零
                    i++;
                    j++;
                } else {    //T[i] != P[j]时,i指针不用动,j指针移动到相应的位置
                    j = next[j];
                }
            }
    
            if (j == needle.length()) { //说明在主串T中找到了模式串P
                return i - j;
            } else {
                return -1;
            }
        }
    
        /**
         * 根据模式串P获取next数组,next[j]表示P[j]匹配失败后,j指针应指向的位置。
         *
         * @param P 模式串
         * @return next数组
         */
        private static int [] getNext(String P) {
            int [] next = new int[P.length()];
            next[0] = -1;
            int k = -1;     //存储当前next[j]对应的k值
    
            int j = 0;
            while (j < P.length()-1) {
                if (k == -1 || P.charAt(k) == P.charAt(j)) { //隐含条件P[0 ... k-1] == P[j-k ... j-1]
                    //当k == -1时,next[j++] = 0,例如next[1] = 0,或者前j字符没有相同前缀后缀时也为0。
                    //当P[k] == P[j]时,由于P[0 ... k-1] == P[j-k ... j-1],则此时P[0 ... k] == P[j-k ... j],所以此时next[j+1] = k+1
                    next[++j] = ++k;
                } else {
                    //当P[k] != P[j]时,next[j+1]必定小于k,此时可以缩小k的值(最小小到-1)
                    k = next[k];
                }
            }
    
            return next;
        }
    

    参考

    相关文章

      网友评论

          本文标题:利用KMP算法解决LeetCode第28题:实现strStr()

          本文链接:https://www.haomeiwen.com/subject/xlwroqtx.html