Codeforces Round #721 (Div. 2) C. Sequence Pair Weight(map)

Posted 2021-09-21 issue是fw

LINK

考虑$a_j==a_i(i>j)$时造成的贡献为

$j\ast (n-i+1)$

因为当一个子数组左端点小于等于$j$,右端点大于等于$i$就可以包含这个对

而且前面有若干个位置等于$a_i$,这些位置是$j_1,j_2.....$

也就是和他们造成的贡献和是$(n-i+1)\ast \sum j$

用个$map$记录一下就好了

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int maxn = 3e5+10;
int n,a[maxn];
unordered_map<int,int>mp;
signed main()
{
	int t; cin >> t;
	while( t-- )
	{
		cin >> n;
		for(int i=1;i<=n;i++)	cin >> a[i];
		int ans = 0;
		for(int i=1;i<=n;i++)
		{
			ans += mp[a[i]]*(n-i+1);
			mp[a[i]] += i;
		}
		cout << ans << endl;
		mp.clear();
	}
}