---

分析

二重循環會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 小球的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。