KMP算法实现原理很简单,难的是对于Next数组求解的代码,网上对于Next数组的求解方法,有好几种不同版本。如果你不断翻看网络上的不同教程,就会觉得越来越复杂。
这里提供一种next数组实现原理,默认next[0]为0。
首先给出这个方法计算的next数组,如果跟你的计算方法不同那就别看了,免得越学越乱。
index:0 1 2 3 4 5
str : a a b a a f
next: 0 1 0 1 2 0
前缀与后缀的概念(这很重要)
匹配串 aabaaf ,j 为指向前缀的最后一个元素, i 为指向后缀的最后一个元素
前缀:不包括最后一个元素的所有子串
后缀:不包括第一个元素的所有子串
next数组本质是求最长的相等前后缀
初始化 i = 1;j = 0时,子串为aa,所以最长相等前后缀的长度为1。(注意:不包括最后一个元素,和第一个元素)
str[ i ] = str [ j ]相等则 j++ ,i通过循环也 +1;
此时 j = 1 i= 2,子串为abb,while循环中为false,因此 j = next[ j-1] ,此时j 已经等于0了,if判断也为false,因此直接next[ 1 ] = 0。
再解释一下while循环。
为什么j = next[ j-1 ]?
next数组代表的意思是,已经有这么长的相同元素。
aabaaf,当匹配到计算到f时,发现不匹配,那么这个时候,字符串看 f 的前一个元素的next数组值,发现是2,这说明已经有两个字符串已经匹配过了,不用在匹配了,要从第三个元素开始匹配,巧合的是,数组的下标是从0 开始的,第三个元素的下标正好是2。
做完这个之后,还不行,因为要看第三个元素,与前面不匹配的元素能否匹配的上,(pp[j] != pp[i]),如果不匹配 j 还要往前走,直到匹配上为止,或者j = 0.
int j = 0;
next[0] = 0;
for(int i = 1; i < pp.length;i++){
//不匹配的情况
while(j > 0 && pp[j] != pp[i]){
j = next[j-1];
}
//匹配的情况
if(pp[j] == pp[i]){
j++;
}
next[i] = j;
}
public static int strStr(String haystack, String needle) {
char[] pp = needle.toCharArray();
char[] ss = haystack.toCharArray();
//初始化next数组
int[] next = new int[needle.length()];
//前缀最后
int j = 0;
//默认索引0为0
next[0] = 0;
for(int i = 1;i < needle.length();i++){
//j前缀的最后一个下标大于0,退无可退
//不相等的情况下,j要往回退,直到要比的下一个数能与现在不匹配的数匹配上
while(j > 0 && pp[j] != pp[i]){
j = next[i-1];
}
//如果相等就都++
if(pp[j] == pp[i]){
j++;
}
//算出最长前后缀
next[i] = j;
}
//匹配都从0开始
for(int x = 0,y = 0;x < haystack.length();x++){
//如果不相等就回退pp
while(y > 0 && ss[x] != pp[y]){
y = next[y-1];
}
//相等++
if(ss[x] == pp[y]){
y++;
}
//匹配成功
if(y == needle.length()){
return x - needle.length() +1;
}
}
return-1;
}