首页 > 代码库 > KMP(字符串匹配)

KMP(字符串匹配)

1、KMP是一种用来进行字符串匹配的算法,首先我们来看一下普通的匹配算法: 


现在我们要在字符串ababcabcacbab中找abcac是不是存在,那么传统的查找方法就是一个个的匹配了,如图:

 

经过六趟匹配之后,终于匹配上了。现在是数据比较小的时候,大家可能没有什么感觉,如果我们的数据是百万级别的,那用这种方法无疑复杂度太高了,很难接受。

 

2、模式匹配的一种改进算法: 


这种改进算法由D.E.Knuth与V.R.Pratt和J.H.Morris同时发现的,因此人们称他为克努特-莫里斯-普拉特操作(简称KMP算法)。此算法可以在O(n + m)的时间数量级上完成串的模式匹配操作。其改进在于:每当一趟匹配过程中出现字符比较不等时,不需要回溯i指针,而是利用已经得到的“部分匹配”结果将模式向右“滑动”尽可能远的一段距离后,继续进行比较。下面先从具体的例子看起。

回顾上面的传统的匹配过程,在第三趟的匹配中,当i = 7、j = 5字符比较不等时,又从i = 4、j = 1重新开始比较。然后,经仔细观察发现,在i = 4和j = 1,i = 5和j = 1以及i = 6和j = 1这3次比较都是不必进行的。因为从第三趟部分比配结果就可得出,主串中第4、5和6个字符必然是‘b‘、‘c‘和‘a‘(即模式串中的第2、3、和4个字符)。因为模式中的第一个字符是a,因此它无需再和这3个字符进行比较,而仅需要将模式串向右滑动3个字符的位置继续进行i = 7、j = 2时字符比较即可。同理,在第一趟匹配中出现字符不等时,仅需要将模式向右移动两个字符的位置继续进行i = 3、j = 1时的字符比较。由此,在整个匹配过程中,i指针没有回溯,如图4.4所示。

3、KMP 


此时我们构建一个next[]数组,这个数组的意思就是,当主串中的第i个字符与模式串中的第j个字符失配时,模式串中的next[j]个字符与第i个字符继续进行比较,显然复杂度提高了很多。

匹配代码:

int index_KMP(){    int stra = strlen(a);    int strb = strlen(b);    int i = -1,j = -1;    while(i < stra && j < strb)    {        if(j == -1 || a[i] == b[j])        {            i++;j++;        }        else        {            j = next[j];        }    }    if(j >= strb)    {//在a中找到了b字符串        return 1;    }    return 0;}
View Code

 

4、求next数组: 


KMP算法实在已知模式串的next函数值的基础上执行的,那么,如何求得模式串的next函数值呢?

从上述讨论可见,此函数值仅取决于模式串本身和相匹配的主串无关。我们可从分析其定义出发用地推的方法求得next函数值。

设next[j] = k,这表明在模式串中存在下列关系:‘p1....pk-1‘ = ‘pj-k+1....pj-1‘。其中k为满足1 < k < j的某个值,此时的next[j + 1] = ?可能有两种情况:

(1)若pk = pj,则表明模式串中‘p1....pk‘ = ‘pj-k+1....pj

这就是说next[j+1] = k+1,即next[j + 1] = next[j] + 1。

(2)若pk != pj,则表明在模式串中‘p1....pk’ != ‘pj-k+1....pj

此时可把求next函数值的问题看成是一个模式匹配的问题,整个模式串既是主串又是模式串,而当前在匹配过程中,已有pj-k+1=p1,pj-k+2=p2,...,pj-1=pk-1,则当pj!=pk时应将模式向右滑动至模式中的第next[k]个字符和主串的第j个字符相比较。若next[k] = k‘,且pj = pk‘,则说明在主串中的第j+1个字符之前存在一个长度为k’(即next[k])的最长子串,和模式串中从首字符起长度为k‘的子串相等,即‘p1...pk‘‘ = ‘pj-k+1...pj‘   (1 < k‘ < k < j)      (4 - 10)

这就是说next[j + 1] = k‘ + 1即next[j + 1] = next[k] + 1。

同理,若pj!=pk‘,则将模式继续向右滑动直至将模式中第next[k‘]个字符和pj对齐,....,依次类推,直至pj和模式中某个字符匹配成功或者不存在任何k‘(1 < k‘ < j)满足等式(4 - 10) 则next[j + 1] = 0。

代码:

void get_next(){    next[0] = -1;    int j = -1;    int i = 0;    int str = strlen(b);    while(i < str - 1)    {        if(j == -1 || b[i] == b[j])        {            next[++i] = ++j;        }        else        {            j = next[j];        }    }}
View Code

 

5、给出一份完整代码: 


#include<iostream>#include<cstdio>#include<cstring>using namespace std;char a[1000];char b[1000];int next[1000];void get_next(){    next[0] = -1;    int j = -1;    int i = 0;    int str = strlen(b);    while(i < str - 1)    {        if(j == -1 || b[i] == b[j])        {            next[++i] = ++j;        }        else        {            j = next[j];        }    }}int index_KMP(){    int stra = strlen(a);    int strb = strlen(b);    int i = -1,j = -1;    while(i < stra && j < strb)    {        if(j == -1 || a[i] == b[j])        {            i++;j++;        }        else        {            j = next[j];        }    }    if(j >= strb)    {//在a中找到了b字符串        return 1;    }    return 0;}int main(){    while(scanf("%s%s",a,b)!=EOF)    {        get_next();        if(index_KMP())        {            printf("YES\n");        }        else        {            printf("NO\n");        }    }    return 0;}
View Code

 

运行结果: