Implement strStr()

每日算法——leetcode系列

问题 Implement strStr()

Difficulty: Easy

Implement strStr().

Returns the index of the first occurrence of needle in haystack, or -1 if needle is not part of haystack.

class Solution {
public:
int strStr(string haystack, string needle) {

}
};

翻译

实现strStr()

难度系数：简单

实现strStr()。

返回匹配时的第一个索引， 如果没有匹配的就返回-1。（感觉原文用针和草堆来形容带诙谐）

思路

strstr

经典题。

假设：

遍历到的needle索引为j, haystack索引为i+j, needle,haystack长度分别为m,n

- 暴力法

遍历haystack和needle，如果haystack[i+j] == needle[j]（匹配）， 则 j++;

如果不等于（失配），i++, j = 0 T(O) = m * n

- KMP

这个得专门写一篇总结的文章。

还有Linux的grep, BM算法

代码

class Solution {
public:
int strStr(string haystack, string needle) {
return strStrKMP(haystack, needle);
}
private:
// brute-force
int strStrBF(string haystack, string needle) {

if (needle.empty()){
return 0;
}
int i = 0;
int hSize =  (int)(haystack.size());
int nSize = (int)(needle.size());
if (hSize < nSize){
return -1;
}
while(i < hSize){
int j = 0;
if (haystack[i + j] == needle[j]){
j++;
}else{
i++;
j = 0;
}
if (j >= nSize){
return i;
}
}
return -1;
}

// KMP
int strStrKMP(string haystack, string needle) {

if (needle.empty()){
return 0;
}
int i = 0;
int hSize = static_cast<int>(haystack.size());
int nSize = static_cast<int>(needle.size());
if (hSize < nSize){
return -1;
}
vector<int> next(nSize, -1);
calcNext(needle, next);
int j = 0;
while (i < hSize) {
if (j == -1 || haystack[i] == needle[j]){
i++;
j++;
}else{
j = next[j];
}
if (j >= nSize){
return i - nSize;
}
}
return -1;
}

void calcNext(const string& needle, vector<int> &next){
int nSize = static_cast<int>(needle.size());
int i = 0;
int j = -1;
while (i < nSize - 1) {
if (j == -1 || needle[i] == needle[j]){
i++;
j++;
next[i] = j;
}else{
j = next[j];
}
}
}
};