Codeforces 1828E Manacher + DP

题意

传送门 Codeforces 1828E Palindrome Partition

题解

观察到对于任一个 “beautiful string” $s[l\cdots r]$，都可以通过求解 $s$ 中以 $l$ 为左边界的最短回文串，将 $s[l\cdots r]$ 表示为唯一的分割形式。Manacher 求解以各个位置为中心的回文串最长长度，用一个栈维护有序的回文串中心位置，不断弹出无法覆盖当前位置的元素，则可以得到 $s$ 中以各个位置为左边界的最短回文串。DP 求解以各个位置为边界的答案即可。总时间复杂度 $O(n)$。

对于此类后推形式的 DP，倒序递推较为方便，可以以 $dp[n]$ 为边界情况；前推时可以以 $dp[0]$ 为边界情况。

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

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int tt;
    cin >> tt;
    while (tt--) {
        int n;
        cin >> n;
        string s;
        cin >> s;
        vector<int> d(n);
        for (int l = 0, r = 0, i = 0; i < n; ++i) {
            int k = r <= i ? 0 : min(d[l + r - i], r - i);
            while (i + k < n && i - 1 - k >= 0 && s[i + k] == s[i - 1 - k]) {
                k += 1;
            }
            if (r < i + k) {
                r = i + k;
                l = i - k;
            }
            d[i] = k;
        }
        vector<int> to(n, -1);
        vector<pair<int, int>> stk;
        for (int i = n - 1; i >= 0; --i) {
            while (!stk.empty() && i < stk.back().second) {
                stk.pop_back();
            }
            if (!stk.empty()) {
                int j = stk.back().first;
                to[i] = 2 * j - i;
            }
            stk.push_back({i, i - d[i]});
        }
        vector<int> dp(n + 1);
        for (int i = n - 1; i >= 0; --i) {
            int j = to[i];
            if(j != -1) {
                dp[i] = dp[j] + 1;
            }
        }
        ll res = 0;
        for (int i = 0; i < n; ++i) {
            res += dp[i];
        }
        cout << res << '\n';
    }

    return 0;
}