KMP——强大的next数组

\(KMP\) 的原理不在这里仔细讲了，主要说说最近刷题总结出的 \(next\) 数组的强大功能。
部分例题来自《信息学奥赛一本通》的配套练习。

---

基于定义——字符串相同先后缀

“基于定义”：咱们求的 \(next\) 数组就是字符串到某一位时最长相同先后缀的长度。
注意 \(next\) 数组求的为“最长”的，那若是想知道一个字符串全部相同的先后缀长度咋办？

举个栗子：

假设一个 \(n\) 位的字符串（下标从 \(1\) 到 \(n\)），\(next[n]=p\)
那么该字符串的子串 \([1,p]\) 与 \([n-p+1,n]\) 应是相同的
设 \(next[p]=q\) ，那么子串 \([1,q]\) 与 \([p-q+1,p]\) 是相同的
综上，子串 \([1,q]\) 与 \([n-q+1,n]\) 是相同的，即 \(next[next[n]]\) 也是该字符串相同先后缀长度

就这样 \(next\) 一遍遍向前找，直到某一位的 \(next\) 为 \(0\)， 拓展出一棵 \(next\) 树（也叫 \(fail\) 树）。

例题 \(bzoj3620\)

\(PROBLEM:\)
求一个长度为 \(n\) 的字符串全部形似 \(A+B+A\) , 且 \(len(A) \geq k,len(B) \geq 1\) 的子串数目。
\(n \leq 15000\)

\(SOLUTION:\)
一个奇妙的事情是这个题 \(O(n^2)\) 能过。
因而枚举每一位为起点，\(KMP\) 的过程当中，\(next\) 值至关于这一段子串 \(len(A)\) 的最大值
若是它小于 \(k\) ，显然不行。
而若 \(它 \times 2+1 > 子串长度\) 也不行。
因此须要找到合适的 \(len(A)\) 知足 \(len(A) \geq k\) 且 \(len(A) \times 2 +1 \leq 子串长度\)
这就用到 \(next\) 树的思想了！
还有一个小优化，用一个数组记录某一段 \(\geq k\) 的最短的相同先后缀长度，把它做为 \(len(A)\) 判断比较快。

代码：

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
 
using namespace std;
 
const int N = 15005;
 
char s[N];
int nxt[N],KK,ok[N],ans;
 
void KMP(char p[]){
    int len=strlen(p+1),k=0;
    nxt[1]=0; ok[0]=ok[1]=-1;
    for(int i=2;i<=len;i++){
        while(k && p[i]!=p[k+1]) k=nxt[k];
        if(p[i]==p[k+1]) k++;
        nxt[i]=k;
         
        if(k<KK) { ok[i]=-1; continue; }
        if(ok[k]==-1) ok[i]=k;
        else ok[i]=ok[k];
        if(ok[i]*2+1<=i) ans++;
    }
}
 
int main()
{
    int len;
    scanf("%s",s+1);
    scanf("%d",&KK);
    len=strlen(s+1);
     
    for(int i=1;i<=len;i++) {
        if(KK*2+1>len-i+1) break;
        KMP(s+i-1);
    }
    printf("%d\n",ans);
     
    return 0;
}

---

拓展功能——字符串循环节

“拓展”：这里的主角为 \(n-next[n]\)

仍是举个栗子：

上图中 \(n-next[n]=3\) ，那 \(3\) 是什么呢？
看那些棕圈圈，\(3\) 其实能够叫作字符串的 “类”循环节，由于字符串并非由这个循环节完完整整组成的。

而若一个字符串有真正的循环节要知足什么条件呢？
答案是 \(n-next[n]\) 整除 \(n\)
一样举个栗子就明了了：

对于全部字符串， \(n-next[n]\) 只是它最短的循环节（类循环节），其余循环节（类循环节）的长度经过 \(next\) 一遍遍向前找求出。
仍是 \(next\) 树的思想，结合栗子便可证实，这里就不赘述了。

！！！
注意：有真正循环节的字符串，全部循环节长度都为最短循环节长度的倍数。而类循环节并不知足这一性质！

