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);
}