2084: [Poi2010]Antisymmetry
Time Limit: 10 Sec Memory Limit: 259 MBSubmit: 432 Solved: 273
[ Submit][ Status][ Discuss]
Description
对于一个01字符串,如果将这个字符串0和1取反后,再将整个串反过来和原串一样,就称作“反对称”字符串。比如00001111和010101就是反对称的,1001就不是。
现在给出一个长度为N的01字符串,求它有多少个子串是反对称的。
Input
第一行一个正整数N (N <= 500,000)。第二行一个长度为N的01字符串。
Output
一个正整数,表示反对称子串的个数。
Sample Input
8
11001011
11001011
Sample Output
7
hint
7个反对称子串分别是:01(出现两次), 10(出现两次), 0101, 1100和001011
hint
7个反对称子串分别是:01(出现两次), 10(出现两次), 0101, 1100和001011
HINT
Source
方法一:二分+哈希
枚举中间点,然后二分长度,用哈希判断是否满足反对称。
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#define F(i,j,n) for(int i=j;i<=n;i++)
#define D(i,j,n) for(int i=j;i>=n;i--)
#define ll long long
#define ull unsigned long long
#define maxn 500005
#define base 233
using namespace std;
int n,a[maxn],b[maxn];
ll ans;
ull fa[maxn],fb[maxn],p[maxn];
char s[maxn];
inline ull hasha(int l,int r)
{
return fa[r]-fa[l-1]*p[r-l+1];
}
inline ull hashb(int l,int r)
{
return fb[l]-fb[r+1]*p[r-l+1];
}
int main()
{
scanf("%d",&n);
scanf("%s",s+1);
F(i,1,n) a[i]=s[i]-'0',b[i]=a[i]^1;
F(i,1,n) fa[i]=fa[i-1]*base+a[i];
D(i,n,1) fb[i]=fb[i+1]*base+b[i];
p[0]=1;F(i,1,n) p[i]=p[i-1]*base;
F(i,1,n-1)
{
int l=1,r=min(i,n-i),mid,tmp=0;
while (l<=r)
{
mid=(l+r)>>1;
if (hasha(i-mid+1,i+mid)==hashb(i-mid+1,i+mid)) tmp=mid,l=mid+1;
else r=mid-1;
}
ans+=tmp;
}
printf("%lld\n",ans);
return 0;
}
方法二:manacher
令0=1,0≠0,1≠1,#=#,然后跑manacher。
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#define F(i,j,n) for(int i=j;i<=n;i++)
#define D(i,j,n) for(int i=j;i>=n;i--)
#define ll long long
#define ull unsigned long long
#define maxn 1000005
using namespace std;
int n,f[maxn];
ll ans;
char ch[maxn],s[maxn];
inline bool judge(char a,char b)
{
return (a=='#'&&b=='#')||(a=='0'&&b=='1')||(a=='1'&&b=='0');
}
inline void manacher()
{
int mx=0,id=0;
F(i,1,n)
{
if (mx>i) f[i]=min(f[id*2-i],mx-i);
else f[i]=0;
while (judge(s[i-f[i]],s[i+f[i]])) f[i]++;
if (i+f[i]>mx) mx=i+f[i],id=i;
ans+=(f[i]>>1);
}
}
int main()
{
scanf("%d",&n);
scanf("%s",ch+1);
s[1]='#';
F(i,1,n) s[i<<1]=ch[i],s[i<<1|1]='#';
n=n<<1|1;
manacher();
printf("%lld\n",ans);
return 0;
}