kmp算法next计算方法_1个公式就弄懂了KMP模式匹配算法,困扰我多年的问题终于解决了

KMP算法可以说是一个很经典的模式匹配算法了,它一种改进的字符串匹配算法,但很多人就是不理解,甚至多看几次之后也没有理解透彻。

我们要查找S字符串串中是否包含P字符串,将P串称之为模式匹配串(以下简称模式串)。

朴素模式串匹配算法

我们先用一个动图来看不用KMP匹配的 朴素模式串匹配算法:

aa0d4d4172cc680f68413db8c24f26e5.gif

浅显易懂吧!但是呢,朴素的模式串匹配算法是有效率问题的!比如像下面这种:

c8787bf020c29a01f6166375b643701a.png

在匹配到第5个字符时,发生了P串和S串不匹配的情形,然后按照朴素的算法,需要回溯 i 指针到 1 位置,回溯 j 指针到 0 位置,然后我们依次移动P串进行匹配计算:

87accb4c3803fd02c5abb1de61db53c7.gif

在i = 3时,此时后面就有S[3] = P[0], S[4] = P[1], S[5] = P[2]:

7f187e73b918293db5596433852195af.png

然后S[6] != P[3]:

058b3a21a8de2e2caf46a53fb01891e0.png

才发现又是不匹配的,其中我们可以用肉眼观察到,当S[1]和P[0]发生不匹配时,P串依次的从S串第1个位置,第2个位置进行比较,只有到S串第3个位置字符是A时才和P串的第0个字符匹配,而且后面的两个字符也能够匹配,但是,S串第5个字符是不匹配的。

假如我们是通过肉眼判断的话,当

3866216e5a82ab5b56117573686a4e82.png

这时不匹配时,我们应该直接移动 j 到

b33bae8ec92f87335a2630c26a706c7a.png

进行匹配判断,因此,朴素的模式匹配是有一部分匹配判断是多余的,我们要想办法把它优化,在S[5] != P[5] 时,直接进行判断 S[5] 是否等于 P[2]。这是因为P[0] == P[3], P[1] == P[4];P串前后有相同的子串,可以看下图知道:

a0000978a78d95d56b72864e708367d2.png

由于S[3] = P[3], S[4] = P[4],而 P串前后有相同的子串:P[0] == P[3], P[1] == P[4],因此,我们可以直接进行判断S[5] 是否 等于 P[2]。也就是说当S[i] 和 P[j] 不匹配时,我们可以决定 j 的下一个位置是什么,从而使 i 指针不产生回溯。

推广到一般情况,我们可以用数学来证明这个事情:

c4080eb8f1b3df7e43e644ec586dc935.png

这就有了KMP算法,我们假设S串下标是i,P串下标是 j。当P串中存在重合的子串时,我们需要分析出j指针指向的下一个位置,为了记录所有 j 取值的情况,用next数组来记录每一个P串的字符不匹配时需要将 j 指针移动到的位置下标。

KMP算法核心

核心思想:模式串中有重合的子串,那么当模式串p[j]匹配失败时,要移动到下标为k的位置进行匹配,k需要满足:0 ~ k 之间的 k+1 个字符与 j-1-k ~ j-1 之间的 k+1 个字符相同。next数组用于存放匹配失败时,next[j]代表需要进行匹配下一个字符的位置p[next[j]]。

1cb6f2f8487abf4a698b839bad0bea5c.png

可以用数学语言表述为:

模式串p中有n个字符,其中

d885d0355502b9292dc1046f11b3d70d.png

如果存在

fce0546c3f1f4e18e3e27b01ed1bfa8b.png

使得

9885899ef2602fed65902958f46e59b1.png

那么令

55ddfffcbc64155d969902bd77fb5f76.png

否则(p中没有字符相同),令

5e499d9c9ce3a53b75ca60eb04dd6785.png

说明:

  • 当 k = 0 时,表示模式串p中前后只有一个字符能够重合,此时要从p[0](从头开始)匹配;
  • 当 k = -1 时,表示模式串中没有可以前后重合的部分子串,此时需要将原始串的指针下标(i)后移。

记住:k = next[j]; 当 s[i] != p[j]时,需要将j指针移动到p串的k下标位置,那么next[j]就用来存储这个k。

具体解释:

  • 第一种情况:j == 0时,s[i]和p[j]不匹配,那j指针不可能再向左移动了,此时应该要i指针向后移动。这种情况,记next[j] = -1;
