东方博宜OJ 2362：前缀和后缀 ← KMP算法

【题目来源】
https://oj.czos.cn/p/2362

【题目描述】
给定若干由小写字母组成的字符串（这些字符串总长≤4×10^5），在每个字符串中求出所有既是前缀又是后缀的子串长度。例如：ababcababababcabab，既是前缀又是后缀的：ab，abab，ababcabab，ababcababababcabab。

【输入格式】
输入若干行，每行一个字符串。

【输出格式】
对于每个字符串，输出一行，包含若干个递增的整数，表示所有既是前缀又是后缀的子串长度。

【输入样例】
ababcababababcabab
aaaaa

【输出样例】
2 4 9 18
1 2 3 4 5

【数据范围】
字符串总长≤4×10^5。

【算法分析】
● next[] 数组的涵义：++next[i] 表示字符串前 i 个字符的最长公共前后缀长度++。

● 为什么递归 ne[] 数组就能找到所有答案？
答：字符串 ababcababababcabab 的 next数组值为：0 1 1 2 3 1 2 3 4 5 4 5 4 5 6 7 8 9。

|--------|---|---|---|---|---|---|---|---|---|----|----|----|----|----|----|----|----|----|
| idx | 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | 10 | 11 | 12 | 13 | 14 | 15 | 16 | 17 | 18 |
| T | a | b | a | b | c | a | b | a | b | a | b | a | b | c | a | b | a | b |
| ne[] | 0 | 1 | 1 | 2 | 3 | 1 | 2 | 3 | 4 | 5 | 4 | 5 | 4 | 5 | 6 | 7 | 8 | 9 |

ne[18] = 9 → 前 18 个字符的最长相等前后缀长度 = 9
ne[9] = 4 → 前 9 个字符的最长相等前后缀长度 = 4
ne[4] = 2 → 前 4 个字符的最长相等前后缀长度 = 2
ne[2] = 0 → 停止
所以所有答案：2 → 4 → 9 → 18

● KMP算法的next数组与前缀表的关系

【算法代码】

cpp 复制代码

#include <bits/stdc++.h>
using namespace std;

const int N=4e5+5;
int ne[N];

void getNext(string t) {
    int len=t.length();
    int i=0,j=-1;
    ne[0]=-1;
    while(i<len) {
        if(j==-1 || t[i]==t[j]) {
            i++,j++;
            ne[i]=j;
        } else j=ne[j];
    }
}

void print(int x) {
    if(x==0) return;
    print(ne[x]);
    cout<<x<<" ";
}

int main() {
    string t;
    while(cin>>t) {
        int len=t.size();
        getNext(t);
        print(ne[len]);
        cout<<len<<endl;
    }
    return 0;
}

/*
in:
ababcababababcabab
aaaaa

out:
2 4 9 18
1 2 3 4 5
*/

【参考文献】
https://blog.csdn.net/hnjzsyjyj/article/details/160215718
https://blog.csdn.net/hnjzsyjyj/article/details/127140892
https://blog.csdn.net/hnjzsyjyj/article/details/146059543