z函数 扩展kmp

设字符数组 str[ ]

对于str中的每个str[i] z函数计算的是str[i ~ j]的长度，也就是以str[i]为首的子字符串对于str lcp的长度。str[i ~ j]为str的一个最长前缀，即j要尽可能的大。

在计算z函数之前，要了解z box。对于每一个i都存在一个z box(z box == str[l ~ r]), 满足z box是str的一个前缀，并且i∈[l, r]，还要满足r尽可能的大。（对于i位置的z box是i - 1位置上更新的z box）

z函数采用迭代的方式计算，可以认为在计算z[i]时， z[1 ~ i - 1]都是被计算过的

由于z box是str中的一个前缀, 因此str[1, 1 + r - l] == z box。因此z[i]  的计算可以看成对 z[i - l + 1]的讨论。这里分为三种情况: 

第一种情况是z[i - l + 1] < r - i + 1(不取等是为了满足r尽可能大这一性质) 因为前缀与z box相等，因此对于i位置的z[i] == z[i - l + 1]

第二种情况是z[i - l + 1] >= r - i + 1此时只能确定在z box内的字符串相等，超出z box的部分要枚举判断

第三种情况是i一开始就在z box的外面，此时直接枚举即可

代码：

#include <iostream>

using namespace std;

const int N = 1e5 + 10;

char str[N];
int z[N];
int n;

int main() {
    cin >> n >> str + 1;
    
    int l = 0, r = 0;
    z[1] = n;
   for(int i = 2; i <= n; i ++ ) {
        if(i > r) {//i在zbox外
            while(str[i + z[i]] == str[1 + z[i]]) z[i] ++;
            l = i, r = i + z[i] - 1;
        } else if(z[i - l + 1] < r - i + 1) z[i] = z[i - l + 1];//情况2
        else {//第3种情况
            z[i] = r - i;
            while(str[i + z[i]] == str[1 + z[i]]) z[i] ++;
            l = i, r = i + z[i] - 1;
        }
    } 
    
    for(int i = 1; i <= n; i ++ ) cout << z[i] << " ";
    return 0;
}