数据结构与算法-KMP算法及优化

KMP 算法基本原理

KMP 算法是根据三位作者(D.E.Knuth,J.H.Morris 和 V.R.Pratt)的名字来命名的,算法的全称是 Knuth Morris Pratt 算法,简称为 KMP 算法。

KMP 算法的核心思想是:我们假设主串是 a,模式串是 b。在模式串与主串匹配的过程中,当遇到不可匹配的字符的时候,我们希望找到一些规律,可以将模式串往后多滑动几位,跳过那些肯定不会匹配的情况。

image.png

第一次匹配的情况如图
当i = 6的时候,匹配失败,说明前面的都能匹配上,我们希望找到x前缀和后缀相同的位置,这样我们就能快速的将j定位到c

image.png

从上图我们可以看出,这个过程跳过了很多多余的比较,那我们怎么呢做呢?
事实上,KMP 算法也可以提前构建一个数组,用来存储模式串中每个前缀的最长可匹配前缀子串的结尾字符下标。我们把这个数组定义为next 数组,很多书中还给这个数组起了一个名字,叫失效函数(failure function)。简单的说就是回溯的位置。

next数组的计算方法

next函数的计算方法是KMP最绕的部分


image.png

在求解next数组的4种情况:

  1. 默认next[1] = 0;
  2. 当 i=0时,表示当前的比应该从头开始.则i++,j++,next[j] = i;
  3. 当 T[i] == T[j] 表示2个字符相等,则i++,j++.同时 next[j] = i; 也就是说下一个字符的应该回溯的位置是i;
  4. 当 T[i] != T[j]` 表示不相等,则需要将i 退回到合理的位置. 则 i = next[i];
    T[i] != T[j] 进行回退. [i,j]范围有没有前缀和后缀;
    [next[i],j] 范围有没有前缀和后缀;(扩大范围)
    最终直到[0,j]范围有没有前缀和后缀;

此时next为01112

我用C写了一份next的求解代码

//----KMP 模式匹配算法---
//1.通过计算返回子串T的next数组;
//注意字符串T[0]中是存储的字符串长度; 真正的字符内容从T[1]开始;
//很多书中还给这个数组起了一个名字,叫失效函数(failure function)。next函数求解也是kmp中最绕的部分
void get_next(String T,int *next) {
    int i = 0;
    int j = 1;
    next[1] = 0;
    while (j<T[0]) {
        if (i==0 || T[i] == T[j]) {
            i++;
            j++;
            next[j] = i; //匹配相同说明前面的字符匹配,成功那么下一个字符的回溯位置就是i,这里的特殊在于当我们要计算 next[j] 的时候,前面的 next[0],next[1],……,next[j-1] 应该已经计算出来了
        } else {
            //如果字符不相同,则i值回溯;
            i = next[i];
        }
    }
}

有了next数组,我们接下来实现KMP算法

思路:

  1. 遍历模式串S,i 是用来标记主串的索引; 遍历模式串, j 是用来标记模式串的索引;
  2. 结束条件是当i > S.length 和 j > T.length;
    如果 i > S.length 但是j 却小于T.length 表示遍历了整个主串,都没有找到与模
    式串匹配的情况
    只有1种可能,就是j > T.length 表示,已经在主串中找到模式串了. 因为你已经顺
    利的把T模式串中的每个字符串正常的依次比较下去了,直到它结束;
  3. 当 j = 0 时,表示此时你需要将模式串从1这个位置与主串i+1这个位置开始比较;
  4. 当 T[i] == T[j], 表示此时当前模式串j 与 主串i 这个2个字符是相等,则j++,i++;
  5. 当 j != 0 并且T[i] != T[j] 时,表示此时需要移动模式串的 j ,那么我们让 j =
    next[j]; 来节省重复的比较次数;
//KMP 匹配算法(1)
//返回子串T在主串S中第pos个字符之后的位置, 如不存在则返回0;
int Index_KMP(String S,String T,int pos){
    
    int i = pos;
    int j = 1;
    
    int next[MAXSIZE];
    get_next(T, next);
    
    while (i <= S[0] && j <= T[0]) {
        //j==0,第一个就不匹配
        if (j == 0 || S[i] == T[j]) {
         //匹配成功i和j同时后移
            i++;
            j++;
        } else {
            //回溯,如果不匹配时,j回退到合适的位置,i值不变;
            j = next[j];
        }
    }
    
    if (j>T[0]) {
        return i-T[0];//i全部匹配完成的下标,要减去T的长度
    }
    return -1;
}

这部分很简单,其实就是字符匹配成功,i和j同时后移,如果匹配不成功,j回溯到合适的位置也就是当前next[j]的值,继续进行匹配。
循环结束,当j大于T的长度,说明匹配成功,否则匹配失败。

失效函数优化

image.png

此时的next为01234
当i == 5时,可以模拟一下,这里又有很多重复比较,j从4一直回溯到0,此时i = 6 ,j = 1,继续比较. 你又没有发现:4到1的这部分比较其实是多余。那么我们要nextval为00004,也就是说我们x不匹配的时候j回溯到4,此时T[4]为a,同样不匹配,那么j回溯到0,此时i++,j++,也就是开始比较主串的下一个字符

那么如何实现呢?

在求解nextVal数组的5种情况:

  1. 默认next[1] = 0;
  2. T[i] == T[j] 且++i,++j 后 T[i] 依旧等于 T[j] 则 nextval[i] = nextval[j]
  3. i = 0, 表示从头开始i++,j++后,且T[i] != T[j] 则nextVal = j;
  4. T[i] == T[j] 且++i,++j 后 T[i] != T[j] ,则nextVal = j;
  5. 当 T[i] != T[j]` 表示不相等,则需要将i 退回到合理的位置. 则 i = next[i];
//KMP 匹配算法(2)next数组优化
//求模式串T的next函数值修正值并存入nextval数组中;
void get_nextVal(String T,int *nextVal){
    int i = 0;
    int j = 1;
    nextVal[1] = 0;
    while (j<T[0]) {
        if (i==0 || T[i] == T[j]) {
            i++;
            j++;
            //如果当前字符与前缀不同,则当前的j为nextVal 在i的位置的值
            if (T[i] != T[j]) {
                nextVal[j] = i;
            } else {
                //如果当前字符相等并且前缀相同,则将前缀的nextVal 值赋值给nextVal 在i的位置,也就是说,让当前j应该回溯的位置和i相同
                nextVal[j] = nextVal[i];
            }
            
        } else {
            //如果字符不相同,则i值回溯;
            i = nextVal[i];
        }
    }
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 213,992评论 6 493
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,212评论 3 388
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 159,535评论 0 349
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,197评论 1 287
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,310评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,383评论 1 292
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,409评论 3 412
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,191评论 0 269
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,621评论 1 306
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,910评论 2 328
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,084评论 1 342
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,763评论 4 337
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,403评论 3 322
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,083评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,318评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,946评论 2 365
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,967评论 2 351

推荐阅读更多精彩内容