字符串模式匹配之KMP算法的next数组详解与C++实现

相信来看next数组如何求解的童鞋已经对KMP算法是怎么回事有了一定的了解,这里就不再赘述,附上一个链接吧:https://www.cnblogs.com/c-cloud/p/3224788.html,里面对KMP算法有详细的讲解,如果你还不了解KMP算法,可以看看~~。

下面就来讲解不容易理解但又很重要的next数组,相信这是你看过的最容易理解的next数组的讲解了(*^_^*)。

---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------

一、首先,next数组是什么?

简单来说,假设我们有一个主串S和一个模式串T,并且想知道S是否包含T,如果包含,那么T第一次出现在S中的首字符在什么位置?有一种暴力求法:当S[i]!=T[j]的时候,j回溯到j=0,而i回溯到i=i-j+1,这种方法简答粗暴,但效率低下,时间复杂度的范围是(最好与最坏情况):O(n+m)~O(n*m),其中,n为主串S的长度,m为模式串T的长度;KMP算法对这种BF算法做了很大的改进,其基本思想是主串S不进行回溯,而是希望某趟在S[i]和T[j]匹配失败后,下标i不回溯,下标j回溯到某个位置K,使得T[K]对准S[i]继续进行比较。显然,关键问题是如何确定位置K。而这里的next数组表示的就是这个K值!需要注意的是,这个K值仅依赖于模式串T本身字符序列的构成,与主串S无关。

二、朴素模式匹配算法BF

这里给出BF代码,很简单,就不做具体说明了。

int BF(string S, string T)
{
    int i = 0;    // S 的下标
    int j = 0;    // T的下标
    int s_len = S.size();    // 字符串长度
    int t_len = T.size();
    if(s_len<t_len)
        return -1;
    while (i < s_len && j < t_len)
    {
        if (S[i] == t[j])  // 相等,则都前进一步
        {
            i++;
            j++;
        }
        else              // 回溯
        {
            i = i - j + 1;
            j = 0;
        }
    }

    if (j == t_len)        // 匹配成功
        return i - j;

    return -1;
}

 

时间复杂度的范围是(最好与最坏情况):O(n+m)~O(n*m),其中,n为主串S的长度,m为模式串T的长度。

三、改进的模式匹配算法KMP

KMP算法的核心是如何求出next数组!next数组其实就是查找T中每一位前面的子串的前后缀有多少位匹配,从而决定j失配时应该回退到哪个位置(前后缀的概念请看附录)。

文字总是枯燥的,如果图文并茂,那就更好了!好了,上图!

这个图画的就是T这个要查找的关键字字符串。假设我们有一个空的next数组,我们的工作就是要在这个next数组中填值。
下面我们用数学归纳法来解决这个填值的问题。
这里我们借鉴数学归纳法的三个步骤(或者说是动态规划?):
1、初始状态
2、假设第j位以及第j位之前的我们都填完了
3、推论第j+1位该怎么填

初始状态我们稍后再说,我们这里直接假设第j位以及第j位之前的我们都填完了。也就是说,从上图来看,我们有如下已知条件:
next[j] == k;
next[k] == 绿色色块所在的索引;
next[绿色色块所在的索引] == 黄色色块所在的索引;
我们来看下面一个图,可以得到更多的信息:

1.由"next[j] == k;"这个条件,我们可以得到A1子串 == A2子串(根据next数组的定义,前后缀那个)。

2.由"next[k] == 绿色色块所在的索引;"这个条件,我们可以得到B1子串 == B2子串。

3.由"next[绿色色块所在的索引] == 黄色色块所在的索引;"这个条件,我们可以得到C1子串 == C2子串。

4.由1和2(A1 == A2,B1 == B2)可以得到B1 == B2 == B3。

5.由2和3(B1 == B2, C1 == C2)可以得到C1 == C2 == C3。

6.B2 == B3可以得到C3 == C4 == C1 == C2

接下来,我们开始用上面得到的条件来推导如果第j+1位失配时,我们应该填写next[j+1]为多少?

next[j+1]即是找T从0到j这个子串的最大前后缀:

#:(#:在这里是个标记,后面会用)我们已知A1 == A2,那么A1和A2分别往后增加一个字符后是否还相等呢?我们得分情况讨论:

(1)如果T[k] == T[j],很明显,我们的next[j+1]就直接等于k+1。用代码来写就是next[++j] = ++k;

