KMP算法

2019-08-13  本文已影响0人  Collie

写在前面

这篇文章针对的是C/C++/Java语言程序,所以我们的下标从零开始。

KMP算法的改进

KMP算法的改进主要体现在当模式串中下标为j的字符p_j与主串中的字符s_i失配时,在一定条件下可以从p_k开始匹配。而不需要从头开始匹配。所以我们需要解决的问题是,当模式串中下标为j的字符p_j与主串中的字符s_i失配时,应使用模式串中哪一个字符与s_i继续比较。

假设当模式串中下标为j的字符p_j与主串中的字符s_i失配时,应使用模式串中下标为k(k < j)的字符p_ks_i继续比较。

递推生成下标跳转表

假设存在线性表nextIndex,其中k = nextIndex[ j ]代表当模式串中下标为j的元素与主串中的元素发生失配时,主串中的这一元素应与下标为k的元素尝试匹配。该线性表称为“下标跳转表”,为优化随机访问,可以采用顺序存储结构。
特别地,若采用有符号数据类型作为下表跳转表的元素,则可以令nextIndex[0] = -1表示指向主串中被比较元素的指针应右移一位,以s_{i+1}与模式串中第一个元素p_0进行比较。
根据上文的分析,我们应在j > 0的前提下进行分析,尝试递推。否则p_{0}...p_{k-1} = p_{j-(k-1)}...p_{j-1}关系不存在,最终产生的跳转表为从0n - 1递增的等差数列,原因从略。显然有nextIndex[1] = 0

假设已知k = nextIndex[ j ],其中j > 0

  1. (k = next[ j ]) > 0时,那么则说明模式串中存在如下关系
    p_{0}...p_{k-1} = p_{j-(k-1)}...p_{j-1}
    1. p_k = p_j,则有p_{0}...p_k= p_{j-(k-1)}...p_{j}
      此时对于p_{j+1}next[ j + 1 ] = k + 1
    2. p_k \neq p_j,则
      p_{0}...p_{k-1} = p_{j-(k-1)}...p_{j-1} \\ p_{0}...p_k \neq p_{j-(k-1)}...p_{j}
      相当于以模式串对模式串本身进行匹配,在p_k处发生了失配。
      此时可能存在k' < k使得p_{0}...p_{k'} = p_{j-(k'-1)}...p_{j},若满足这一条件,说明next[ j + 1 ] = k' + 1可以通过对k进行迭代的方法尝试找到满足的k'
      若不存在k'满足上述条件,则令next[ j + 1 ] = 0
  2. (k = next[ j ]) == 0
    (k = next[ j ]) == 0代表当模式串中p_j与主串中s_i失配时,需要跳转到模式串的第一个字符p_0与主串中的s_i进行比较。
    当模式串中p_{j+1}与主串中的s_i失配时,必有p_j = s_{i-1}。此时若有p_j = p_0,则有s_{i-1} = p_0。因此,若p_j = p_0,则next[ j + 1 ] = 1,否则,next[ j + 1 ] = 0

据此,可以写出递推生成“下标跳转表”的KMP算法C++例程如下

inline std::vector<int> generateNextArray(
                const std::string &pattern
        );
        
        inline int match(
                const std::string &s, const std::string &pattern
        )
        {
            std::vector<int> next = generateNextArray(pattern);
            for (
                    int i = 0, j = 0; i < s.length();
                    )
            {
                if (s[i] == pattern[j])
                {
                    if (j == pattern.length() - 1)
                        return (i - j);//found
                    else ++i, ++j;
                }
                else
                {
                    j = next[j];
                    if (j == -1)
                    {
                        ++i, ++j;
                    }
                }
            }
            return -1;
        }
        
        
        inline std::vector<int> generateNextArray(
                const std::string &pattern
        )
        {
            std::vector<int> next(pattern.length());
            next[0] = -1;
            next[1] = 0;
            for (
                    std::vector<int>::size_type i = 1;
                    i < (pattern.length() - 1);
                    ++i
                    )
            {
                for (
                        int k = next[i];; k = next[k]
                        )
                {
                    if (pattern[i] == pattern[k])
                    {
                        next[i + 1] = k + 1;
                        break;
                    }
                    else if (k == 0)
                    {
                        next[i + 1] = 0;
                        break;
                    }
                    //else continue;
                }
            }
            return next;
        }
上一篇 下一篇

猜你喜欢

热点阅读