leetcode28、找出字符串中第一个匹配项的下标

给你两个字符串 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) {
        vector<int> next(needle.size(), 0);
        for (int i = 1; i < needle.size(); i++) {
            if (needle[next[i-1]] == needle[i]) next[i] = next[i-1] + 1;
            else {
                int temp = next[i-1];
                while (needle[temp] != needle[i] && temp) temp = next[temp-1];
                if (needle[temp] == needle[i]) next[i] = temp + 1;
                else next[i] = 0;
            }
        }
        int res = -1;
        int p1 = 0, p2 = 0;
        while (p1 < haystack.size()) {
            if (haystack[p1] == needle[p2]) {
                p1++;
                p2++;
                if (p2 == needle.size()) {
                    res = p1 - p2;
                    return res;
                }
            } else if (!p2) p1++;
            else p2 = next[p2-1];
        }
        return res;
    }
};

0 ms 100%

6.5 MB 37.58%