4b17959ca79f099f2803118da8c99f36.png
  • 第二种情况:j == 1时, s[i]和p[j]不匹配,那j指针显然是要移动到0位置的。这种情况,记next[j] = 0;
c152932be4aaca14c73ee7397896a99d.png
  • 第三种情况:p[j] = p[k] 时,有 next[j+1] = next[j] + 1;

证明:

因为p[j]之前有 p[j-k ~ j-1] = p[0 ~ k-1]; 所以 next[j] = k;

那此时有 p[j] = p[k];可以得到 p[j-k ~ j-1] + p[j] == p[0 ~ k-1] + p[k];

即:p[j-k ~ j] == p[0 ~ k];

那么 next[j+1] = k + 1 = next[j] + 1;

0a29067e6991cf0dae004c51924dc165.png
  • 第四种情况:p[j] != p[k] 时,此时应该是 k = next[k];
400c3dfb1fb1cd17c958778707ef60e9.png

代码中为了保证循环依次计算j之前的重合子串,用j+1表示当前匹配不成功的字符位置,则计算next数组的算法如下:

void getNext(char *p, int next[]){    int j = 0, k = -1, len = strlen(p);    if (len == 0)    return;    next[0] = -1;    while (j < len-1)    {        if (k == -1 || p[j] == p[k])            next[++j] = ++k; //next[j+1]记录p[j+1]匹配失败时需要跳转的位置        else            k = next[k];    }}

由此可以写出KMP算法:

int KMP(char s[], char p[]){    int[] next = getNext(p);    int i = 0, j = 0;    while (s[i] && p[j])    {        if (j == -1 || s[i] == p[j])            i++, j++;        else        j = next[j]; // j回到指定位置    }    if (p[j])        return -1;    return i - j;}

改进的KMP算法

回顾第三种情况:p[j] = p[k] 时,有 next[j+1] = next[j] + 1;

6497c8094f47ca9db6214d2ae6f8888e.png

当s[i] != p[j]时,需要将 j 移动到 next[j],然而由于 p[j] == p[next[j]],导致移动j之后 s[i] 和 p[next[j]]仍然不匹配,此时需要继续将j移动到next[next[j]]进行比较;

显然,当p[j] 与 s[i]不匹配时,若p[j] == p[next[j]],就可以跳过j移动到next[j]这一步,直接进行移动到下一步。

因此,修正后的nextval数组如下:

9185054e09d81f7dfdf7cf1da5fbc0ea.png

改进思想:

在保证

p[i] = p[j - k + i],(i = 0, 1, ..., k)

后,再继续检查 p[j] 是否与 p[k] 相等?(此时原本是令 next[j] = k)

因为此时 k 记录的是 p[j] 发生不匹配时,需要直接跳到下次比较的位置;

所以:

  • 如果 p[k] = p[j] ,则同样发生不匹配,这时应该令 next[j] = next[k](因为 p[k] 与 p[j] 相同,则当前字符 p[j] 不匹配时,p[k]同样也不匹配,next[k]记录的是 p[k] 不匹配时需要跳转的位置,因此 p[j] 不匹配应该直接跳转到 next[k] 位置进行比较);
  • 如果 p[k] != p[j],那么当 p[j] 不匹配时才应该跳转到 p[k] 再次进行匹配,此时应令 next[j] = k。

改进算法如下:

void getNextval(char p[], int nextval[]){    int j = 0, k = -1, len = strlen(p);    if (len == 0)        return;    nextval[j] = k;    while (j < len-1)    {        if (k == -1 || p[j] == p[k])        {            ++j, ++k;            if(p[j] != p[k])                nextval[j] = k;            else            nextval[j] = nextval[k];        }        else            k = nextval[k];    }}

kmp匹配的算法代码如下:

/*** kmp模式匹配子串* 参数:* *str - 源串* *substr - 目标串*/int kmp(char *str, char *substr){    int i=0, j=0, slen = strlen(str), plen = strlen(substr);    int *nextval = (int *) calloc(plen, sizeof(int));    getNextval(substr, nextval);    while(i < slen && j < plen)    {        if( j == -1 || str[i] == substr[j])            i++, j++;        else            j = nextval[j];    }    if(j == plen)        return i - j;    else        return 0;}

如果觉得这篇文章对你有帮助,就帮我点赞并转发吧!我会持续分享IT、科技、技术方面的知识、经验,如果你喜欢的话,可以加个关注收藏一下吧!@IT研究僧大师兄