leetcode 28. 实现 strStr() KMP next数组图解 (c++) 2021-07-15

最新推荐文章于 2024-03-01 07:00:00 发布

sususuqingxiao

最新推荐文章于 2024-03-01 07:00:00 发布

阅读量59

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/sususuqingxiao/article/details/118755599

版权

leetcode 专栏收录该内容

42 篇文章 1 订阅

订阅专栏

在这里插入图片描述

来源：https://www.cnblogs.com/c-cloud/p/3224788.html
在这里插入图片描述

class Solution {
public:
    int strStr(string haystack, string needle)
    {
        if (needle.size() == 0)
            return 0;

        vector<int> next(needle.size(), 0);
        getNext(needle, next);
        
        for (int i = 0, j = 0; i < haystack.size();i++)
        {        
            while ( j > 0 && haystack[i] != needle[j])
                j = next[j - 1];

            if (haystack[i] == needle[j])
                j++;

            if(j == needle.size())
                return (i + 1 - needle.size());
        }
        return -1;
    }
private:
    void getNext(string& s, vector<int>& next);
};

inline void Solution::getNext(string& s, vector<int>& next)
{
    for (int i = 1, k = 0; i < s.size(); ++i)
    {
        while(k > 0 && s[i] != s[k])
        {
            k = next[k - 1];
        }

        if (s[i] == s[k])
            k++;

        next[i] = k;
    }
    return;
}