LeetCode:28. 找出字符串中第一个匹配项的下标(KMP算法)
跟着carl学算法,本系列博客仅做个人记录,建议大家都去看carl本人的博客,写的真的很好的!
代码随想录
LeetCode:28. 找出字符串中第一个匹配项的下标(KMP算法)
给你两个字符串 haystack 和 needle ,请你在 haystack 字符串中找出 needle 字符串的第一个匹配项的下标(下标从 0 开始)。如果 needle 不是 haystack 的一部分,则返回 -1 。
示例 1:
输入:haystack = “sadbutsad”, needle = “sad”
输出:0
解释:“sad” 在下标 0 和 6 处匹配。
第一个匹配项的下标是 0 ,所以返回 0 。
示例 2:
输入:haystack = “leetcode”, needle = “leeto”
输出:-1
解释:“leeto” 没有在 “leetcode” 中出现,所以返回 -1 。
注意next
数组的求法,注意都是先找不相等的时候然后再去判断是否相等,多注意代码注释
public int strStr(String haystack, String needle) {
// 求next数组 最长相等前后缀的数组
int[] next = getNext(needle);
int j = 0;
for(int i = 0; i < haystack.length(); i++){
while(j > 0 && needle.charAt(j) != haystack.charAt(i)){
j = next[j - 1];
}
if(needle.charAt(j) == haystack.charAt(i)){
j++;
}
// 如果j已经遍历完了 则说明在haystack中存在needle
if(j == needle.length()){
// 返回第一个匹配项的下标
return i - needle.length() + 1;
}
}
return -1;
}
private int[] getNext(String needle){
int[] next = new int[needle.length()];
// i之前的字符的最长相等前后缀的长度
int j = 0;
for(int i = 1; i < needle.length(); i++){
// 注意这里面的 必须是先判断不等 然后再去判断相等,如果反过来就错了,比如 aabaaac;
// 这个while最终的结果是要么 j回到0 要么找到等于i位置的字符
while(j > 0 && needle.charAt(j) != needle.charAt(i)){
j = next[j - 1];
}
// 如果找到了和i位置字符相等的字符 则最长相等前后缀需要加1,如果没找到则是0
if(needle.charAt(j) == needle.charAt(i)){
j++;
}
next[i] = j;
}
return next;
}