看到这道题第一个想到前缀树,咳咳咳,有点简化哈,但是思想差不多嘛。
就是创建一个快指针一个慢指针来遍历目标字符串,找一个一个的子字符串,然后用一个指针遍历目标子字符串,遍历的过程中判断是否相等。
我相信我的代码比我说的话更好懂(/捂脸)
public int strStr(String haystack, String needle) {
if(needle == null || needle.length() < 1){
return 0;
}
int begin = 0; //慢指针,指向子字符串的开始
int position = 0; //快指针,指向子字符串的结尾
int target = 0; //指向needle字符串
while(position < haystack.length()){
if(haystack.charAt(position) == needle.charAt(target)){//可能相等
target++;
position++;
if(target == needle.length()){//确实就是这个子字符串了
return begin;
}
}else if(target != 0){//说明之前子字符串有部分等于需要的字符串,但是匹配失败了
target = 0; //调零,重新等待遍历needle
position = ++begin;//快指针回调,选择下一个子字符串
}else{
begin = ++position;//相当于begin++,position++
}
}
return -1;
}
看了题解之后,发现原来是kmp算法可以用在这。使用kmp算法之后的代码如下:
class Solution {
public int strStr(String haystack, String needle) {
if(needle == null || haystack == null || needle.length() < 1) return 0;
int x = 0; //指向haystack字符串
int y = 0; //指向needle字符串
int[] next = getNext(needle); //获取next数组
while(x < haystack.length() && y < needle.length()){
if(haystack.charAt(x) == needle.charAt(y)){ //遍历子串和目标字符串元素相等,则两个指针一起向后移。
x++;
y++;
}else if(next[y] == -1){//回溯的时候判断一下如果是next[0]的话就不需要回溯了。
x++;
}else{// y指针向前回溯
y = next[y];
}
}
return y == needle.length() ? x - y : -1;
}
public int[] getNext(String str){ // 创建next数组,aabaaf对应-1 0 1 0 1 2
if(str.length() == 1) return new int[]{-1};
int[] next = new int[str.length()];
next[0] = -1;
next[1] = 0;
int i =2;
int cn = 0;
while(i < next.length){
if(str.charAt(i-1) == str.charAt(cn)){
next[i++] = ++cn;
}else if(cn > 0){
cn = next[cn];
}else{
next[i++] = 0;
}
}
return next;
}
}