2019浙江省程序设计竞赛 K：Strings in the Pocket（马拉车算法）

版权声明：本文为博主原创文章，顺手点个赞叭~有问题欢迎指出(*╹▽╹*) https://blog.csdn.net/qq_41117236/article/details/89641231

【题面】

Strings in the Pocket

【题解】

题意：给定两个字符串s,t，能否通过一次反转操作使得s变成t，输出可能情况的数目。

思路：如果两个字符串相同，①不相同的部分不可通过反转操作转化，答案为0 ②不相等的部分可通过反转操作转化，则从不相等部分的两边往外匹配相同的部分，则答案就是往外匹配的长度+1；如果两个字符串相同，③用马拉车算法寻找以所有位置s[i]为中心的最长回文串，答案就是所有最长回文串长度/2求和的结果。

马拉车算法小结+模板题

【代码】

#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int mxn=2e6+5;
char s[mxn],t[mxn];
char tmp[2*mxn];
ll Len[2*mxn];
int len;
bool check(int l,int r)
{
    for(int i=l,j=r;i<=r;i++,j--)
        if(s[i]!=t[j])
            return false;
    return true;
}
void manacher()
{
    ll mx=0,pos=0;
    for(ll i=1;i<len;i++){
        if(i<mx) Len[i]=min(Len[2*pos-i],mx-i);
        else Len[i]=1LL;
        while(tmp[i-Len[i]]==tmp[i+Len[i]]) Len[i]++;
        if(mx<i+Len[i]){
            pos=i;
            mx=i+Len[i];
        }
    }
}
ll init()
{
    tmp[0]='@'; tmp[1]='#';
    int j=2;
    for(int i=0;i<len;i++){
        tmp[j++]=s[i];
        tmp[j++]='#';
    }
    tmp[j]=')';
    len=j;
    manacher();
    ll ans=0;
    for(int i=0;i<j;i++)
        ans+=Len[i]/2LL;
    return ans;
}
int main()
{
    int _; scanf("%d",&_);
    while(_--){
        scanf("%s%s",s,t);
        len=strlen(s);
        int l=-1,r=-1;
        int f=1;
        for(int i=0;i<len;i++)
            if(s[i]!=t[i]){
                f=0;
                if(l==-1) l=i;
                r=i;
            }
        ll ans;
        if(!f){
            if(!check(l,r)) ans=0;
            else{
                ans=1;
                for(int i=r+1,j=l-1;i<len,j>=0;i++,j--){
                    if(s[i]==s[j]) ans++;
                    else break;
                }
            }
        }
        else ans=init();
        printf("%lld\n",ans);
    }
}