实现 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) {
int i=0, j=0;
if(needle.size()<1)
return 0;
if(haystack.size()<needle.size())
return -1;
int first=i;
while(first<haystack.size())
{
first = i;
int second = j;
while(haystack[first]==needle[second]&&first<haystack.size()&&second<needle.size()){
first++;
second++;
}
// cout<<"first: "<<first;
// cout<<"second: "<<second;
if(second>=needle.size())
return i;
i++;
}
return -1;
}
};
升级后
class Solution {
public:
int strStr(string haystack, string needle) {
int i = 0;
int j = 0;
while(i< haystack.size() && j < needle.size())
{
if(haystack[i]==needle[j])
{
i++;
j++;
}
else
{
i=i-j+1;
j=0;
}
}
if(j>=needle.size())
{
return i-needle.size();
}
else
return -1;
}
};
KMP算法
class Solution {
public:
vector<int> getNext(string needle)
{
vector<int> next;
int j=0,t=-1;
next.push_back(-1);
while(j < needle.size()-1)
{
if(t==-1||needle[j]==needle[t])
{
next.push_back(t+1);
t++;
j++;
}
else
{
t = next[t];
}
}
return next;
}
int strStr(string haystack, string needle) {
if(needle.empty())
return 0;
vector <int> next = getNextVal(needle);
for(auto c : next)
{
cout<<c;
}
int i=0,j=0;
int m = needle.size();
int n = haystack.size();
while(i<n&&j<m)
{
if(j==-1||haystack[i]==needle[j])
{
i++;
j++;
}
else{
j=next[j];
}
}
return j>=m? i-j:-1;
}
vector<int> getNextVal(string needle)
{
vector <int> nextval;
int j = 0, t = -1;
nextval.push_back(-1);
int m = needle.size();
while(j<m-1)
{
if(t==-1||needle[j]==needle[t])
{
if(needle[j+1]!=needle[t+1])
nextval.push_back(t+1);
else
nextval.push_back(nextval[t+1]);
t++;
j++;
}
else
t = nextval[t];
}
return nextval;
}
};