[bzoj2084][雜湊]Antisymmetry
阿新 • • 發佈:2018-12-12
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;
}