SSL2520 2014年汕头市选拔赛普级组 小球（math）

2014年汕头市选拔赛普级组 小球

Description



分析：考虑第i个小球和它前面的小球对答案的贡献，我们知道Ci-Cj不是1就是0，当第Ci=0时前面有影响的小球Cj=1，当Ci=1时相反，所以只需要记录一下前i个小球有多少个为1，多少为0。那么i*j怎么算呢，显然i是不变的，那么我们只需要用数组记录前面有影响的小球位置和即可，ans=x*i-sum[i-1]，x为前面有影响的小球个数。

代码

#include <cstdio>
#define maxn 200000
#define ll long long
using namespace std;

ll sum1[maxn],sum0[maxn],a[maxn],b[maxn],x[maxn];
int n;

int main()
{
scanf("%d",&n);
for (int i=1;i<=n;i++)
{
scanf("%d",&x[i]);
if (x[i]==0)
{
a[i]=a[i-1]+1;
b[i]=b[i-1];
sum0[i]=sum0[i-1]+i;
sum1[i]=sum1[i-1];
}
else
{
a[i]=a[i-1];
b[i]=b[i-1]+1;
sum1[i]=sum1[i-1]+i;
sum0[i]=sum0[i-1];
}
}
long long ans=0;
for (int i=2;i<=n;i++)
if (x[i]==0) ans+=b[i-1]*i-sum1[i-1];
else ans+=a[i-1]*i-sum0[i-1];
printf("%lld",ans);
}