题目
实现 strStr() 函数。
给定一个 haystack 字符串和一个 needle 字符串,在 haystack 字符串中找出 needle 字符串出现的第一个位置 (从0开始)。如果不存在,则返回 -1。
示例 1:
输入: haystack = “hello”, needle = “ll”
输出: 2
示例 2:
输入: haystack = “aaaaa”, needle = “bba”
输出: -1
说明:
当 needle 是空字符串时,我们应当返回什么值呢?这是一个在面试中很好的问题。
对于本题而言,当 needle 是空字符串时我们应当返回 0 。这与C语言的 strstr() 以及 Java的 indexOf() 定义相符。
法一:超时
class Solution {
public:
int strStr(string haystack, string needle) {
if(needle=="")return 0;
bool found=false;
int hindex=0,nindex=0,hlen=haystack.length(),nlen=needle.length();
int index=hindex;
while(!found&&hindex<hlen&&nindex<nlen){
while(hindex<hlen&&nindex<nlen&&haystack[hindex]==needle[nindex]){
hindex++;
nindex++;
}
if(nindex==nlen)found=true;
else {
index++;
hindex=index;
nindex=0;
}
}
if(found)return index;
return -1;
}
};
法二:kmp
前缀:不包含尾字母,后缀:不包含首字母。前缀表:该字符结尾的字符串的最长相等前后缀。例如aabaaf的前缀表值为0,1,0,1,2,0。若在f不匹配,显然f前的aa是匹配的,那么下次我们可以把串从b处开始比较,因为前缀后缀aa一样。前缀表就是要求的next数组,b处就是f前的a的前缀表值2,也就是说下一个比较位置在不匹配位置的前一个前缀表值。
class Solution {
public:
void getNext(int* next, const string& s) {
int j = 0;//j表示前缀末尾位置,i表示后缀末尾位置
next[0] = 0;
//可以理解为0~j和0~i两个串匹配
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; //此时0~j和i-j~i是相等前后缀,j就是其长度
}
}
int strStr(string haystack, string needle) {
if (needle.size() == 0) return 0;
int next[needle.size()];
getNext(next, needle);
int j = 0;
for (int i = 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;
}
};