【ybtoj 高效进阶 2.3】 【kmp】 周期长度和

【ybtoj 高效进阶 2.3】 【kmp】 周期长度和

题目

解题思路

题目出得奇奇怪怪，一点都不可爱
首先要将x的前缀都做一遍kmp
所谓最大周期即找一个最短的公共前后缀
这个前后缀可以通过一直向前那next值找到，可用类似并查集的方法优化
用总长减去这个前后缀的长度
图解
好吧，有点简陋

代码

#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
char x[1000200];
int n, k = 0, next[1000020];
long long ans;
int find(int y) {
    if (next[y] != 0)
        return next[y] = find(next[y]);
    return y;
}  //向前找最短的前后缀
int main() {
    scanf("%d", &n);
    for (int i = 1; i <= n; i++) cin >> x[i];
    for (int i = 1; i < n; i++) {
        while (k > 0 && x[i + 1] != x[k + 1]) k = next[k];
        if (x[i + 1] == x[k + 1])
            k++;
        next[i + 1] = k;
    }  //跑kmp
    for (int i = 1; i <= n; i++) ans += i - find(i);
    printf("%lld\n", ans);
    return 0;
}