(2)如果T[k] != T[j],那么我们只能从已知的,除了A1,A2之外,最长的B1,B3这个前后缀来做文章了。

那么B1和B3分别往后增加一个字符后是否还相等呢?

由于next[k] == 绿色色块所在的索引,我们先让k = next[k],把k挪到绿色色块的位置,这样我们就可以递归调用"#:"标记处的逻辑了。

由于j+1位之前的next数组我们都是假设已经求出来了的,因此,上面这个递归总会结束,从而得到next[j+1]的值。

 

我们唯一欠缺的就是初始条件了:next[0] = -1,  k = -1, j = 0另外有个特殊情况是k为-1时,不能继续递归了,此时next[j+1]应该等于0,即把j回退到首位。

即 next[j+1] = 0; 也可以写成next[++j] = ++k;

接下来就是代码实现next数组(C++版):

 1 int* getNext(string T)
 2 {
 3     int T_len = T.size();
 4     int* next = new int[T_len];    // 声明next数组    
 5     int i = 0;    // T的下标
 6     int j = -1;
 7     next[0] = -1;
 8     while (i < T_len)
 9     {
10         if (j == -1 || T[i] == T[j])
11         {                                    
12                 next[++i] = ++j;
13         }
14         else
15             j = next[j];
16     }
17     return next;
18 
19 }    

KMP优化:

如果T[k] == T[j],很明显,我们的next[j+1]就直接等于k+1。用代码来写就是next[++j] = ++k;可是我们知道,第j+1位是失配了的,如果我们回退j后,发现新的j(也就是此时的++k那位)跟回退之前的j也相等的话,必然也是失配。所以还得继续往前回退。

 1 int* getNext(string T)
 2 {
 3     int T_len = T.size();
 4     int* next = new int[T.size()];    // 声明next数组    
 5     int i = 0;    // T的下标
 6     int j = -1;
 7     next[0] = -1;
 8     while (i < T_len)
 9     {
10         if (j == -1 || T[i] == T[j])
11         {
12             if (T[i + 1] == T[j + 1])    //KMP优化
13                 next[++i] = next[++j];
14             else
15                 next[++i] = ++j;
16         }
17         else
18             j = next[j];
19     }
20     return next;
21 
22 }

完整代码C++:

 1 #include <iostream>
 2 #include <string>
 4 using namespace std;
 5  //获取next数组
 6 int* getNext(string T)
 7 {
 8     int* next = new int[T.size()];    // 声明next数组
 9     int T_len = T.size();
10     int i = 0;    // T的下标
11     int j = -1;
12     next[0] = -1;
13     while (i < T_len)
14     {
15         if (j == -1 || T[i] == T[j])
16         {
17             if (T[i + 1] == T[j + 1])    //KMP优化
18                 next[++i] = next[++j];
19             else
20                 next[++i] = ++j;
21         }
22         else
23             j = next[j];
24     }
25     return next;
26 
27 }
28 
29 // KMP算法,在 S 中找到 T 第一次出现的位置 
30 int KMP(string S, string T)    // S为主串,T为模式串
31 {
32     int* next = getNext(T);
33     int i = 0;        // S下标
34     int j = 0;        // T下标
35     int s_len = S.size();
36     int t_len = T.size();
37     while (i < s_len && j < t_len)
38     {
39         if (j == -1 || S[i] == T[j])    //T 的第一个字符不匹配或S[i] == T[j]
40         {
41             i++;
42             j++;
43         }
44         else
45             j = next[j];        // 当前字符匹配失败,进行跳转
46     }
47     if (j == t_len)            // 匹配成功
48         return i - j;
49     return -1;
50 }
51 
52 
53 int main()
54 {
55     string S = "bbc abcdab abcdabcdabde";    
56     string T = "abcdabd";
57     int num = KMP(S, T);
58     cout << num<<endl;
59     system("pause");
60     return 0;
61 }

 

附录:关于前后缀

来一张图片说明吧:

由上图所得, "前缀" 指除了自身以外,一个字符串的全部头部组合;"后缀" 指除了自身以外,一个字符串的全部尾部组合。

参考文献:

[1]王红梅, 胡明, 王涛. 数据结构(C++版)[M]. 北京:清华大学出版社, 2011:83-85.

[2]唐小喵的博客:http://www.cnblogs.com/tangzhengyue/p/4315393.html#3831240


 

posted @ 2017-12-28 11:07  ~君莫笑~  阅读(6326)  评论(2编辑  收藏  举报