Strings in the Pocket ZOJ - 4110 | manacher算法

第一道马拉车的题。。纪念一下花了一个小时学的🐎拉🚗算法

题意：
给你两个串 s 和 t ， 问能否翻转 s 串的子串让 s 和 t 一样，问可以翻转的种类有多少个？
分析：
如果两个字符串相同时用 Manacher 求出回文串的个数。如果不同的话，找到那部分串，以这个串为中心向两边扩展 （可以证明左右扩展总是同步的），如果两边是回文串，那就 ans ++。
代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
//typedef __int128 lll;
#define close() ios::sync_with_stdio(0), cin.tie(0), cout.tie(0)
#define mem(a, b) memset(a, b, sizeof(a))
const ll mod = 1e9 + 7;
const int maxn = 2e6 + 10;
const int inf = 0x3f3f3f3f;
char a[maxn], b[maxn];
char ma[maxn << 1];
int mp[maxn << 1];

ll manacher(char s[])
{
    int l = 0;
    ma[l++] = '$', ma[l++] = '#';
    for(int i = 0; s[i]; i++)
        ma[l++] = s[i], ma[l++] = '#';
    ma[l] = 0;
    int maxx = 0, pos = 0;
    ll res = 0;
    for(int i = 0; i < l; i++)
    {
        mp[i] = maxx > i ? min(mp[2 * pos - i], maxx - i) : 1;
        while(ma[i + mp[i]] == ma[i - mp[i]]) mp[i]++;
        if(i + mp[i] > maxx) maxx = i + mp[i], pos = i;
    }
    for(int i = 0; i < l; i++) res += mp[i] / 2;
    return res;
}

int main()
{
    int t; cin >> t;
    while(t--)
    {
        scanf("%s%s", a, b);
        int len = strlen(a);
        int l = 0, r = len - 1;
        for(; l < len; l++) if(a[l] != b[l]) break;
        for(; r >= 0; r--) if(a[r] != b[r]) break;
        if(r == -1)
        {
            cout << manacher(a) << endl;
            continue;
        }
        int flag = 0;
        for(int i = l; i <= r; i++)
            if(a[i] != b[l + r - i]) 
            {
                flag = 1;
                break;
            }
        if(flag)
        {
            cout << 0 << endl;
            continue;
        }
        int sum = 1;
        l--, r++;
        while(l >= 0 && r <= len - 1)
            if(a[l] == b[r])
                sum++, l--, r++;
            else break;
        cout << sum << endl;
    }
}