例题1 \(bzoj1511\)

\(PROBLEM:\)
一个串是有限个小写字符的序列,特别的,一个空序列也能够是一个串. 一个串 \(P\) 是串 \(A\) 的前缀, 当且仅当存在串 \(B\) , 使得 \(A = PB\). 若是 \(P \neq A\) 而且 \(P\) 不是一个空串,那么咱们说 \(P\) 是 \(A\) 的一个 \(proper\) 前缀. 定义 \(Q\) 是 \(A\) 的周期, 当且仅当 \(Q\) 是 \(A\) 的一个 \(proper\) 前缀而且 \(A\) 是 \(QQ\) 的前缀(不必定要是 \(proper\) 前缀). 好比串 \(abab\) 和 \(ababab\) 都是串 \(abababa\) 的周期. 串 \(A\) 的最大周期就是它最长的一个周期或者是一个空串(当 \(A\) 没有周期的时候), 好比说, \(ababab\) 的最大周期是 \(abab\). 串 \(abc\) 的最大周期是空串. 给出一个串,求出它全部前缀的最大周期长度之和.
\(串长度 \leq 10^6\)

\(SOLUTION:\)
其实题中说的最大周期就是 \(\neq A\) 的最长“类循环节”
\(next\) 树的思想，用一个数组记录每一个“点”在该“树”上最小的非零祖先，不然会超时

#include<cstdio>
#include<iostream>
#include<algorithm>

using namespace std;

const int N = 1000005;
typedef long long ll;

int n;
int nxt[N],snxt[N];
char s[N];

int main()
{
    scanf("%d",&n);
    scanf("%s",s+1);
    
    ll ans=0;
    int k=0;
    nxt[1]=0; snxt[1]=1;
    for(int i=2;i<=n;i++){
        while(k && s[i]!=s[k+1]) k=nxt[k];
        if(s[i]==s[k+1]) k++;
        nxt[i]=k; 
        snxt[i]=(k?snxt[k]:i);
        ans+=i-snxt[i];
    }
    printf("%lld\n",ans);
    
    return 0;
}

例题2 \(bzoj4974\)

\(PROBLEM:\)
一个串 \(T\) 是 \(S\) 的循环节，当且仅当存在正整数 \(k\)，使得 \(S\) 是 \(T^k\) (即 \(T\) 重复 \(k\) 次)的前缀，好比 \(abcd\) 是 \(abcdabcdab\) 的循环节。给定一个长度为 \(n\) 的仅由小写字符构成的字符串 \(S\)， 请对于每一个 \(k(1 \leq k \leq n)\)，求出 \(S\) 长度为 \(k\) 的前缀的最短循环节的长度 \(per_i\) 。小 \(Q\) 告诉你 \(n\) 以及 \(per_1,per_2,...,per_n\)，请找到一个长度为 \(n\) 的小写字符串 \(S\)，使得 \(S\) 能对应上 \(per\) 。
\(n \leq 10^5\)

\(SOLUTION:\)
能够发现，\(per_i\) 值其实就是最短“类循环节”长度，也就是 \(n-next[i]\)
因而咱们能够求出全部 \(next\) 值，而后进行逆向 \(KMP\) ，得出原字符串。

代码：

#include<cstdio>
#include<iostream>
#include<algorithm>
 
using namespace std;
 
const int N = 100005;
 
int n;
int nxt[N],vis[26];
char s[N];
 
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",&nxt[i]),nxt[i]=i-nxt[i];
     
    s[1]='a';
    for(int i=2;i<=n;i++){
        if(nxt[i]!=0) { s[i]=s[nxt[i]]; continue; }
        for(int j=0;j<26;j++) vis[j]=0;
        int k=nxt[i-1];
        while(k!=0) vis[s[k+1]-'a']=1,k=nxt[k];
        vis[s[k+1]-'a']=1;
        for(int j=0;j<26;j++)
            if(!vis[j]) { s[i]='a'+j; break; }
    }
    printf("%s",s+1);
     
    return 0;
}