C. Sequence Pair Weight（组合计数）

Posted 2021-06-18 CCSU_Cola

题目链接

题意：给一个长度为n的序列，问所有的字串的贡献总和为多少？若$ai==aj$，且$i!=j$,该子串的贡献加1。

例：

4
1 2 1 1
的贡献总和为6。
[1,2] having 0 valid unordered pairs;
[2,1] having 0 valid unordered pairs;
[1,1] having 1 valid unordered pair;
[1,2,1] having 1 valid unordered pairs;
[2,1,1] having 1 valid unordered pair;
[1,2,1,1] having 3 valid unordered pairs.

思路：两个相等的数字前后分别可以去掉$i$个和$(n-i+1)$个，所以根据这个来算贡献即可。每一个出现的数乘上之前出现的相同的数的$i$即为答案。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
int p[100010];
map<ll,ll>mp;
int main(){
	int t;
	scanf("%d",&t);
	while(t--){
	int n;
	mp.clear();
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		scanf("%d",&p[i]);
	}
	ll ans=0;
	for(int i=1;i<=n;i++){
		ans+=(n-i+1)*mp[p[i]];
		mp[p[i]]+=i;
	}
	printf("%lld\\n",ans);
	}
	return 0;
}