数据结构与算法---KMP算法

KMP算法是数据结构与算法中串的经典算法案例,KMP是由三位学者同时发现(D.E.Knuth,J.H.Morris和V.R.Pratt),然后就以他们命名的算法。

串的模式匹配

串的模式匹配也就是一种串的定位操作,比如:
提供一个主串T和一个模式串S,需要获得模式串S在T中的位置。

BF算法

BF算法(Brute-Force)顾名思义就是暴力搜索算法,这也是比较容易想到的一种方法。


数据结构与算法---KMP算法_第1张图片
BF配图*1
  • 算法首先对S串和T串从首位进行比对匹配
  • 如果匹配相同双方同时进行下一个比对
  • 如果匹配不相同,S串回溯到本轮比对的开头的下一个数据,T回到开头进行新一轮的比对。

BF算法分析

暴力破解的算法相对简单,这里不再过多讲解算法细节。

值得注意思考的是:
假如主串S的长度是N,模式串T的长度是M。举个最坏的例子,S:aaaaaaaab T:aaaab 不做任何优化的情况下,需要进行至少MN次比对。

所以,该算法的时间复杂度在O(NM)数量级上。

KMP算法

KMP算法提出来就是为了优化暴力搜索遗留给我们的思考。当我们遇到思考题中S,T时,什么时候可以减少回溯?怎么避免不必要的回溯操作?

数据结构与算法---KMP算法_第2张图片
KMP配图*2

通过*2图可以引发我们的思考:
当模式串T比对到T1的时候,位置1、2、3分别匹配成功,但是位置4并匹配失败。如果是BF算法,模式串T将会从头并从位置2进行比对。我们通过人脑的思考,模式串这时候应该头对着3,S串的4和T1进行比对才为最优,即T2位置之前的暴力计算都可以省略。最后到T3配对成功。

通过观察可以看出,当比对过程中发生不匹配的时候,不匹配的位置之前的子串如果有首尾呼应的属性的话,就可以实现减少回溯。
这里的首尾呼应即串存在最大(不包括自己)的公共前缀串后缀串。并且,这是模式串的属性,自然只和模式串有关。

为了描述模式串这个属性,需要引进一个NEXT数组,该数组存贮这当前位置之前子串的最大公共前后缀的长度(即位置)。

当发生匹配不成功的时候,主串S串不再进行回溯,模式串会根据NEXT数组进行相应的回溯。

[TIPS] 我们可以换一种角度思考,以abcab串为例,当发生不匹配的时候,说明前面的是匹配的,前后缀串和中间串这种情况也是一一对应的,如果继续匹配下去前缀串和中间串必定不会匹配成功,也必然是浪费时间的过程,这样一想是不是好理解些了。另外,如有交叉串(如ababab)也是也只是省略了中间串,性质还是一样的。

NEXT数组

NEXT数组是匹配到相应位置时应该回溯的位置(长度),这个回溯的位置和它之前的子串有关。这也导致NEXT数组中存放的都向后偏移一位。

以abab为例,当匹配到第二个b时发生了不匹配,这时说明前面的aba是匹配的,回溯的位置当然是第一个a的位置后面的一个,即回溯到1,这样才能和当前与b比对的进行吻合。这里强调的是NEXT数组里面放的是子串信息。

另外,匹配的时候我们只需要进位比较下一个就行了。如果不匹配的话有两种情况:
1、S串的某位和T串的首位进行比较;
2、S串的某位与T串的非首位进行比较;

面对第一种情况,S串进位,T不变;面对第二种情况,需要参考NEXT数组进行进一步的比对。例如图*2中的T2状态,进一步匹配位置4后,发现还是不匹配,需要继续跟进前面的子串,一直到T串的首尾。

可以看出来,这个数组存放的数据是,首尾是一个标志位可以用※来代替,当匹配到这一位的时候,会进行相应操作。剩下的便是子串的最大前后缀了。所以,abab的NEXT数组是 [※,0,0,1]。一般,为了方便操作和代码简洁※用-1表示。

