leetcode Implement strStr()

原题链接：https://leetcode.com/problems/implement-strstr/

Description

Implement strStr().

Returns the index of the first occurrence of needle in haystack, or -1 if needle is not part of haystack.

kmp水一发。。

class Solution {
public:
    void get_next(string needle) {
        next.push_back(0);
        for (size_t i = 1, j = 0; i < n; i++) {
            while (j > 0 && needle[i] != needle[j]) j = next[j - 1];
            if (needle[i] == needle[j]) j++;
            next.push_back(j);
        }
    }
    int kmp(string haystack, string needle) {
        for (size_t i = 0, j = 0; i < m; i++) {
            while (j > 0 && haystack[i] != needle[j]) j = next[j - 1];
            if (haystack[i] == needle[j]) j++;
            if (j == n) return i - n + 1;
        }
        return -1;
    }
    int strStr(string haystack, string needle) {
        n = needle.length();
        m = haystack.length();
        if((!n && !m) || !n) return 0;
        if(!m && n) return -1;
        get_next(needle);
        return kmp(haystack, needle);
    }
private:
    size_t n, m;
    vector<int> next;
};