2018ICPC 南京 Mediocre String Problem 扩展KMP + Manacher

本文链接：https://blog.csdn.net/weixin_43485513/article/details/109488231

本文介绍了一种计算特定字符串组合成回文串的算法。通过将字符串SSS和TTT进行特殊处理，利用扩展KMP算法计算每个后缀与TTT串的最长公共前缀，并结合回文串特性，实现高效求解。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目大意为计算 $S$ 串的子串 $s$ 串和 $T$ 中的前缀 $t$ 串拼起来是回文串的种数的总贡献，要求 $s$ 串长度大于 $t$ ， $s + t$ 是回文串， $s$ 和 $t$ 串非空，拼起来的串视作不同种当且仅当 $s$ 或 $t$ 串的位置不同。

我们把 $s$ 串分成 $s_1+s_2$ 两个部分，其中 $s_1=reverse(t)$ ， $s_2=palindrome$ 。显而易见的，我们可以把整个 $S$ 串倒过来，计算它的每个后缀与 $T$ 串的最长公共前缀，也就是扩展 $K M P$ 做的事（我这里将 $S$ 串和 $T$ 串按 $T +^{'} ∣^{'} + S$ 连接，这样求出 $e x$ 数组，就求出了 $S$ 串的每个后缀在 $T$ 串中的最长公共前缀，而 $^{'} ∣^{'}$ 则用来避免他们计算错误）。

这样我们就得到了每个 $s_1$ 串，此时的 $s$ 串变为 $s_2+s_1$ ， $s_1=t$ ， $s_2=palindrome$ 。我们只要再求出倒置后的 $S$ 串中的回文串，然后差分贡献即可（每个回文串对它的右半径都有贡献，用差分才能在线性时间内求出）。

接下来就是求 $S$ 串中的回文串，然后添加贡献，再遍历 $S$ 串。累加差分值，当前字符为字母时计算对于当前的 $S$ 串的后缀的贡献（为什么当前字符为字母才算，因为当前字符为回文串的右边界时，才计算贡献，而在实际的 $S$ 串中，右边界只会是字母）。

$t o l$ 为以当前字符为右边界的回文串总数， $e x$ 为 $S$ 串的每个后缀与 $T$ 串的最长公共前缀，显而易见的 $e x$ 的每个长度都能产生一个贡献，所以总贡献就是 $e x [i + 1] * t o l$ （为什么是 $i + 1$ ，因为是当前 $i$ 视作回文串的一部分，所以起始位置为 $i + 1$ 的后缀才用来当 $s_1$ 串）。

挺好的一道题，ICPC的题目质量还是不错的

#include<bits/stdc++.h>
using namespace std;
#define endl '\n'
#define IOS     ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
typedef long long LL;
const int maxn = 2e6 + 5;
char s[maxn], t[maxn];
int r[maxn], ex[maxn], sum[maxn];
int mid, rb, slen, tlen, len, cnt;
inline void init(){
    tlen = strlen(t);
    cnt = 1;
    s[0] = '~';
    s[1] = '|';
    for(int i = tlen - 1; ~i; i--){
        s[++cnt] = t[i];
        s[++cnt] = '|';
    }
    for(int t = 1; t <= cnt; t++){
        if(t < rb) r[t] = min(r[(mid<<1) - t], rb - t);
        else     r[t] = 1;
        while(s[t - r[t]] == s[t + r[t]]) r[t]++;
        if(r[t] + t > rb) rb = r[t] + t, mid = t;
        sum[t]++, sum[t + r[t]]--;
    }
}

inline void getex(){
    int n = len;
    for(int i = 1, l = 0, r = 0; i < n; i++){
        if(i <= r) ex[i] = min(r - i + 1, ex[i - l]);
        while(i + ex[i] < n && s[ex[i]] == s[i + ex[i]]) ex[i]++;
        if(i + ex[i] - 1 > r) l = i, r = i + ex[i] - 1;
    }
}

int main(){
    IOS;
    cin >> t;
    init();
    cin >> s;
    slen = strlen(s);
    s[slen++] = '|';
    len = slen + tlen;
    for(int i = 0; slen < len; i++, slen++) s[slen] = t[tlen - i - 1];
    getex();
    slen = len - tlen - 1;
    LL tol = 0, ans = 0;
    for(int i = 1; i <= cnt; i++){
        tol += sum[i];
        if(i % 2) continue;
        ans += ex[slen + 1 + i / 2] * tol;
    }
    cout << ans << endl;
}