LeetCode 28 . 实现 strstr() (KMP算法)

是 The Lin 呀

已于 2022-03-17 14:18:34 修改

阅读量220

点赞数 1

分类专栏： LeetCode 文章标签：算法 leetcode c++

于 2022-03-11 11:52:49 首次发布

本文链接：https://blog.csdn.net/weixin_44798320/article/details/123420768

版权

LeetCode 专栏收录该内容

18 篇文章 0 订阅

订阅专栏

该博客介绍了如何使用KMP算法在给定字符串haystack中查找子串needle的首次出现位置。首先，通过构建前缀数组next来优化匹配过程，然后遍历haystack并利用next数组进行查找。当needle为空字符串时，返回0，符合C语言的strstr()和Java的indexOf()规定。示例展示了不同情况下的输出结果。

摘要由CSDN通过智能技术生成

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、解题思路

KMP算法
核心思想：定义和字串长度相同的前缀数组
例如：
字串：
abcabcd
对应的前缀数组为：
0001230

class Solution {
public:
    int strStr(string haystack, string needle) {
        int haystack_len = haystack.length(), needle_len = needle.length();
        if( needle_len == 0){
            return 0;
        }
        //创建一个 和子串同样长度的 前缀数组
        vector<int> next(needle_len);
        //获取 前缀数组的值
        for(int i = 1, j = 0 ; i < needle_len; ++i){
            while(j > 0 && needle[j] != needle[i]){
                j = next[j-1];
            }
            if(needle[j] == needle[i]){
                j++;
            }
            next[i] = j;
        }
        //通过前缀数组，完成算法
        for(int i = 0 , j = 0; i<haystack_len; ++i){
            while(j > 0 && needle[j] != haystack[i]){
                j = next[j - 1];
            }
            if(needle[j] == haystack[i]){
                j++;
            }
            if(j == needle_len){
                return i - j + 1;
            }
        }
        return -1;
    }
};