【bzoj2084】【poi2010】【Antisymmetry】【manacher】

Description

对于一个01字符串，如果将这个字符串0和1取反后，再将整个串反过来和原串一样，就称作“反对称”字符串。比如00001111和010101就是反对称的，1001就不是。
现在给出一个长度为N的01字符串，求它有多少个子串是反对称的。

Input

第一行一个正整数N (N <= 500,000)。第二行一个长度为N的01字符串。

Output

一个正整数，表示反对称子串的个数。

Sample Input

8
11001011

Sample Output

7

hint

7个反对称子串分别是：01(出现两次), 10(出现两次), 0101, 1100和001011

题解：规定0=1,0!=0,1!=1跑manacher即可。

代码：

#include<iostream>
#include<cstdio>
#define N 2000000 
using namespace std;
char s[N],ss[N];
int n,mx,p,f[N];
int manacher(){
  long long ans(0);
  ss[0]='$';ss[1]='#';mx=1;p=1;
  for (int i=1;i<=n;i++) ss[i<<1]=s[i],ss[i<<1|1]='#';n=n*2+1;
  for (int i=1;i<=n;i++){
    f[i]=max(0,min(mx-i,f[2*p-i]));
    while(ss[f[i]+i]=='#'&&ss[i-f[i]]=='#'||max(ss[f[i]+i],ss[i-f[i]])=='1'&&min(ss[f[i]+i],ss[i-f[i]])=='0') f[i]++;
    if (i+f[i]>mx) mx=f[i]+i,p=i;
    ans+=f[i]>>1;
  }
  return ans;
}
int main(){
  scanf("%d",&n);scanf("%s",s+1);
  printf("%lld",manacher());
}