BZOJ2084[Poi2010] Antisymmetry

BZOJ2084[Poi2010] Antisymmetry

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

Solution：

挺不错的一道题..（其实是因为你自己YY出来了吧）

最开始想枚举一点，计算另一点，然后就怎么也弄不出来。后来可以发现一些性质：

• 反对称子串长度一定是偶数（这不废话吗）
• 反对称子串的中心串一定是反对称子串：设原串的区间为$[L,R]$，则$[L+1,R-1],[L+2,R-2],[L+3,R-3]$等等都是反对称子串。

一旦发现了这第二个性质，就非常简单了。枚举中心点，二分答案最长的以此为中心的反对称子串，哈希一下就好了。复杂度$O(nlogn)$。

话说这题好像用Manacher算法可以做到$O(n)$，以后再说吧。

#include<stdio.h>
#include<iostream>
#define P 1000000007
#define B 200019
#define M 500005
#define ll long long
using namespace std;
int Hash1[M],Hash2[M],Base[M],n;
char str[M],str1[M];
bool check(int i,int j){
    int t1=((Hash1[j]-1LL*Hash1[i]*Base[j-i])%P+P)%P;
    int t2=((Hash2[n-i+1]-1LL*Hash2[n-j+1]*Base[j-i])%P+P)%P;
    return t1==t2;
}
void swap(char &a,char &b){char t=a;a=b;b=t;}
int min(int a,int b){return a<b?a:b;}
int main(){
    ll ans=0;
    scanf("%d",&n);
    scanf("%s",str+1);
    Base[0]=1;
    for(int i=1;i<=n;i++)
        Base[i]=1LL*Base[i-1]*B%P;
    for(int i=1;i<=n;i++)
        Hash1[i]=(1LL*Hash1[i-1]*B+str[i])%P;
    for(int i=1;i<=n;i++)
        str1[i]=(str[i]=='0'?'1':'0');
    int i1=1,j1=n;
    while(i1<=j1)swap(str1[i1++],str1[j1--]);
    for(int i=1;i<=n;i++)
        Hash2[i]=(1LL*Hash2[i-1]*B+str1[i])%P;
    for(int i=1;i<n;i++){
        int j=i+1;
        if(str[i]==str[i+1])continue;
        int L=0,R=min(i-1,n-i-1),res=L;
        while(L<=R){
            int mid=(L+R)>>1;
            if(check(i-mid,i+1+mid)){
                L=mid+1;
                res=mid;
            }else R=mid-1;
        }
        ans+=res+1;
    }
    cout<<ans<<endl;
    return 0;
}