字符串_马拉车算法

马拉车算法与kmp算法的优化方法基本一样,就是把已经求出来的回文片段用来算接下来要求的回文中心(来回颠倒),在此附上一道马拉车算法题目的代码。

点击查看代码
#include <bits/stdc++.h>
using namespace std;

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int t;
    cin >> t;
    while (t--) {
        string s;
        int n;
        cin >> n;
        cin >> s;
        vector<int> d1(n), d2(n);
        int L = 0, R = -1;
        for (int i = 0; i < n; i++) {
            int k;
            if (i > R) {
                k = 1;
            } else {
                int j = L + R - i;
                k = min(d1[j], R - i + 1);
            }
            while (i - k >= 0 && i + k < n && s[i - k] == s[i + k]) {
                k++;
            }
            d1[i] = k;
            if (i + k - 1 > R) {
                L = i - k + 1;
                R = i + k - 1;
            }
        }
        L = 0, R = -1;
        for (int i = 0; i < n; i++) {
            int k;
            if (i > R) {
                k = 0;
            } else {
                int j = L + R - i + 1;
                k = min(d2[j], R - i + 1);
            }
            while (i - k - 1 >= 0 && i + k < n && s[i - k - 1] == s[i + k]) {
                k++;
            }
            d2[i] = k;
            if (i + k - 1 > R) {
                L = i - k;
                R = i + k - 1;
            }
        }
        vector<int> le(n), ri(n);
        for (int i = 0; i < n; i++) {
            if (i + d1[i] - 1 < n) {
                ri[i + d1[i] - 1] = max(d1[i] * 2 - 1, ri[i + d1[i] - 1]);
            }
            if (i + d2[i] - 1 < n && i + d2[i] - 1 >= 0) {
                ri[i + d2[i] - 1] = max(d2[i] * 2, ri[i + d2[i] - 1]);
            }
            if (i - d1[i] + 1 >= 0) {
                le[i - d1[i] + 1] = max(d1[i] * 2 - 1, le[i - d1[i] + 1]);
            }
            if (i - d2[i] >= 0) {
                le[i - d2[i]] = max(d2[i] * 2, le[i - d2[i]]);
            }
        }
        for (int i = 0; i + 1 < n; i++) {
            le[i + 1] = max(le[i + 1], le[i] - 2);
        }

        for (int i = n - 1; i > 0; i--) {
            ri[i - 1] = max(ri[i - 1], ri[i] - 2);
        }
        int ans = 0;
        for (int i = 0; i < n - 1; i++) {
            ans = max(ri[i] + le[i + 1], ans);
        }
        cout << ans << '\n';
    }
    return 0;
}

d1[i]表示的是奇数回文,d2[i]表示的是偶数回文。
posted @ 2026-07-14 20:58  hybrz  阅读(0)  评论(0)    收藏  举报