差值中位数（二分法）

题意

给出一个长度为 N 的数组 a，并用这个数组生成一个新数组 ans[i]，满足ans[]=abs(a[i]-a[j])（i != j），求出数组 ans 的中位数。
输入要求：
多组输入，每次输入一个 N ，表示有 N 个数，之后输入一个长度为N的序列 a。
输出要求：
输出新数组ans的中位数
sample input：

4
1 3 2 4
3
1 10 2

sample output：

1
8

思路

• 暴力解法：将ans所有值计算出来，排序后求中位数，显然复杂度过高，不可取。
• 采用二分法：
• 将数组a排序后，假设 j > i，a[j] - a[i] <= P，a[j] <= P + a[i]，可以通过判断在各种不同的 i 时，共有多少 j 可以满足 a[j] - a[i] <= P，如此，可以判断 P 在 ans 数组中的位置，在前半部分还是后半部分，就可以进行二分。
• 定义了两个函数：find 用于查找数组 a 中第一个大于等于 t 的元素的位置，check 用于判断 val 在 ans 中处于什么位置，两个函数代码如下：

int find(int t) {//找到数组a中第一个大于等于t的元素的位置
	int l = 0, r = n - 1, ans = -1;
	while (l <= r) {
		int min = (l + r) / 2;
		if (a[min] >= t) {
			ans = min;
			r = min - 1;
		}
		else {
			l = min + 1;
		}
	}
	return ans;
}

bool check(int val) {//返回true：中位数比val大		返回false：中位数比val小
	int count = 0;
	for (int i = 0; i < n - 1; i++) {
		int t = find(a[i] + val);
		if (t != -1)
			count += n - find(a[i] + val);
	}
	return count > m;
}

• 通过这两个函数，我们可以将中位数的大小一直缩小，直至得到精确的中位数为止。

代码

#include <vector>
#include <algorithm>
#include <iostream>
using namespace std;

int a[100000];
int m, n;
int ans = -1;

int find(int t) {//找到数组a中第一个大于等于t的元素的位置
	int l = 0, r = n - 1, ans = -1;
	while (l <= r) {
		int min = (l + r) / 2;
		if (a[min] >= t) {
			ans = min;
			r = min - 1;
		}
		else {
			l = min + 1;
		}
	}
	return ans;
}

bool check(int val) {//返回true：中位数比val大		返回false：中位数比val小
	int count = 0;
	for (int i = 0; i < n - 1; i++) {
		int t = find(a[i] + val);
		if (t != -1)
			count += n - find(a[i] + val);
	}
	return count > m;
}


int main() {
	while (scanf_s("%d", &n) != EOF) {
		for (int i = 0; i < n; i++) {
			scanf_s("%d", &a[i]);
		}
		sort(a, a + n);
		m = n * (n - 1) / 4;
		int l = 0, r = a[n - 1] - a[0];
		while (l <= r) {
			int min = (l + r) / 2;
			if (check(min)) {
				l = min+1;
				ans = min;
			}
			else {
				r = min - 1;
			}
		}
		cout << ans << endl;
	}
}