1. 程式人生 > >BZOJ 2084 [Poi2010]Antisymmetry(manacher)

BZOJ 2084 [Poi2010]Antisymmetry(manacher)

bsp 鏈接 namespace manacher pre .com cnblogs 條件 clas

【題目鏈接】 http://www.lydsy.com/JudgeOnline/problem.php?id=2084

【題目大意】

  對於一個01字符串,如果將這個字符串0和1取反後,
  再將整個串反過來和原串一樣,就稱作“反對稱”字符串。
  比如00001111和010101就是反對稱的,1001就不是。
  現在給出一個長度為N的01字符串,求它有多少個子串是反對稱的。

【題解】

  修改manacher的判定條件,對該串進行計算即可。

【代碼】

#include <cstdio>
#include <cstring>
using namespace std;
const int N=1000010; 
int n,m,i,r,p,f[N<<1];
long long ans;
char a[N],s[N<<1];
int min(int a,int b){return a<b?a:b;}
bool check(char x,char y){
    if(x==‘#‘&&y==‘#‘)return 1;
    if((x-‘0‘)+(y-‘0‘)==1)return 1;
    return 0;
}
void manacher(char *a){
    for(i=1;i<=n;i++)s[i<<1]=a[i],s[i<<1|1]=‘#‘;
    s[0]=‘$‘,s[1]=‘#‘,s[m=(n+1)<<1]=‘&‘;
    for(r=p=0,f[1]=1,i=1;i<m;ans+=f[i++]>>1){
        for(f[i]=r>i?min(r-i,f[p*2-i]):0;check(s[i-f[i]],s[i+f[i]]);f[i]++);
        if(i+f[i]>r)r=i+f[i],p=i;
        //printf("%d\n",f[i]); 
    }
}
int main(){
    scanf("%d",&n);
    scanf(" %s",a+1); 
    manacher(a);
    printf("%lld\n",ans);
    return 0;    
}

BZOJ 2084 [Poi2010]Antisymmetry(manacher)