leetcode-cn 实现strStr()

题目如图:
实现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作者的强大🤣
indexOf()
实现strStr()

posted @ 2019-06-11 13:23  平湖  阅读(130)  评论(0编辑  收藏  举报