实现 strStr() 函数。
给定一个 haystack 字符串和一个 needle 字符串,在 haystack 字符串中找出 needle 字符串出现的第一个位置 (从0开始)。如果不存在,则返回 -1。
示例 1:
输入: haystack = "hello", needle = "ll"
输出: 2
示例 2:
输入: haystack = "aaaaa", needle = "bba"
输出: -1
说明:
当 needle 是空字符串时,我们应当返回什么值呢?这是一个在面试中很好的问题。
对于本题而言,当 needle 是空字符串时我们应当返回 0 。这与C语言的 strstr() 以及 Java的 indexOf() 定义相符。
暴力法
直接拿needle去和haystack进行匹配,若能成功匹配则输出具体位置,否则输出-1
库函数解法:
public int strStr(String haystack, String needle) {
if(needle.length()<1)
return 0;
if( haystack.length() < 1 ||needle.length()>haystack.length())
return -1;
return haystack.indexOf(needle);
}
复杂度分析:
时间复杂度:O(N^2)
空间复杂度:O(1)
KMP解法:
public int strStr(String haystack, String needle) {
if(needle.length()<1)
return 0;
if(haystack == null || needle.length() < 1 ||needle.length()>haystack.length())
return -1;
char[] hayArr = haystack.toCharArray();
char[] needArr = needle.toCharArray();
int index1 = 0; //主串(haystack)的位置
int index2 = 0; //模式串(needle)的位置
while (index1 < hayArr.length && index2 < needArr.length) {
if (hayArr[index1] == needArr[index2]) { // 当两个字符相等则比较下一个
index1++;
index2++;
} else {
index1 = index1 - index2 + 1; // 一旦不匹配,index1后退
index2 = 0; // index2归零
}
}
if (index2 == needArr.length) { //说明完全匹配
return index1 - index2;
} else {
return -1;
}
}
复杂度分析:
时间复杂度:O(N)
空间复杂度:O(1)