马拉车算法——求回文子串个数zoj4110

zoj的测评姬好能卡时间。。

求回文子串的个数：只要把p[i]/2就行了：

　　如果s_new[i]是‘#’，算的是没有中心的偶回文串

　　反之是奇回文串

/*
给定两个字符串s,t
结论：s，t不相同的第一个字符下标为l，最后一个字符下标为r
         如果l==r，那么不存在解
        如果l<r,那么翻转s的一个子串时，一定是将s[l]和s[r]互相翻转
反证：假设存在s[l]和s[r+k]的翻转方法，
    因为s[r+k]=t[r+k]=t[l],且s[l]=t[r+k]，可得s[l]=t[l]，矛盾
l-k同理

所以本题只需要考虑三种情况：
    1.两个串相等：求s的回文子串
    2.l==r,无解
    3.l<r,首先s[l,r]翻转后和t[l,r]相等，然后找以s[l,r]为中心的回文子串
*/
#include<bits/stdc++.h>
using namespace std;
#define maxn 5000006
char s[maxn],t[maxn],s_new[maxn];
int p[maxn];

int init(){
    int len=strlen(s),j=;
    s_new[]='$',s_new[]='#';
    for(int i=;i<len;i++){
        s_new[j++]=s[i];
        s_new[j++]='#';
    }
    s_new[j]=;
    return j;
}
long long manacher(){
    long long res=;
    int len=init(),mx=,id;
    for(int i=;i<len;i++){
        if(mx>i)p[i]=min(p[*id-i],mx-i);
        else p[i]=;
        while(s_new[i-p[i]]==s_new[i+p[i]])p[i]++;
        if(mx<i+p[i])mx=i+p[i],id=i;
        res+=(long long )p[i]/;
    }
    return res;
}
int main(){
    int T;cin>>T;
    while(T--){
        scanf("%s%s",s,t);
        int len=strlen(s),l=0x3f3f3f3f,r=;
        for(int i=;i<len;i++)
            if(s[i]!=t[i]){l=i;break;}
        for(int i=len-;i>=;i--)
            if(s[i]!=t[i]){r=i;break;}

        if(l==r){puts("");continue;}
        if(l>r){cout<<manacher()<<endl;continue;}

        int flag=;
        for(int i=l,j=r;i<=r;i++,j--)
            if(s[i]!=t[j])flag=;
        if(flag){puts("");continue;}
        else {
            int ans=;
            while(l!= && r!=len-){
                --l;++r;
                if(s[l]==s[r])ans++;
                else break;
            }
            cout<<ans<<endl;
        }
    }
}