CF383A Solution

题解

对于一对方向相反的牛，它们之间的顺序是无关紧要的，因为一定有一头会看到对方被挤奶，所以只需要考虑方向相同的牛。对于向左的牛，从右到左挤奶，反之亦然。

代码

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=2e5+10;
int a[N],l1[N];
inline int read()
{
	int s=0,w=1; char ch=getchar();
	while(ch<'0' || ch>'9') {if(ch=='-') w=-1; ch=getchar();}
	while(ch>='0' && ch<='9') s=s*10+ch-'0',ch=getchar();
	return s*w;
}
signed main()
{
	int n=read(),cnt=0;
	for(int i=1;i<=n;i++) a[i]=read(),l1[i]=l1[i-1]+a[i];
	for(int i=n;i>=1;i--)
		if(!a[i]) cnt+=l1[i];
	printf("%lld",cnt); 
	return 0;
}