[每日一题] 122. 实现 strStr()(字符串、KMP算法、多方法）

1. 题目来源

链接：实现 strStr()
来源：LeetCode

2. 题目说明

实现 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() 定义相符。

3. 题目解析

方法一：STL中find函数抖个机灵

参见代码如下：

// 执行用时 :0 ms, 在所有 C++ 提交中击败了100.00%的用户
// 内存消耗 :9.3 MB, 在所有 C++ 提交中击败了17.23%的用户

class Solution {
public:
    int strStr(string haystack, string needle) {
        return haystack.find(needle);
    }
};

方法二：遍历解法

这道题需要在一个字符串中找另一个字符串第一次出现的位置，那首先要做一些判断：

• 如果子字符串为空，则返回0
• 如果子字符串长度大于母字符串长度，则返回 -1
• 然后开始遍历母字符串，这里并不需要遍历整个母字符串，而是遍历到剩下的长度和子字符串相等的位置即可，这样可以提高运算效率
• 然后对于每一个字符，都遍历一遍子字符串，一个一个字符的对应比较，如果对应位置有不等的，则跳出循环，如果一直都没有跳出循环，则说明子字符串出现了，则返回起始位置即可

参见代码如下：

// 执行用时 :4 ms, 在所有 C++ 提交中击败了93.63%的用户
// 内存消耗 :9.3 MB, 在所有 C++ 提交中击败了17.69%的用户

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

方法三：优化遍历解法

针对方法二，也可以写的更加简洁一些：

• 开头直接套两个 for 循环，不写终止条件
• 然后判断假如j到达 needle 的末尾了，此时返回 i；若此时 i + j 到达 haystack 的长度了，返回 -1；否则若当前对应的字符不匹配，直接跳出当前循环

参见代码如下：

// 执行用时 :8 ms, 在所有 C++ 提交中击败了58.17%的用户
// 内存消耗 :9.2 MB, 在所有 C++ 提交中击败了29.70%的用户

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

方法四：KMP算法求解

关于KMP算法详细内容，网上一大堆相关技术博文，在此采用KMP模板进行解题。

// 执行用时 :4 ms, 在所有 C++ 提交中击败了93.63%的用户
// 内存消耗 :9.5 MB, 在所有 C++ 提交中击败了9.10%的用户

class Solution {
public:
    int strStr(string haystack, string needle) {
        return kmp(haystack, needle);
    }

    vector<int> getNext(string p) {
        int n = p.size(), k = -1, j = 0;
        vector<int> next(n, -1);
        while (j < n - 1) {
            if (k == -1 || p[j] == p[k]) {
                ++k; ++j;
                next[j] = (p[j] != p[k]) ? k : next[k];
            } else {
                k = next[k];
            }
        }
        return next;
    }
    int kmp(string s, string p) {
        int m = s.size(), n = p.size(), i = 0, j = 0;
        vector<int> next = getNext(p);
        while (i < m && j < n) {
            if (j == - 1 || s[i] == p[j]) {
                ++i; ++j;
            } else {
                j = next[j];
            }
        }
        return (j == n) ? i - j : -1;
    }
};