- 实现 strStr() 函数。
给你两个字符串 haystack 和 needle ,请你在 haystack 字符串中找出 needle 字符串出现的第一个位置(下标从 0 开始)。如果不存在,则返回 -1 。
说明:
当 needle 是空字符串时,我们应当返回什么值呢?这是一个在面试中很好的问题。
对于本题而言,当 needle 是空字符串时我们应当返回 0 。这与 C 语言的 strstr() 以及 Java 的 indexOf() 定义相符。
示例 1:
输入:haystack = “hello”, needle = “ll”
输出:2
示例 2:
输入:haystack = “aaaaa”, needle = “bba”
输出:-1
示例 3:
输入:haystack = “”, needle = “”
输出:0
提示:
0 <= haystack.length, needle.length <= 5 * 104
haystack 和 needle 仅由小写英文字符组成
方法一:暴力求解
class Solution {
public:
int strStr(string haystack, string needle) {
int m = needle.size();
if (m==0) return 0;
int n = haystack.size();
for (int i=0;i<n-m+1;i++){
bool flag = false;
for (int j=0; j<m ;j++){
if (haystack[i+j]==needle[j]){
flag = true;
}
else{
flag = false;
break;
}
}
if (flag){
return i;
}
}
return -1;
}
};
- 时间复杂度O(n*m)
- 空间复杂度O(1)
方法二:KMP
class Solution {
public:
void getNext(string & needle, int *next){
int m = needle.size();
int j = 0;
next[0] = 0;
for(int i=1; i<m;i++){
while(j>0 && needle[i]!=needle[j]){
j = next[j-1];//回退到最长公共前后缀的最长公共前后缀的前缀结尾
}
if (needle[i]==needle[j]){
j++;
}
next[i] = j;
}
}
int strStr(string haystack, string needle) {
int m = needle.size();
if (m==0) return 0;
int n = haystack.size();
int next[m];
getNext(needle, next);
int j = 0;
for(int i=0;i<n;i++){
while(j>0 && haystack[i]!=needle[j]){
j = next[j-1];
}
if (haystack[i]==needle[j])
j++;
if (j==m){
return i-j+1;
}
}
return -1;
}
};
注意点:
- j:表示当前字串s[:i]的最长公共前后缀的前缀的结尾的下一个字符的下标。也是当前字串s[:i]的最长公共前后缀的长度。
- j=next[j-1]:当不匹配的时候需要将j回退到s[:i]字串的最长公共前后缀的最长公共前后缀出,这样才能保证字符s[i]前由与模板前缀的公共区域。
- 时间复杂度:O(n+m)
- 空间复杂度:O(n)