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
HINT
Source
鸣谢 JZP
定义 1=0,0=1 1 = 0 , 0 = 1 跑manacher即可
#include<bits/stdc++.h>
using namespace std;
#define rep(i,j,k) for(int i = j;i <= k;++i)
#define repp(i,j,k) for(int i = j;i >= k;--i)
#define rept(i,x) for(int i = linkk[x];i;i = e[i].n)
#define P pair<int,int>
#define Pil pair<int,ll>
#define Pli pair<ll,int>
#define Pll pair<ll,ll>
#define pb push_back
#define pc putchar
#define mp make_pair
#define file(k) memset(k,0,sizeof(k))
#define ll long long
const int q = 19930726;
int n;
char s[2001000];
int p[2001000];
int read(){int sum = 0;char c = getchar();bool flag = true;while(c < '0' || c > '9') {if(c == '-') flag = false;c = getchar();}while(c >= '0' && c <= '9') sum = sum * 10 + c - 48,c = getchar();if(flag) return sum;else return -sum;}
bool check(int x,int y)
{
if(x == '1' && y == '0') return true;
if(x == '0' && y == '1') return true;
if(x == '#' && y == '#') return true;
return false;
}
void manachar()
{
int pos = 0 , mx = 0;
rep(i,1,n)
{
if(i%2==0) continue;
p[i]=i<mx?min(p[pos*2-i],mx-i):1;
while(check(s[i-p[i]],s[i+p[i]])) p[i]++;
if(i + p[i] > mx) mx = i + p[i] , pos = i;
}
}
int main()
{
n = read();
rep(i,1,n) {scanf("%c",&s[2*i]);s[2*i-1]='#';}s[2*n+1] = '#';
n*=2;n++;s[0] = '?';s[n+1] = '!';
manachar();
ll ans = 0;
rep(i,1,n) if(i%2 == 1) ans += (p[i]-1)/2;
printf("%lld\n",ans);
return 0;
}