题意
给出一个长度为 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;
}
}