leetcode-cn 实现strStr()
题目如图:
其实这道题相当于让我们自己手写indexOf(),平时用惯了api,手写起来不是很容易,我自己就换了好几种写法,代码如下:
private static int strStr(String haystack, String needle) {
if (haystack == null || needle == null) {
return -1;
}
int needleLength = needle.length();
if (needleLength == 0) {
return 0;
}
int stackLength = haystack.length();
if (needleLength > stackLength) {
return -1;
}
int i = 0, j = 0, count = 0;
// i < stackLength - (needleLength - j - 1) 目的是为了减少查找次数
// 例如 A = "aabbccdef" B = "def" 当 j = 0 的时候最多只能到A串的'd',后面的“ef”没必要查找
// 当 j = 1 的时候 最多只能到 A 中的 'e' 相当于动态限制
for (; j < needleLength && i < stackLength - (needleLength - j - 1); i++, j++) {
while (haystack.charAt(i) != needle.charAt(j) && i < stackLength - (needleLength - j - 1)) {
// 全部遍历了还没找到 只针对于 needleLength = 1 或者两个字符串长度相等
if (i == stackLength - 1 || needleLength == stackLength) {
return -1;
}
if (count == 0) {
// 没有找到过,就一直找下去
++i;
} else {
// 找到过,但是中间某个不匹配了,要重新找 即 j 指针指向 0
// 例如 abccbcdd 和 bcd 之前匹配到 bc count = 2
// 然后 c 和 d 不匹配,所以需要重新匹配,i 从之前的 index 即 (i - count) 再加 1 位出发就好
j = 0;
i = (i - count) + 1;
count = 0;
}
}
if (++count == needleLength && i < stackLength - (needleLength - j - 1)) {
return i - count + 1;
}
}
return -1;
}
看了第一名的代码,愈发觉得jdk作者的强大🤣