POI2010 Antisymmetry

Antisymmetry

对于一个01字符串，如果将这个字符串0和1取反后，再将整个串反过来和原串一样，就称作“反对称”字符串。比如00001111和010101就是反对称的，1001就不是。

现在给出一个长度为N的01字符串，求它有多少个子串是反对称的。

输入

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

输出

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

Antisymmetry和回文串比较像,所以可以用Manacher算法解决.
Manacher主要是利用了回文串的对称性,利用求过的回文串来求之后的回文串,
它通过记录回文串达到的最右端和这个回文串的中点来实现上面的想法.

如:

$$?????XXXXLLLLLLXXXmidXXXRRRRRXXX?????$$
X 表示到达最右端的回文串覆盖区域,
mid表示它的中点
R 表示求的回文.
所以R和L是相同的,R可以直接从L得到.
(直到R要超出X,才需要再算)

反对称的要求就是从中间向两边扩展的每两个数相加都是$1$,
如

$$01010010110101$$
(它就满足这个要求)
把这个作为匹配回文串的要求就可以了Manacher了.

但是Manacher时还要注意两点:
1) 由于我们需要的是长度为偶数的回文串,所以只需考虑 ‘#’ 的位置(就是Manacher添加的那些字符).
2) 答案要 long long (-_-||)

#include <cstdio>
#include <iostream>
using namespace std;
const int M=500005;
char num[M*2];
int R[M*2];
int n,len;

bool chk(int x) {
    if (num[x-R[x]]!=num[x+R[x]]) return num[x-R[x]]-'0'+num[x+R[x]]-'0'==1;
    else return num[x+R[x]]=='#';
}

int main() {
    char str[M];
    scanf("%d%s",&n,str);
    num[0]='!';
    for (int i=0;i<n;++i) {
        num[(i<<1)+1]='#';
        num[(i<<1)+2]=str[i];
    }
    num[(n<<1)+1]='#';
    len=(n<<1)+1;
    long long ans=0;
    int pos=0,mx=0;
    for (int i=1;i<=len;i+=2) {
        R[i]= mx>i?min(R[(pos<<1)-i],mx-i):1;
        while (chk(i)) R[i]++;
        if (R[i]+i>mx&&(i&1)) {
            mx=R[i]+i;
            pos=i;
        }
        if (i&1) ans+=(R[i]-1)>>1;
    }
    cout<<ans<<endl;
    return 0;
}