Educational Codeforces Round 115 (Rated for Div. 2) C 二分函数lower_bound upper_bound 的结合运用

题目

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 ;
}