KMP代码解析

KMP算法主要分为两部分完成:

  • NEXT数组的获取
  • 使用NEXT数组进行回溯查找
NEXT数组的获取

当提供一个串T,求出他的NEXT数组。我们从递推的角度看(引进长当前长度len、索引i参数):

当第i个元素和第len个元素进行匹配,匹配成功,NEXT[i+1] = len+1;

当第i个元素和第len个元素匹配失败,但是注意,这时候i前面的len个元素和i前面的len的前面的len个元素是相匹配的。这时候需要考虑着len个元素是否是首尾呼应的。这里举个例子便好理解,串ababcababa,当我们匹配到最后一个a的时候,发现a和c位置匹配不成功,这时候abab是相互匹配的,它的前后缀串是ab,所以比较ab后面的这个元素是否和最后一位的a相同,这时已经递推给子串了,len = NEXT[len]。经过匹配,发现相同,这个时候需要将NEXT[i+1] = len+1。如果一直不相同,则会一直递推到len = -1,然后等于0。

void getKMPNext(char T[]) {
    //l是前缀相同的长度
    int l = -1, i = 0;
    next[0] = -1;
    int TL = (int)strlen(T);
    while (i < TL) {
        if (l == -1 || T[i] == T[l]) {
            l++;
            i++;
            next[i] = l;
        } else {
            l = next[l];
        }
    }
}
使用NEXT数组进行回溯查找

在讲NEXT数组的时候,这边的逻辑就已经比较清晰了。设i、j分别是S、T串的索引,并同时从0开始。当索引下的元素值相等,i++、j++;当不相等,将前面最大子串最大前缀的长度获取进行下一步的比对,j=NEXT[i];

int KMPMatchingFunc(char S[], char T[]) {
    getKMPNext(T);   //将KMP数组初始化一下
    
    int i = 0, j = 0;
    //得到两个串的长度。
    int SL = (int)strlen(S);
    int TL = (int)strlen(T);
    
    while (i < SL && j < TL) {
        if (j == -1 || S[i] == T[j]) {
            i++;
            j++;
        } else {
            
            j = next[j];
        }
    }
    if (j >= TL) {
        return i-j;

    } else {
        return 999;
    }
}

算法的优化

KMP和BF算法都可以作进一步的优化。当在一直没有发生匹配的时候,这两种算法都将i、j都以串的长度M、N次进行完了才结束的,根据实际情况是没有必要的。当模式串T的首尾已经和M-N+1进行对齐的时候,剩下的就没必要在比较了。因为,剩下的长度已经不足以支撑该模式串T的匹配成功。

int KMPMatchingFunc(char S[], char T[]) {
    getKMPNext(T);   //将KMP数组初始化一下
    
    int i = 0, j = 0;
    //得到两个串的长度。
    int SL = (int)strlen(S);
    int TL = (int)strlen(T);
    
    while (i < SL && j < TL) {
        if (j == -1 || S[i] == T[j]) {
            i++;
            j++;
        } else {
            //优化
            if ( i - j + 1 + TL > SL ) {
                return 999;
            }
            j = next[j];
        }
    }
    return i-j;
}

小结

KMP算法通过计算模式串的NEXT数组需要N个数量级的时间复杂度,通过模式串进行匹配需要M个时间复杂度。所以,KMP算法的时间复杂度是O(M+N)。

个人认为,对于绝大多数情况,加法的时间复杂度还是优于乘法的。但是,具体问题具体分析,如果有些业务本身的模式串规律性不强,使用KMP也显得有点复杂。对于有些较长的模式串、并且基础元素不是特别繁多,那么里面的规律相对较多,KMP绝对是不二选择。

串的模式匹配git(仅供参考)

欢迎指出错误,学习,交流,提意见!!一起进步、学习~~

你可能感兴趣的:(数据结构与算法---KMP算法)