SP687 Repeats

news2024/12/15 14:14:58

题意

给定字符串，求重复次数最多的连续重复子串

思路

有两个方法

方法1

直接枚举单个循环节的长度 $ki$ 。

然后将字符串划分成 $n / ki$ 个区间，这样如果只用枚举每一个区间就可以做到 $O (n l o g n)$ 。

考虑到一个重复子串如果循环了 $t$ 次，那么有至少 $t - 1$ 个区间是被完全覆盖的。我们只需要用哈希判断相邻的两个区间是不是一样的。

假设现在连续 $d$ 个区间是一样的，那么我们现在的目标就是确定这个子串到底是循环了 $d$ 还是 $d + 1$ 。因为我们已经可以确定循环节，因此左右两边需要添加的字符都是固定的，我们只用再在相邻的两个区间上二分+哈希，就能求出这一段最多扩展的长度，然后就可以判断了。

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

const int N = 5e4 + 10;
const ll mod = 998244353;

int n; char s[N];
ll val[N], hsh[N];

ll gethsh(int l, int r) {
    return (hsh[r] - hsh[l - 1] * val[r - l + 1] % mod + mod) % mod;
}

void solve() {
    cin >> n;
    char ss[2];
    val[0] = 1;
    for (int i = 1; i <= n; i++) {
        cin >> ss;
        val[i] = val[i - 1] * 357 % mod;
        hsh[i] = (hsh[i - 1] * 357 + ss[0]) % mod;
    }
    int ans = 1;
    // int ki = 3, l = 5, r;
    for (int ki = 1; ki <= n / 2; ki++) {
        for (int l = 1, r; l <= n; l = r + 1) {
            r = l + ki - 1;
            ll flag = gethsh(l, r);
            while (r + ki <= n && gethsh(r + 1, r + ki) == flag) {
                r = r + ki;
            }
            int zuo, you;
            int L = max(1, l - ki), R = l - 1, mid, pos = l;
            while (L <= R) {
                mid = (L + R) / 2;
                if (gethsh(mid, l - 1) == gethsh(mid + ki, l - 1 + ki)) {
                    pos = mid;
                    R = mid - 1;
                }
                else {
                    L = mid + 1;
                }
            }
            zuo = pos;
            L = r + 1, R = min(n, r + ki), pos = r;
            while (L <= R) {
                mid = (L + R) / 2;
                if (gethsh(r + 1, mid) == gethsh(r + 1 - ki, mid - ki)) {
                    pos = mid;
                    L = mid + 1;
                }
                else {
                    R = mid - 1;
                }
            }
            you = pos;
            ans = max(ans, (you - zuo + 1) / ki);
        }
    }
    cout << ans << endl;
}

int main() {
    // freopen("in.in", "r", stdin);
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int T; cin >> T;
    while (T--) {
        solve();
    }
}