CF傳送門

題目大意：給定一個長度為$n$的序列$a$，求${\sum }_{1\le l<r\le n}{\sum }_{l\le x<y\le r}[a_x=a_y]$，即求每個連續區間中相等數對個數之和。

思路：對于這種計數問題，如果直接求不好計算，一般求每個值對答案的貢獻。
首先考慮對于$a_i=a_j(i<j)$，那么其對答案產生的貢獻就是$i\times (n?j+1)$，那對于$a_i=a_j=a_k(i<j<k)$,其對答案多產生的貢獻顯然是$(i+j)\times (n?k+1)$
那么我們可以維護一個$map$，
$ans={\sum }_{i=1}^{n}map[a[i]]\times (n?i+1)$
每次更新$map[a[i]]+=i$來維護前綴和
$PS$：記得開$longlong$！！！
AC代碼：

#include <bits/stdc++.h> using namespace std; const int N=1e5+10; typedef long long ll;int t,n,a[N]; map<ll,ll>cnt;int main(){cin>>t;while(t--){cin>>n;for(int i=1;i<=n;i++) cin>>a[i];cnt.clear();ll ans=0;for(int i=1;i<=n;i++){ans+=cnt[a[i]]*(n-i+1);cnt[a[i]]+=i;}cout<<ans<<endl;}return 0; }

總結

以上是生活随笔為你收集整理的CF1527C Sequence Pair Weight的全部內容，希望文章能夠幫你解決所遇到的問題。

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