扩展kmp算法（z函数）

定义

资料来源：https://oi-wiki.org/string/z-func/

---

初号版本：O(n²)

就是纯暴力做法

void get_z(char *str){
    int len=strlen(str);
    z[0]=0;
    for (int i=1;i<len;i++){
        int j=i;
        while (str[j]==str[j-i]&&j<len) j++;
        z[i]=j-i;
    }
}

---

优化版本：O(n)

我们设定一个区间[L,R]表示目前最靠后的一个匹配段。z[i]储存i点的z函数。
那么根据z函数的定义有：当我遍历到第i个元素时，会有这几种情况：

• i≤R 目前这个i在以前的遍历过程中已经经过了。可以利用以前的结果减少计算量。因为区间[L,R]显然是和区间[0,R-L+1]是一样的（根据z函数的定义）所以z[i]是显然和z[i-L]一样的！ 但是z[i-L]有时是过大的，因为从i开始向后走不到z[i-L]可能就超过r了！(i+z[i-L]>=r).所以我们令z[i]=min(z[i-L],R-i+1)
• i>R 目前这个i无法利用以前遍历过程中的信息，我们只能从i使用朴素算法。并且注意更新[L,R],保证区间[L,R]始终是最靠后的一个匹配段。

void get_z(char *str){
    MT(z,0);
    int len=strlen(str);
    z[0]=0;
    int L=0,R=0;
    for (int i=1;i<len;i++){
        if (i<=R) z[i]=min(z[i-L],R-i+1);
        while (str[z[i]]==str[i+z[i]]&&(i+z[i]<len)) z[i]++;
        if (i+z[i]-1>R) L=i,R=i+z[i]-1;
    }
}

P.S证明复杂度的过程见https://oi-wiki.org/string/z-func/