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

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

串的模式匹配

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

BF算法

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


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

BF算法分析

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

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

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

KMP算法

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

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(仅供参考)

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

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 204,445评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,889评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,047评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,760评论 1 276
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,745评论 5 367
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,638评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,011评论 3 398
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,669评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,923评论 1 299
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,655评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,740评论 1 330
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,406评论 4 320
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,995评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,961评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,197评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,023评论 2 350
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,483评论 2 342

推荐阅读更多精彩内容