实现 strStr()
实现 strStr() 函数。
给你两个字符串 haystack 和 needle ,请你在 haystack 字符串中找出 needle 字符串出现的第一个位置(下标从 0 开始)。如果不存在,则返回 -1 。
说明:
当 needle 是空字符串时,我们应当返回什么值呢?这是一个在面试中很好的问题。
对于本题而言,当 needle 是空字符串时我们应当返回 0 。这与 C 语言的 strstr() 以及 Java 的 indexOf() 定义相符。
示例 1:
输入:haystack = "hello", needle = "ll"
输出:2
示例 2:
输入:haystack = "aaaaa", needle = "bba"
输出:-1
示例 3:
输入:haystack = "", needle = ""
输出:0
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/implement-strstr
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
代码:
class Solution {
public int strStr(String haystack, String needle) {
int n = haystack.length(),m=needle.length();
if(m == 0){ //排除特殊情况
return 0;
}
for(int i = 0;i+m<=n;i++){ //指针指定haystack,我们只需要判断i可能取的范围
boolean flag = true;
for(int j = 0;j<m;j++){ //指针指定needle
if(haystack.charAt(i+j) != needle.charAt(j)){ //当haystack和needle没有相等的部分时候
flag = false;
break;
}
}
if(flag){
return i; //返回i的下标
}
}
return -1;
}
}
class Solution {
public int strStr(String haystack, String needle) {
if(needle.length()>haystack.length())return -1; //特殊情况
if(needle.length()==0)return 0;
int index1 = 0,index2=0,res=0;
while(index2<needle.length()&&index1<haystack.length()){ //遍历两个字符串
if(needle.charAt(index2)==haystack.charAt(index1)){ //判断两个字符串相等时
index1++;
index2++;
}else{
index1 = index1 - index2 + 1; //不相等就回退到相等字符串时下一个坐标
index2 = 0;
}
}
if(index2==needle.length()){ //当相等的时候直接返回差值
return index1-index2;
}
return -1;
}
}
KMP算法:
class Solution {
public int strStr(String haystack, String needle) {
int[] next = new int[needle.length()];
if (needle.length() == 0) return 0;
for(int rigth=0,left = 1;left<needle.length();left++){
while(rigth>0 && needle.charAt(rigth)!=needle.charAt(left)){
rigth = next[rigth-1]; //不相等,回退左指针
}
if(needle.charAt(rigth)==needle.charAt(left)){ //当匹配时,更新右指针
rigth++;
}
next[left] = rigth; //把前缀码写入到next数组中
}
for(int i=0,j=0;i<haystack.length();i++){
while(j>0&&haystack.charAt(i)!=needle.charAt(j)){
j=next[j-1];
}
if(haystack.charAt(i)==needle.charAt(j)){
j++;
}
if(j==needle.length())return i-needle.length()+1;
}
return -1;
}
}