题目:
题意:
有一个长度为 n n n的 01 01 01序列
我们需要求出有多少个三元组形式如 l , p , r ( l < p < r ) l,p,r(l<p<r) l,p,r(l<p<r)能够满足 s p = 1 s_p=1 sp=1且 [ l , p ] [l,p] [l,p]中的 1 1 1的个数与 [ p , r ] [p,r] [p,r]中 1 1 1的个数相同
分析:
我们会发现如果枚举 p p p的话时间复杂度是无法承受的,那么换一个思路,我们枚举 l l l
显然对于一个合法的三元组, l ∼ r l\sim r l∼r的 1 1 1个数一定是一个奇数
而 0 0 0显然是不决定是否合法的,我们就将 0 、 1 0、1 0、1合为一体,统计每个 1 1 1右边有多少个 0 0 0,这样在计算答案的时候能够统计完全贡献
当然对于 1 1 1左边的 0 0 0,我们可以正序扫一遍,边扫边统计,单独计算贡献
代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<algorithm>
#include<queue>
#define LL long long
using namespace std;
inline LL read()
{
LL s=0,f=1; char c=getchar();
while(c<'0'||c>'9') {
if(c=='-') f=-1;c=getchar();}
while(c>='0'&&c<='9') {
s=s*10+c-'0';c=getchar();}
return s*f;
}
char s[1000005];
LL a[2],r0[1000005];
int main()
{
freopen("puzzle.in","r",stdin);
freopen("puzzle.out","w",stdout);
LL op=read(),n=read(),len=0;
scanf("%s",&s);
for(LL i=0;i<n;i++) len+=s[i]-'0';
LL c=0;
for(LL i=n-1,j=len;i>=0;i--)
{
c++;
if(s[i]=='1') {
r0[j--]=c;c=0;}
}
for(LL i=1;i<=len;i+=2) a[1]+=r0[i];
for(LL i=2;i<=len;i+=2) a[0]+=r0[i];
c=0;
LL ans=0,j=0;
for(LL i=0;i<n;i++)
{
if(s[i]=='1')
{
j++;
a[j&1]-=r0[j];
ans+=c*(r0[j]-1)+(c+1)*a[j&1];
c=0;
}
else c++;
}
cout<<ans;
return 0;
}