难度:简单
实现 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() 定义相符。
题目解析:
直接暴力匹配。还有一种解法是KMP,由于较难理解,后期再优化
参考代码:
class Solution {
public:
int strStr(string haystack, string needle) {
if (!needle.size()) return 0;
if (!haystack.size()) return -1;
//先构造pattern
int j = -1, i = 0;//j在后面,i在前面
vector<int> b(needle.size() + 1);
b[i] = j;
while (i < needle.size())
{
while (j >= 0 && needle[i] != needle[j]) j = b[j];
i++, j++;
b[i] = j;
}
j = 0, i = 0; //j这回是text的, i是pattern的
while (j < haystack.size())
{
while (i >= 0 && needle[i] != haystack[j]) i = b[i];
i++, j++;
if (i == needle.size())
{
return j - needle.size();
}
}
return -1;
}
};
暴力:
class Solution {
public:
int strStr(string haystack, string needle) {
if(needle.empty())
return 0;
if(haystack.empty())
return -1;
for(int i = 0; i < haystack.size(); i++)
{
for(int j = 0; j < needle.size(); j++)
{
if(haystack[i+j] != needle[j])
break;
if(j == needle.size()-1)
return i;
}
if(haystack.size() - i < needle.size())//当遍历的位置到最后的长度小于needle的时候就说明不可能再完成匹配
break;
}
return -1;
}
};