1. 程式人生 > >[bzoj2084][雜湊]Antisymmetry

[bzoj2084][雜湊]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

題解

一個合法的串一定是一個偶數串 並且01對應… 取個反 就是問你迴文串的長度 雜湊二分一下沒了

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#include<queue>
#include<vector>
#include<ctime>
#include<map>
#define LL long long
#define mp(x,y) make_pair(x,y)
#define ULL unsigned long long
using namespace
std; const ULL HA=569; inline int read() { int f=1,x=0;char ch=getchar(); while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();} while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();} return x*f; } inline void write(int x) { if(x<0)putchar('-'),x=-x; if(x>9)write(x/10); putchar
(x%10+'0'); } inline void print(int x){write(x);printf(" ");} ULL h1[510000],h2[510000],tmp[510000]; char ch[510000]; ULL get1(int l,int r){return h1[r]-h1[l-1]*tmp[r-l+1];} ULL get2(int l,int r){return h2[l]-h2[r+1]*tmp[r-l+1];} int main() { int len=read(); tmp[0]=1;for(int i=1;i<=500000;i++)tmp[i]=tmp[i-1]*HA; scanf("%s",ch+1); for(int i=1;i<=len;i++)h1[i]=h1[i-1]*HA+ch[i]-'0'; for(int i=len;i>=1;i--)h2[i]=h2[i+1]*HA+((ch[i]-'0')^1); LL ans=0; for(int i=1;i<len;i++) if(ch[i]!=ch[i+1]) { int l=1,r=min(len-(i+1)+1,i),ret; while(l<=r) { int mid=(l+r)/2; if(get1(i-mid+1,i)==get2(i+1,i+1+mid-1))ret=mid,l=mid+1; else r=mid-1; } ans+=(LL)ret; } printf("%lld\n",ans); return 0; }