【LeetCode每日一题】28. 实现 strStr()

最新推荐文章于 2023-07-25 10:18:42 发布

guangcheng0312q

最新推荐文章于 2023-07-25 10:18:42 发布

阅读量250

点赞数

文章标签：字符串 leetcode html nginx 哈希表

本文链接：https://blog.csdn.net/guangcheng0312q/article/details/115922564

版权

【LeetCode每日一题】28. 实现 strStr()

今日题目是一个比较经典的字符串与模式串匹配问题，KMP，而在Linux之类的系统中也经常有strStr等字符串函数，当然C、Java、Python也有类似函数的实现，不同语言实现不一样，一定要注意处理模式串为0的情况。

题目：

实现 strStr() 函数。

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

示例 1：

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

题解：

直接暴力法

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

KMP

dp[状态][字符] = 下一个状态

采用状态机解决KMP匹配问题，需要注意模式串长度为0的情况。

class Solution {
public:
    vector<vector<int>> dp;

    void kmp(string& needle) {
        int m = needle.size();
        dp = vector<vector<int>>(m, vector<int>(256,0));
        int prev = 0;
        dp[0][needle[0]] = 1;
        for (int j = 1; j < m; j++) {
            for (int c = 0; c < 256; c++) {
                if (c == needle[j]) {
                    dp[j][c] = j + 1;
                } else {
                    dp[j][c] = dp[prev][c];
                }
            }
            prev = dp[prev][needle[j]];
        }
    }
    int strStr(string haystack, string needle) {
        int n = haystack.size(), m = needle.size();
        if (m == 0) return 0;
        kmp(needle);
        int prev = 0;
        for (int i = 0; i < n; i++) {
            prev = dp[prev][haystack[i]];
            if (prev == m) return i - m + 1;
        }

        return -1;
    }
};