Codeforces Round 936 (Div. 2) - A. Median of an Array （数学）

$$A.MedianofanArray$$
$$timelimitpertest:1second$$
$$memorylimitpertest:256megabytes$$
$$input:standardinput$$
$$output:standardoutput$$

给你一个由 $n$ 个整数组成的数组 $a$ 。

数组 $q_1,q_2,\dots ,q_k$ 的中位数是 $p_{\lceil \frac{k}{2}\rceil }$ ，其中 $p$ 是按非递减顺序排序的数组 $q$ 。例如，数组 $[9,5,1,2,6]$ 的中位数是 $5$ ，如在排序数组 $[1,2,5,6,9]$ 中，索引 $\lceil \frac{5}{2}\rceil =3$ 处的数字是 $5$ ；数组 $[9,2,8,3]$ 的中位数是 $3$ ，如在排序数组 $[2,3,8,9]$ 中，索引 $\lceil \frac{4}{2}\rceil =2$ 处的数字是 $3$ 。

您可以选择一个整数 $i$ ( $1\le i\le n$ )，并在一次操作中将 $a_i$ 增加 $1$ 。

你的任务是找出增加数组中位数所需的最少操作次数。

注意数组 $a$ 不一定包含不同的数。

输入

每个测试由多个测试用例组成。第一行包含一个整数 $t$ ( $1\le t\le 10^4$ ) - 测试用例的数量。然后是测试用例的描述。

每个测试用例的第一行包含一个整数 $n$ ( $1\le n\le 10^5$ ) - 数组 $a$ 的长度。

每个测试用例的第二行包含 $n$ 个整数 $a_1,a_2,\dots ,a_n$ ( $1\le a_i\le 10^9$ ) - 数组 $a$ 。

保证所有测试用例中 $n$ 的值之和不超过 $2\cdot 10^5$ 。

输出

对于每个测试用例，输出一个整数 - 增加数组中位数所需的最少操作数。

---

在排序之后，位于 $\lceil \frac{n}{2}\rceil$ 下标处的数就被认为是中位数，暂时使这个下标为 $p$ ，如果想要改变中位数的数值，那么就要使得 $p$ 下标之后的数都满足大于或者等于 $a[p]$ ，因为只有这样才能够满足数组的有序顺序不会改变。

所以直接先取出中位下标然后取扫后面有多少数和中位数相等。
不用管大于中位数的数是因为，就算中位数加 $1$ 了之后他们也是大于等于中位数，不会影响整体顺序。

代码：

#include<iostream>
#include<vector>
#include<cmath>
#include<algorithm>
using namespace std;
const int N = 1e5+10;

int a[N];

void solve(){
	int n;cin >> n;
	for(int i = 1;i <= n;i++)cin >> a[i];

	int mid = (n+1)/2;//取出中位下标

	int res = 1;//先提前加上中位数本身应该加的1

	for(int i = mid + 1;i <= n;i++){
		if(a[i] == a[mid])res++;
	}
	cout << res << endl;
	return;
}

int main(){
	int t;cin >> t;

	while(t--)
		solve();

	return 0;
}