给你两个字符串 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 。
class Solution {
public:
vector<int> next;
void GetNext(string& s)
{
int j = 0;
next[0] = 0;
for(int i = 1; i < s.size(); i++)
{
while(j>0 && s[i]!=s[j])
j = next[j-1];
if(s[i] == s[j])
j++;
next[i] = j;
}
}
int strStr(string haystack, string needle) {
next.resize(needle.size(), 0);
GetNext(needle);
for(int i = 0, j = 0; i < haystack.size(); i++)
{
while(j>0 && haystack[i] != needle[j])
j = next[j-1];
if(haystack[i] == needle[j])
j++;
if(j == needle.size())
return i-needle.size()+1;
}
return -1;
}
};