题目
Monocarp 有一个由 n 个整数组成的数组 a。让我们将 k 表示为这些元素的数学平均值(请注意,k 可能不是整数)。
由 n 个元素组成的数组的数学平均值是元素的总和除以这些元素的数量(即总和除以 n)。
Monocarp 想要从 a 中恰好删除两个元素,以便剩余 (n-2) 个元素的数学平均值仍然等于 k。
您的任务是计算位置 [i,j] (i<j) 对的数量,如果删除这些位置上的元素,则剩余 (n-2) 个元素的数学平均值等于 k(即,它等于原始数组 a) 的 n 个元素的数学平均值。
输入
第一行包含一个整数 t (1≤t≤104)——测试用例的数量。
每个测试用例的第一行包含一个整数 n (3≤n≤2⋅105)——数组中元素的数量。
第二行包含一个整数序列 a1,a2,…,an (0≤ai≤109),其中 ai 是数组的第 i 个元素。
所有测试用例的 n 总和不超过 2⋅105。
输出
打印一个整数 — 位置对的数量 [i,j] (i<j) 使得如果删除这些位置上的元素,则剩余 (n-2) 个元素的数学平均值等于 k(即,它等于原始数组 a) 的 n 个元素的数学平均值。
题解思路
将序列排序,求出全部的平均值 , 我们枚举每个数 ,二分查找出 之后的符合 相加也是平均值的数 , upper 是求大于等于的数 lower 是求大于的数 。
相减就是等于的数 。
这题卡longlong 的,因为n为 2e5 方案数可能会达到 1e10左右 。
AC代码
#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <vector>
#include <algorithm>
#include <map>
#include <string>
using namespace std;
const int INF = 0x3f3f3f3f;
double a[200100] ;
int main ()
{
ios::sync_with_stdio(false);
int T ;
cin >> T ;
while ( T -- )
{
long double sum = 0 ;
long long n , ans = 0 ;
cin >> n ;
for (int i = 1 ; i <= n ; i++ )
{
cin >> a[i] ;
sum += a[i] ;
}
sort( a + 1 , a + 1 + n ) ;
double sk = (2.0)*sum / n ;
for (int i = 1 ; i <= n ; i++ )
{
ans += upper_bound(a+1+i , a+1+n , sk - a[i] ) - lower_bound(a+1+i , a+1+n , sk - a[i] ) ;
}
cout << ans <<"\n" ;
}
return 0 ;
}