C – TT 的神秘礼物(Week4作业)

一位重度爱猫人士面临好友的挑战,需从给定数组生成新数组,并找出新数组的中位数。通过优化算法,采用二分查找与二分法,巧妙解决了时间复杂度过高的问题。

题目

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;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值