Description
给出一个长度为n的01串,问其有多少子串满足反串等于该串按位取反后的串
Input
第一行一整数n表示串长,之后一个长度为n的01串(1<=n<=5e5)
Output
输出满足条件的子串数
Sample Input
8
11001011
Sample Output
7
Solution
满足条件的子串必为偶数长度且从中心开始往两边是一个0一个1,用Manacher求最长回文串的方法,更新时只更新以填充位为中心的位置(偶数长度),往两边拓展时有三种情况:## 01 10,最后统计这种“回文串”的数量即可
Code
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
#define maxn 555555
char ss[2*maxn];
int ma[2*maxn];
void Manacher(char *s,int len)
{
int l=0;
ss[l++]='$';
ss[l++]='#';
for(int i=0;i<len;i++)
{
ss[l++]=s[i];
ss[l++]='#';
}
ss[l]=0;
int mx=0,id=0;
for(int i=0;i<l;i++)
{
ma[i]=mx>i?min(ma[2*id-i],mx-i):1;
while(ss[i+ma[i]]=='#'&&ss[i-ma[i]]=='#'||ss[i+ma[i]]=='0'&&ss[i-ma[i]]=='1'||ss[i+ma[i]]=='1'&&ss[i-ma[i]]=='0')ma[i]++;
if(ss[i]=='#'&&i+ma[i]>mx)
{
mx=i+ma[i];
id=i;
}
}
}
char s[maxn];
int main()
{
int n;
while(~scanf("%d",&n))
{
scanf("%s",s);
Manacher(s,n);
ll ans=0;
for(int i=3;i<2*n;i+=2)ans+=ma[i]/2;
printf("%lld\n",ans);
}
return 0;
}