这是一道马拉车的变异版,我们只需要在统计回文串的时候将判断条件改为右侧加左侧等于1即可,我们还要注意不可以以数字为中心扩张,因为可能会出现左侧是对称而右侧不是的情况。
#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<iomanip>
#include<cstring>
#include<string>
#include<ctime>
#include<cmath>
#include<algorithm>
using namespace std;
char s[2000000];
char ns[2000000];
int f[2000000];
int main()
{
int n;
scanf("%d",&n);
scanf("%s",s+1);
int len=strlen(s+1);
int p=0,mx=0;
int top=0;
ns[top]='$';
ns[++top]='#';
for(int i=1;i<=len;i++)
{
ns[++top]=s[i]-'0';
ns[++top]='#';
}
ns[++top]='@';
long long ans=0;
for(int i=1;i<=top;i+=2)
{
int tmp=1;
if(i<mx)
{
int o=2*p-i;
tmp=min(mx-i+1,f[o]);
}
while((i-tmp>=0) && (ns[i-tmp]=='#' || ns[i-tmp]+ns[i+tmp]==1)) tmp++;
f[i]=tmp;
if(i+tmp-1>mx)
{
mx=i+tmp-1;
p=i;
}
if(ns[i]=='#') ans+=f[i]/2;
}
cout<<ans;
return 0;
}