洛谷P3501

Description

對於一個 \(0/1\) 串，若是取反後再將整個串反過來和原串同樣，就稱做「反對稱」字符串

給出一個長度爲 \(n\) 的 \(0/1\) 串，求它有多少個反對稱子串

Solution

哈希題

不難想到，預先處理出正向原串哈希值與反向取反串的哈希值

常規作法，暴力枚舉區間斷定，哈希值相等則加一

可是數據範圍 \(1\le n\le 500\ 000\) 顯然會超時

因而考慮二分

二分枚舉每一個區間的對稱軸，每次對答案的貢獻就是其半徑

Code

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#define maxn 1000100
#define LL long long
#define uLL unsigned long long

using namespace std;

int n;
LL ans;
char s[maxn];
uLL b=1e9+7,Get[maxn],val[maxn],V[maxn];

inline int read(){
	int s=0,w=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-') w=-1;ch=getchar();}
	while(ch>='0'&&ch<='9') s=(s<<1)+(s<<3)+ch-'0',ch=getchar();
	return s*w;
}

inline int gethash2(int l,int r){return V[l]-V[r+1]*Get[r-l+1];}
inline int gethash1(int l,int r){return val[r]-val[l-1]*Get[r-l+1];}
inline bool check(int l,int r) {return gethash1(l,r)==gethash2(l,r);}

inline int erfen(int i){
	int l=0,r=min(i,n-i),Ans=0;
	while(l<=r){
		int mid=(l+r)/2;
		if(check(i-mid+1,i+mid)){l=mid+1;Ans=mid;} 
		else r=mid-1;
	}
	return Ans;
}

int main(){
	n=read();scanf("%s",s+1);
	int len=strlen(s+1);Get[0]=1;
	for(register int i=1;i<=n;i++){
		val[i]=val[i-1]*b+(uLL)s[i];	
		Get[i]=Get[i-1]*b;
	}
	for(register int i=n;i>=1;i--){
	    if(s[i]=='1') V[i]=V[i+1]*b+(uLL)'0';
		else V[i]=V[i+1]*b+(uLL)'1';
	}
//	for(register int i=1;i<n;i++) 
//	    for(register int j=i+1;j<=n;j++)
//	    		if(gethash1(i,j)==gethash2(i,j)) ans++;
    for(register int i=1;i<=n;i++) ans+=erfen(i);
	printf("%lld",ans);
	return 0;
}