实现 strStr() 函数。给你两个字符串 haystack 和 needle ,请你在 haystack 字符串中找出 needle 字符串出现的第一个位置(下标从 0 开始)。如果不存在,则返回 -1 。
说明:
当 needle 是空字符串时,我们应当返回什么值呢?这是一个在面试中很好的问题。
对于本题而言,当 needle 是空字符串时我们应当返回 0 。这与 C 语言的 strstr() 以及 Java 的 indexOf() 定义相符。
示例 1:
输入:haystack = "hello", needle = "ll"
输出:2
方法一:暴力匹配
我们可以让字符串needle 与字符串haystack 的所有长度为m的子串均匹配一次。
为了减少不必要的匹配,我们每次匹配失败即立刻停止当前子串的匹配,对下一个子串继续匹配。如果当前子串匹配成功,我们返回当前子串的开始位置即可。如果所有子串都匹配失败,则返回−1。
class Solution{
public int strStr(String haystack,String needle){
int n=haystack.length(),m=needle.length();
for(int i=0;i+m<=n;i++){
boolean flag=true;
for(int j=0;j<m;j++){
if(haystack.charAt(i+j)!=needle.charAt(j)){
flag=false;
break;
}
}
if(flag){
return i;
}
}
return -1;
}
}
方法二:Knuth-Morris-Pratt 算法
class Solution{
public int strStr(String haystack,String needle){
int n=haystack.length(),m=needle.length();
if(m==0){
return 0;
}
int[] pi=new int[m];
for(int i=1,j=0;i<m;i++){
while(j>0 && needle.charAt(i)!=needle.charAt(j)){
j=pi[j-1];
}
if(needle.charAt(i)==needle.charAt(j)){
j++;
}
pi[i]=j;
}
for(int i=0,j=0;i<n;i++){
while(j>0 && haystack.charAt(i)!=needle.charAt(j)){
j=pi[j-1];
}
if(haystack,charAt(i)==needle.charAt(j)){
j++;
}
if(j==m){
return i-m+1;
}
}
return -1;
}
}