字符串——实现 strStr()（leetcode 28）

题目描述

实现 strStr() 函数。

给你两个字符串 haystack 和 needle ，请你在 haystack 字符串中找出 needle 字符串出现的第一个位置（下标从 0 开始）。如果不存在，则返回  -1 。

说明：

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

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

示例 1：

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

示例 2：

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

示例 3：

输入：haystack = "", needle = ""
输出：0

提示：

    0 <= haystack.length, needle.length <= 5 * 104
    haystack 和 needle 仅由小写英文字符组成

算法分析

https://leetcode-cn.com/problems/implement-strstr/solution/shua-chuan-lc-shuang-bai-po-su-jie-fa-km-tb86/

代码

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

时间复杂度分析

时间复杂度：O(n+m)，其中 n是字符串 haystack的长度，m是字符串 needle的长度。我们至多需要遍历两字符串一次。

空间复杂度：O(m)，其中 m是字符串 needle的长度。我们只需要保存字符串 needle的前缀函数。