实现strstr()

实现 strStr() 函数。

给定一个 haystack 字符串和一个 needle 字符串，在 haystack 字符串中找出 needle 字符串出现的第一个位置 (从0开始)。如果不存在，则返回 -1。

• 示例 1:

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

• 示例 2:

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

• 说明:

当 needle 是空字符串时，我们应当返回什么值呢？这是一个在面试中很好的问题。

对于本题而言，当 needle 是空字符串时我们应当返回 0 。这与C语言的 strstr() 以及 Java的 indexOf() 定义相符。

class Solution {
public:
    int strStr(string haystack, string needle) {
        if (haystack.size() < needle.size()) {
            return -1;
        }
        int i, j;
        for (i=0; i<haystack.size() - needle.size()+1; i++) {
            for (j=0; j<needle.size(); j++) {
                if (haystack[i+j] != needle[j]) {
                    break;
                }
            }
            if (j == needle.size()) {
                return i;
            }
        }
        return -1;
    }
};

• Sunday 算法

class Solution {
public:
    int strStr(string haystack, string needle) {
        int hsize = haystack.size();
        int nsize = needle.size();
        unordered_map<char, int> offset;
        for(int i=0; i<nsize; i++) {
            offset[needle[i]] = nsize - i;
        }
        int i=0;
        while(i<=hsize-nsize) {
            if (haystack.substr(i, nsize) == needle) {
                return i;
            } else {
                if (i+nsize > hsize-1) {
                    return -1;
                } else {
                    if (offset.find(haystack[i+nsize]) != offset.end()) {
                        i += offset[haystack[i+nsize]];
                    } else {
                        i+= nsize+1;
                    }
                }
            }
        }
        return -1;
    }
};

• KMP 算法

class Solution {
public:
    int strStr(string haystack, string needle) {
        if (needle.empty()) {
            return 0;
        }
        int hSize = haystack.size();
        int nSize = needle.size();
        vector<int> next(nSize, 0);
        get_next(next, needle);

        int i=0;
        int j=0;

        while(i<hSize) {
            if (haystack[i] == needle[j]) {
                i++;
                j++;
            } else {
                if (j == 0) {
                    i++;
                } else {
                    j=next[j-1];
                }
            }
            if (j==nSize) {
                return i-j;
            }
        }
        return -1;
    }

    void get_next(vector<int> &next, const string &needle) {
        int n = needle.size();
        int i=1, j=0;
        while(i<n) {
            if (needle[i] == needle[j]) {
                j++;
                next[i]=j;
                i++;
            } else {
                if (j == 0) {
                    next[i]=0;
                    i++;
                } else {
                    j = next[j-1];
                }
            }
        }
    }
};

参考链接

bilibili KMP讲解
来源：力扣（LeetCode）