实现 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() 定义相符。
方法一:
逐位判断是否包含,java中indexOf(String)函数也是这个原理。
class Solution {
public int strStr(String haystack, String needle) {
int len1 = haystack.length();
int len2 = needle.length();
for(int i = 0; i <= len1 - len2; i++){
if(haystack.substring(i, i + len2).equals(needle))
return i;
}
return -1;
}
}
方法二:
双指针分别指向两个串,每次匹配前先判断第一位是否相同,不用每一位都遍历。
class Solution {
public int strStr(String haystack, String needle) {
int len1 = haystack.length();
int len2 = needle.length();
if(len2 == 0)
return 0;
int pos1 = 0;
while(pos1 <= len1 - len2){
while(pos1 <= len1 - len2 && haystack.charAt(pos1) != needle.charAt(0))
++pos1;
int curLen = 0, pos2 = 0;
while(pos2 < len2 && pos1 < len1 && haystack.charAt(pos1) == needle.charAt(pos2)){
++pos1;
++pos2;
++curLen;
}
if(curLen == len2)
return pos1 - len2;
pos1 = pos1 - curLen + 1;
}
return -1;
}
}