题目
实现 strStr() 函数。
给定一个 haystack 字符串和一个 needle 字符串,在 haystack 字符串中找出 needle 字符串出现的第一个位置 (从0开始)。如果不存在,则返回 -1。
示例 :
输入: haystack = "hello", needle = "ll"
输出: 2
输入: haystack = "aaaaa", needle = "bba"
输出: -1
说明:
当 needle 是空字符串时,我们应当返回什么值呢?这是一个在面试中很好的问题。
对于本题而言,当 needle 是空字符串时我们应当返回 0 。这与C语言的 strstr() 以及 Java的 indexOf() 定义相符。
标签
字符串;双指针;滑动窗口
解题思路
方法一:子串逐一比较
将长度为 L 的滑动窗口沿着 haystack 字符串逐步移动,并将窗口内的子串与 needle 字符串相比较,时间复杂度为 O((N - L)L)
方法二:双指针
只有子串的第一个字符跟 needle 字符串第一个字符相同的时候才需要比较。可以一个字符一个字符比较,一旦不匹配了就立刻终止。如果完全匹配成功,则返回字串的开始位置;若失败,便回溯。
代码
public class Leetcode28_Implement_strStr {
public static void main(String[] args) {
String haystack = "hello";
String needle = "ll";
System.out.println(Leetcode28_Implement_strStr.strStr2(haystack, needle));
}
public static int strStr(String haystack, String needle) {
if (needle.length() == 0) {
return 0;
}
return haystack.indexOf(needle);
}
//子串逐一比较:滑动窗口
public static int strStr1(String haystack, String needle) {
int len1 = haystack.length();
int len2 = needle.length();
for (int i = 0; i < len1 - len2 + 1; i++) {
if (haystack.substring(i, len2 + i).equals(needle)) {
return i;
}
}
return -1;
}
//双指针
public static int strStr2(String haystack, String needle) {
int len1 = haystack.length();
int len2 = needle.length();
if (len2 == 0) {
return 0;
}
int pn = 0;
//移动pn指针,直到pn所指向位置的字符与needle字符串第一个字符相等。
while (pn < len1 - len2 + 1) {
//只有子串的第一个字符跟 needle 字符串第一个字符相同的时候才需要比较
while (pn < len1 - len2 + 1 && haystack.charAt(pn) != needle.charAt(0)) {
pn++;
}
//寻找匹配字串
int currLen = 0, pL = 0;
while (pn < len1 && pL < len2 && haystack.charAt(pn) == needle.charAt(pL)) {
pn++;
pL++;
currLen++;
}
//如果找到了,返回索引
if (currLen == len2) {
return pn - len2;
}
//否则,回溯
pn = pn - currLen + 1;
}
return -1;
}
}
来源:力扣(LeetCode)