题目

给你两个字符串 haystackneedle ,请你在 haystack 字符串中找出 needle 字符串的第一个匹配项的下标(下标从 0 开始)。如果 needle 不是 haystack 的一部分,则返回 -1

示例 1:

输入:haystack = "sadbutsad", needle = "sad"
输出:0
解释:"sad" 在下标 0 和 6 处匹配。
第一个匹配项的下标是 0 ,所以返回 0 。

示例 2:

输入:haystack = "leetcode", needle = "leeto"
输出:-1
解释:"leeto" 没有在 "leetcode" 中出现,所以返回 -1 。

提示:

  • 1 <= haystack.length, needle.length <= 104
  • haystackneedle 仅由小写英文字符组成

解答

  1. 暴力解法

首先判断haystack的长度是否小于needle,若小于则直接返回-1.然后对haystack进行遍历,若haystack[i]==needle[0],则说明haystack同needle的第一个字符匹配了,此时直接求haystack在i上长度为needle长度的子串并比较他们是否相等,若相等则直接返回i即可,否则继续遍历haystack。C++解法如下:

class Solution {
public:
    int strStr(string haystack, string needle) {
        int m = haystack.length(); 
        int n = needle.length(); // 分别求两个字符串的长度
        if (n == 0) return 0; // 此判断符合strStr函数的原始定义
        if (m < n) return -1; // 若haystack长度小于needle长度,则直接返回-1
        for (int i = 0; i < m; i++) { // 遍历匹配第一个字母
            if (haystack[i] == needle[0]) { // 若第一个字母成功匹配
                string temp = haystack.substr(i, n); // 直接求下标为i,长度同needle相等的子串temp
                if (temp == needle) { // 判断temp与needle是否相等
                    return i; // 若相等,则直接返回i
                }
            }
        }
        return -1; // 否则返回-1
    }
};
  1. KMP算法

KMP的详细介绍:https://programmercarl.com/0028.%E5%AE%9E%E7%8E%B0strStr.html

class Solution {
public:
    void getNext(int* next, const string& s) {
        int j = -1;
        next[0] = j;
        for(int i = 1; i < s.size(); i++) { // 注意i从1开始
            while (j >= 0 && s[i] != s[j+1]) { //前后缀不相同了
                j = next[j];
            }
            if (s[i] == s[j + 1]) { // 找到相同的前后缀
                j++;
            }
            next[i] = j; // 将j(前缀的长度)赋给next[i]
        }
    }
    int strStr(string haystack, string needle) {
        if (needle.size() == 0) {
            return 0;
        }
        int next[needle.size()];
        getNext(next, needle);
        int j = -1;
        for (int i = 0; i < haystack.size(); i++) { // 注意i从0开始
            while(j >= 0 && haystack[i] != needle[j + 1]) { // 不匹配
                j = next[j]; // j寻找之前匹配的位置
            }
            if (haystack[i] == needle[j + 1]) { // 匹配,j和i同时向后移动
                j++;
            }
            if (j ==(needle.size() - 1))  { // 文本串s里出现了模式串t
                return (i - needle.size() + 1);
            }
        }
        return -1;
    }
};