---

分析

二重循环会TLE，所以就要思考如何优化。（研究了很久）
当$i<j$时，（$i=j$不可能，$i>j$会重复）且颜色不同
那么$ans+=j-i$
一重循环时j的增加数量取决于i的个数，而i前缀和就好了。
用两个数组a和b，a[j]表示Ci为j时的下标和，b[j]表示Ci为j时的个数。然后水水地到了第一。

---

代码

#include <cstdio> #include <cctype> using namespace std; long long ans,a[2],b[2],n; int in(){int ans=0; char c=getchar();while (!isdigit(c)) c=getchar();while (isdigit(c)) ans=ans*10+c-48,c=getchar();return ans; } int main(){n=in();for (int i=1;i<=n;i++){ bool x=in(); a[x]+=i; b[x]++;ans+=i*b[1-x]-a[1-x];}return !printf("%lld",ans); }

总结

以上是生活随笔为你收集整理的ssl 2520 小球的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得生活随笔网站内容还不错，欢迎将生活随笔推荐给好友。