Leetcode.28 找出字符串中第一个匹配项的下标（完整代码）

---

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

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

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

---

1. 核心代码

class Solution {
public:
    void getNext(int* next, const string& s)
    {
        int j = -1;
        next[0] = j;
        for (int i = 1; i < s.size(); i++)
        {
            while (j >= 0 && s[i] != s[j + 1])
                j = next[j];
            if (s[i] == s[j + 1])
                j++;
            next[i] = j;
        }
    }
    int strStr(string haystack, string needle) 
    {
        if (needle.size() == 0)
            return -1;
        //int next[needle.size()];
        int* next = new int[needle.size()];
        getNext(next, needle);
        int j = -1;
        for (int i = 0; i < haystack.size(); i++) 
        {
            while (j >= 0 && haystack[i] != needle[j + 1])
                j = next[j];
            if (haystack[i] == needle[j + 1])
                j++;
            if (j == (needle.size() - 1))
                return (i - j);
        }
        return -1;
    }
};

2. 完整代码

#include
#include
#include
#include
using namespace std;
class Solution {
public:
    void getNext(int* next, const string& s)
    {
        int j = -1;
        next[0] = j;
        for (int i = 1; i < s.size(); i++)
        {
            while (j >= 0 && s[i] != s[j + 1])
                j = next[j];
            if (s[i] == s[j + 1])
                j++;
            next[i] = j;
        }
    }
    int strStr(string haystack, string needle) 
    {
        if (needle.size() == 0)
            return -1;
        //int next[needle.size()];
        int* next = new int[needle.size()];
        getNext(next, needle);
        int j = -1;
        for (int i = 0; i < haystack.size(); i++) 
        {
            while (j >= 0 && haystack[i] != needle[j + 1])
                j = next[j];
            if (haystack[i] == needle[j + 1])
                j++;
            if (j == (needle.size() - 1))
                return (i - j);
        }
        return -1;
    }
};

int main()
{
    string s1, s2;
    getline(cin, s1);
    getline(cin, s2);
    Solution S;
    cout << S.strStr(s1, s2);
}

---

总结

kmp算法的使用，此题目需要反复刷，里面的小坑较多，比如说int数组不能用变量定义长度，需要动态分配内存空间、next数组的引入等。