1、问题描述
实现 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提示:
1 <= haystack.length, needle.length <= 104
haystack 和 needle 仅由小写英文字符组成来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/implement-strstr
2、代码实现
class Solution {
public int strStr(String haystack, String needle) {
int i = 0;
int j = 0;
int[] next = getNext(needle);
while(i < haystack.length() && j < needle.length()){
if(j == -1 || haystack.charAt(i) == needle.charAt(j)){
i++;
j++;
}else {
j = next[j];
}
}
if(j == needle.length()){
return i - j;
}else {
return -1;
}
}
//next[i]的值表示下标为i的字符前的字符串的最长公共前后缀
public int[] getNext(String needle){
int[] next = new int[needle.length()];
int j = 0;
int k = -1;
next[j] = -1;
while(j < needle.length() - 1){
if(k == -1 || needle.charAt(j) == needle.charAt(k)){
j++;
k++;
next[j] = k;
}else {
k = next[k];
}
}
return next;
}
}