public class KMPDemo {
public static void main(String[] args) {
int substrIndexByKMP = getSubstrIndexByKMP("aabacbaabacabc".toCharArray(), "aabaca".toCharArray());
System.out.println(substrIndexByKMP);
int substrIndexByKMP2 = getSubstrIndexByKMP("aaaaa".toCharArray(), "ab".toCharArray());
System.out.println(substrIndexByKMP2);
}
/**
* 比如 str="aabacbaabacabc" patten="aabaca"
* @param str
* @param pattern
@return pattern在str中第一次出现的位置,返回其第一次字符出现的index下标
* 注意:返回-1表示没有找到符合要求的字串
* 设str.length=m,pattern.length=n
* 最坏时间复杂度:o(m+n)
* 其中处理求next数组的时间复杂度为o(n),循环匹配的时间复杂度为o(m)
* 循环匹配的时间复杂度为o(m),这是因为str上的指针从来不回溯,虽然有极少情况下不会进行i++
* 但总体而言时间复杂度为o(m) 好吧,这里还是不是很理解,先放一放
*/
public static int getSubstrIndexByKMP(char[] str,char[] pattern){
//先得到nextVal数组
int[] nextValArr = getNextValArr(pattern);
int i=0,j=0;
while(i < str.length && j < pattern.length){
if (j==-1 || str[i] == pattern[j]){
//一般如果比较字符不相等,i是不用动的,只需要j移动即可
//但是如果j==0,即子串的第一位就不匹配,则需要同时移动i和j,因此next[0]统一规定为-1,方便处理
i++;
j++;
}else{
j=nextValArr[j];
}
}
if (j >= pattern.length){
return i-pattern.length;
}
return -1;
}
/**
* nextVal是对next的优化
* @param pattern
* @return
*/
public static int[] getNextValArr(char[] pattern){
int[] next = getNextArr(pattern);
int[] nextVal = new int[next.length];
nextVal[0] = -1;
for (int i = 1; i < pattern.length; i++) {
if (pattern[i] == pattern[next[i]]){
//这里之所以用nextVal[next[i]]而非next[next[i]],
// 是因为pattern[next[i]]和pattern[next[next[i]]]还可能相等,
// 但pattern[nextVal[next[i]]]和pattern[nextVal[i]]却不可能相等,
// 因为如果相等,i在更小的时候走到这里的时候就会再往前取一个值了
nextVal[i] = nextVal[next[i]];
}else {
nextVal[i] = next[i];
}
}
return nextVal;
}
/**
* 关键是得到next数组
* 参考 https://www.bilibili.com/video/BV16X4y137qw/?spm_id_from=333.337.search-card.all.click
* @param pattern
* @return
*/
private static int[] getNextArr(char[] pattern) {
int[] next = new int[pattern.length];
next[0] = -1;
int i=0,j=-1;
while(i<pattern.length - 1){
if (j==-1 || pattern[i] == pattern[j]){
next[++i] = ++j;
}else {
j = next[j];
}
}
return next;
}
}
KMP算法完整版 包括求next数组的算法及其优化
最新推荐文章于 2024-08-20 17:09:37 发布