扩展kmp--Z函数

原创已于 2024-04-02 20:41:56 修改 · 386 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#算法

于 2024-04-02 15:02:05 首次发布

字符串专栏收录该内容

7 篇文章

订阅专栏

本文介绍了如何使用暴力循环和扩展KMP算法计算字符串的最长公共前缀。暴力方法通过逐个字符比较，而扩展KMP利用已知信息优化搜索过程。Manacher算法在此背景下有所启发。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1. 定义

我们约定对于字符串 $s$ ，长度为 $sz$

$s [i :]$ 为 $s_is_{i+1}\cdots s_{sz-1}$

$z$ 函数定义为

$s [0 :] 与 s [i :]$ 的最长公共前缀长度。

2. 实现

2.1 暴力

直接从前往后暴力循环

std::vector<int> ExtendKmp::z_function_trivial(std::string s) {

    int sz = static_cast<int>(s.size());
    std::vector<int> z(sz, 0);


    for (int i = 1;i < sz; ++i) {
        while ( i + z[i] < sz && s[z[i]] == s[i + z[i]])
                z[i]++;
    }


    return z;
}

2.2 扩展kmp

实际上与 $km p$ 好像没啥关系？感觉跟 $mana c h er$ 有点像。

核心的思想就是，当 $i + z [i] > j$ 时， $z [j] = z [j - i]$

假设前缀区间为 $[l, r]$

当 $r > i$ 时， $s [i, r] = s [i - l, r - l], z [i] = z [i - l]$
当 $r <= i$ 时，$$

std::vector<int> ExtendKmp::z_function(std::string s) {

    int sz = static_cast<int>(s.size());
    std::vector<int> z(sz, 0);

    int l = 0, r = 0;

    for (int i = 1; i < sz; ++i) {
        if ( i <= r && i + z[i - l] + 1 < r) {
            z[i] = z[i - l];
        } else{
            z[i] = std::max(0, r - i + 1);

            while (i + z[i] < sz && s[z[i]] == s[i + z[i]])
                ++z[i];
        }

        if ( i + z[i] > r) {
            l = i;
            r = i + z[i] ;
        }
    }

    return z;
}