题目
TT 是一位重度爱猫人士,每日沉溺于 B 站上的猫咪频道。有一天,TT 的好友 ZJM 决定交给 TT 一个难题,如果 TT 能够解决这个难题,ZJM 就会买一只可爱猫咪送给 TT。
任务内容是,给定一个 N 个数的数组 cat[i],并用这个数组生成一个新数组 ans[i]。新数组定义为对于任意的 i, j 且 i != j,均有 ans[] = abs(cat[i] – cat[j]),1 <= i < j <= N。试求出这个新数组的中位数,中位数即为排序之后 (len+1)/2 位置对应的数字,’/’ 为下取整。
TT 非常想得到那只可爱的猫咪,你能帮帮他吗?
Input
多组输入,每次输入一个 N,表示有 N 个数,之后输入一个长度为 N 的序列 cat, cat[i] <= 1e9 , 3 <= n <= 1e5
Output
输出新数组 ans 的中位数
Sample Input
4
1 3 2 4
3
1 10 2
Sample Output
1
8
题目思路
1、可以使用暴力枚举出结果数组ans,再找到中位数。但时间复杂度太高;
2、使用二分查找,先确定P的取值范围,在这个范围内二分查找P是否是中位数;
3、如何确定当前P是否是中位数
·去掉绝对值 Xj - Xi = P
·移项 Xj = Xi +P
·枚举Xi,计算有多少个Xj满足 Xj>=Xi+P
·判断根据计算结果判断当前P在中位数左边还是右边(结果数组长度为n(n-1)/2,结果数组的中有(n(n-1)+1)2个数大于等于中位数P)
4、在计算满足上述不等式Xj的个数时,可以利用二分查找,自己实现一个find函数,也可以使用lower_bound函数返回第一个大于等于该数字的下标,时间复杂度也是logn。
5、其实在枚举P的过程中,有许多不合法的P(真实不存在结果数组中,但存在于P的取值范围内),但是分析以后可以知道,我们要找的是第一次出现中位数排名的数字,也就是说,在P取值范围内,P的左边的数字求得的满足Xj个数与P可能一致,但是并不会跳过P去找到不合法的值。
代码实现
#include<stdio.h>
#include<algorithm>
using namespace std;
int n;
int main()
{
while(~scanf("%d",&n))
{
int cat[n];
for(int i=0;i<n;i++)
{
scanf("%d",&cat[i]);
}
sort(cat,cat+n);
int l=0;//错在这里了.....
int r=cat[n-1]-cat[0];
int num=(n*(n-1)/2)/2;
while(l+1<r){
int mid=(r+l)>>1;
int m=0;
for(int i=0;i<n;i++)
m=m+n-(lower_bound(cat,cat+n,cat[i]+mid)-cat);
if(m>num)
l=mid;
else
r=mid;
}
printf("%d\n",l);
}
return 0;
}