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 仅由小写英文字符组成
代码一
纯暴力,双重循环,但双重循环10^8 ,过是能过,但时间消耗有点大
class Solution {
public:
int strStr(string haystack, string needle) {
int hlen=haystack.length();
int nlen=needle.length();
if(hlen<nlen) return -1;
for(int i=0;i<hlen;i++){
int cnt=i;
for(int j=0;j<nlen;j++){
if(needle[j]==haystack[i]){
i++;
}
else{
i=cnt;
break;
}
if(j==nlen-1) return cnt;
}
}
return -1;
}
};
代码二
kmp算法,分成两部分
第一部分是求 needle 部分的前缀函数,我们需要保留这部分的前缀函数值。
第二部分是求 haystack 部分的前缀函数,我们无需保留这部分的前缀函数值,只需要用一个变量记录上一个位置的前缀函数值即可。当某个位置的前缀函数值等于 m 时,说明我们就找到了一次字符串 needle 在字符串 haystack 中的出现(因为此时真前缀恰为字符串 needle,真后缀为以当前位置为结束位置的字符串 haystack 的子串),我们计算出起始位置,将其返回即可。
class Solution {
public:
int strStr(string haystack, string needle) {
int n=haystack.size(),m=needle.size();
//if(m==0 || n<m ) return -1;//m是空串,leetcode卡的不是很严
if(m==0) return 0;//上面两种都能过
vector<int> pi(m);
for(int i=1,j=0;i<m;i++){
while(j>0&&needle[i]!=needle[j]){
j=pi[j-1];
}
if(needle[i]==needle[j]){
j++;
}
pi[i]=j;
}
for(int i=0,j=0;i<n;i++){
while(j>0&&haystack[i]!=needle[j]){
j=pi[j-1];//记录haystack中符合needle的前缀
}
if(haystack[i]==needle[j]){
j++;
}
if(j==m) return i-m+1;
}
return -1;
}
};