KMP算法

125 阅读1分钟

前言

kmp算法是一个字符串匹配算法,由于做过好几次,每次都会忘,因此做一个记录。

leetcode题目链接实现strStr()

KMP

关键在于计算待匹配字符串的前缀数组。

个人感觉这篇文章写的比较好【宫水三叶】简单题学 KMP 算法

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

    int strStr(string haystack, string needle) {
        if (needle.size() == 0) {
            return 0;
        }

        vector<int> next(needle.size(), 0);
        getNext(next, needle);

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