【ACWing】3188. manacher算法

题目地址：

https://www.acwing.com/problem/content/description/3190/

给定一个长度为$n$的由小写字母构成的字符串，求它的最长回文子串的长度是多少。

输入格式：
一个由小写字母构成的字符串。

输出格式：
输出一个整数，表示最长回文子串的长度。

数据范围：
$1\le n\le 10^7$

可以用Manachar算法。先将整个字符串做如下处理，将这个字符串每两个字符之间插入一个不在字母表里的字符，比如#，然后在首尾也添加#，再在首尾添加两个不同字符，不能是字母表或者是#，比如可以在开头添加$结尾添加^。例如，对于字符串aab，处理完之后字符串就变为了$#a#a#b#^。令处理完的字符串为$t$。那么，每个$s$里的回文子串都能对应于$t$的某个长度为奇数的回文子串（我们要求在$t$里取的子串以#开头结尾），接下来考虑对$t$求每个位置为中心的情况下最大的回文子串半径，设$p[i]$为以$t[i]$为中心的最长回文子串的半径，$p[0]=0$，为无效值。同时维护两个变量$mr$和$mi$，分别是右端点最靠右的最长回文子串的中点位置和右端点位置 + 1，接下来递推。从左向右遍历到$t[i]$，如果$i<mr$，那么找到$j=2mi-i$，我们证明 p [ i ] ≥ min ⁡ { p [ j ] , m r − i } p[i]\ge \min\{p[j], mr-i\} p[i]≥min{p[j],mr−i}。如果$p[j]\le mr-i$，那么以$i$为中点的最长回文子串半径$p[i]$显然可以取到$p[j]$，并且在以$j$为中点的最长回文子串到达边界的时候，$p[i]$有可能更大；如果$p[j]>mr-i$，那么以$i$为中点的最长回文子串右边界可以取并且最多取到$mr$，否则就会得出$mr$可以继续拓展的结论，矛盾。无论怎样，$p[i]$都可以从 min ⁡ { p [ j ] , m r − i } \min\{p[j], mr-i\} min{p[j],mr−i}开始递增枚举。如果$i\ge mr$，那么$p[i]$可以从$1$开始枚举。当枚举停下来的时候（即再扩大就不回文的时候），更新$mr$和$mid$。求得的所有$p[i]$的最大值如果是$mp$的话，容易验证，$mp-1$就是$s$的最长回文子串的长度。代码如下：

#include <iostream>  
#include <cstring>
using namespace std;

const int N = 2e7 + 10;
int n;
char a[N], b[N];
int p[N];
int res;

// 构造新串
void init() {
    int k = 0;
    b[k++] = '$', b[k++] = '#';
    for (int i = 0; i < n; i++) b[k++] = a[i], b[k++] = '#';
    b[k++] = '^';
    n = k;
}

void manacher() {
    int mr = 0, mid;
    // 最后的'^'就不用枚举了
    for (int i = 1; i < n - 1; i++) {
    	// 初始化p[i]
        if (i < mr) p[i] = min(p[mid * 2 - i], mr - i);
        else p[i] = 1;
		
		// 向外扩展
        while (b[i - p[i]] == b[i + p[i]]) p[i]++;

        if (i + p[i] > mr) {
            mr = i + p[i];
            mid = i;
        }

        res = max(res, p[i] - 1);
    }
}

int main() {
    scanf("%s", a);
    n = strlen(a);
    init();
    manacher();
    
    printf("%d\n", res);
    return 0;
}

时空复杂度$O(n)$。