马拉车 ZOJ 4110_zoj - 4110-CSDN博客

本文链接：https://blog.csdn.net/flyzer/article/details/89676544

锴迪的马拉车

时间复杂度：O ( n )

#include<iostream>
#include<iomanip>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<string>
#include<queue>
#include<stack>
#include<vector>
#include<map>
#include<set>
#include<sstream>
#include<cstdlib>
#include<time.h>
#define rep(i,a,n) for(int i=a;i<=n;i++)
#define repp(i,n,a) for(int i=n;i>=a;i--)
#define ll long long
#define ull unsigned long long
#define mem(a,x) memset((a),(x),sizeof ((a)))//x只能是0或-1或false或true
#define debug(x) cout<<"X: "<<(x)<<endl
#define de cout<<"************"<<endl
#define lowbit(x) ((x)&(-x))
#define lson rt<<1
#define rson rt<<1|1
#define gcd(a,b) __gcd(a,b)
#define lcm(a,b) a*b/(__gcd(a,b))
#define inf 0x3f3f3f3f//1e9+6e7
#define Eps 1e-8
#define Mod 1e9+7
#define maxn 100010
const double pi=acos(-1.0);
using namespace std;
const int N=2000010;
char a[N],b[N];
char str[N<<1];    //str为构造出来的字符串
int hw[N<<1];  //hw[i]表示的是以i为对称轴的最长回文子串的长度
int len;
//数组全是从1开始的
void manacher()
{
    //hw数组不用初始化
    for(int i=1; i<=len; i++) str[i*2-1]='%',str[i*2]=a[i];
    str[len=len*2+1]='%';
    int id=0,mx=0;
    for(int i=1; i<=len; i++)
    {
        if(i<mx)
            hw[i]=min(hw[id*2-i],mx-i);
        else
            hw[i]=1;
        //对于hdu4513，要求前半段递增，需要在下面加上&&str[i-hw[i]]<=str[i-hw[i]+2]
        while(hw[i]+i<=len && i-hw[i]>=1 && str[i+hw[i]]==str[i-hw[i]])
            hw[i]++;
        if(hw[i]+i>mx)
            id=i,mx=i+hw[i];
    }
    for(int i=1;i<=len;i++)
        hw[i]--;
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        getchar();
        ll ans=0;
        scanf("%s%s",a+1,b+1);
        len=strlen(a+1);
        if(strcmp(a+1,b+1)==0)
        {
            manacher();
            rep(i,1,len)
                ans+=(hw[i]+1)/2;
            printf("%lld\n",ans);
            continue;
        }
        int l,r;
        rep(i,1,len)
            if(a[i]!=b[i])
                {l=i;break;}
        repp(i,len,1)
            if(a[i]!=b[i])
                {r=i;break;}
        int f=0;
        rep(i,l,r)
        {
            if(a[i]==b[r-i+l])
                continue;
            else
                {f++;break;}
        }
        int cnt=0;
        if(f==0)
        {
            for(int i=l-1,j=r+1; i>=1&&j<=len; i--,j++)
            {
                if(a[i]==a[j])
                    cnt++;
                else
                    break;
            }
            cnt++;
        }
        if(f==0)
            printf("%d\n",cnt);
        else
            printf("0\n");
    }
    return 0;
}