力扣---28. 找出字符串中第一个匹配项的下标

给你两个字符串 haystack 和 needle ，请你在 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
• haystack 和 needle 仅由小写英文字符组成

class Solution {
public:
    int strStr(string haystack, string needle) {
        // 找到needle字符串在haystack字符串中的第一个匹配项的下标
        size_t pos = haystack.find(needle);
        
        // 如果找到匹配项，返回下标，否则返回-1
        return (pos != std::string::npos) ? pos : -1;
        //std::string::npos是一个常量，表示无效定位或者未找到
    }
};

KMP

#include <iostream>
#include <vector>
#include <string>

// KMP算法的前缀函数
std::vector<int> computeLPSArray(const std::string& pattern) {
    int m = pattern.size();
    std::vector<int> lps(m, 0);
    int len = 0;
    int i = 1;

    while (i < m) {
        if (pattern[i] == pattern[len]) {
            len++;
            lps[i] = len;
            i++;
        } else {
            if (len != 0) {
                len = lps[len - 1];
            } else {
                lps[i] = 0;
                i++;
            }
        }
    }
    return lps;
}

// KMP算法实现
int strStr(const std::string& haystack, const std::string& needle) {
    if (needle.empty()) return 0;
    int n = haystack.size();
    int m = needle.size();

    if (m > n) return -1;

    std::vector<int> lps = computeLPSArray(needle);
    int i = 0, j = 0;

    while (i < n) {
        if (needle[j] == haystack[i]) {
            i++;
            j++;
        }
        if (j == m) {
            return i - j;
        } else if (i < n && needle[j] != haystack[i]) {
            if (j != 0) {
                j = lps[j - 1];
            } else {
                i++;
            }
        }
    }
    return -1;
}

int main() {
    std::string haystack = "hello";
    std::string needle = "ll";
    int result = strStr(haystack, needle);
    std::cout << "Index: " << result << std::endl; // 输出2

    haystack = "aaaaa";
    needle = "bba";
    result = strStr(haystack, needle);
    std::cout << "Index: " << result << std::endl; // 输出-1

    return 0;
}