题目描述
实现 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() 定义相符。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/implement-strstr
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路
用KMP算法进行字符串匹配,直到第一次匹配,返回haystack中的下标(haystack(匹配串)当前下标减needle(模式串)的当前下标)。
代码(c++)
class Solution {
public:
vector<int> next;
void getNext(string needle,int len){
int j=-1;
next.push_back(-1);
for(int i=1;i<len;i++){
while(j!=-1&&needle[i]!=needle[j+1]){
j=next[j];
}
if(needle[i]==needle[j+1]){
j++;
}
next.push_back(j);
}
}
int strStr(string haystack, string needle) {
if(needle.length()==0) return 0;
int n=haystack.length();
int m=needle.length();
getNext(needle,m);
int j=-1;
for(int i=0;i<n;i++){
while(j!=-1&&haystack[i]!=needle[j+1]){
j=next[j];
}
if(haystack[i]==needle[j+1]){
j++;
}
if(j==m-1){//匹配成功
return i-j;
}
}
return -1;
}
};
getNext函数可以改进为:
void getNext(string needle,int len){
int j=-1;
next.push_back(-1);
for(int i=1;i<len;i++){
while(j!=-1&&needle[i]!=needle[j+1]){
j=next[j];
}
if(needle[i]==needle[j+1]){
j++;
}
if(j==-1||needle[i+1]!=needle[j+1]){
next.push_back(j);
}
else{
next.push_back(next[j]);
}
}
}
如果要计算模式串出现的次数,KMP算法可以改为:
int strStr(string haystack, string needle) {
if(needle.length()==0) return 0;
int n=haystack.length();
int m=needle.length();
int ans=0;
getNext(needle,m);
int j=-1;
for(int i=0;i<n;i++){
while(j!=-1&&haystack[i]!=needle[j+1]){
j=next[j];
}
if(haystack[i]==needle[j+1]){
j++;
}
if(j==m-1){//匹配成功
ans+=1;
j=next[j];//让j回退到next[j]继续匹配
}
}
return ans;
}