程序设计思维与实践 Week4 作业 (3/4/数据班) C - TT 的神秘礼物

题意

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 的中位数

思路

这是一道二分答案的题，ans[] = abs(cat[i] - cat[j])，我们要求ans数组的中位数，首先我们将cat数组由小到大排序之后可以将绝对值去除，我们可以得到
cat[j] - cat[i] = p，移项得到cat[j] = cat[i] + p，p最大值为cat[n-1] - cat[0]
由中位数的性质我们可以知道，如果p要是中位数的话，那么小于等于p的值由 (len+1)/2 个，所以cat[j] <= cat[i] + p , (j > i)，cat[j]也有 (len+1)/2，在此题中为 (n * (n - 1)/2 - 1)/2个。
由于p范围已知[0, cat[n-1] - cat[0] ]，所以我们对p进行二分，对于二分的这个p，对cat中每一个i值，我们再次进行二分，来寻找满足cat[j] <= cat[i] + p的cat[j]的个数，所有的个数加起来就是p再ans数组中的位次，与上面的中位数位次比较，继续对p进行二分，直到结束。

总结

第一次接触二分答案的题，感觉比较新颖，主要是判断如何对答案进行二分。

代码

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

int cat[100005];

int binary(int *cat,int i,int p,int n)
{
	int ans=-1;
	
	int l=i+1,r=n-1;
	while(l<=r)
	{
		int mid=(l+r)>>1;
	
		if(cat[mid]<=cat[i]+p)
		{
			l=mid+1;
			ans=mid;
		}
		else
			r=mid-1;
	}
	return ans-i;
}
int main()
{
	int n;
	while(scanf("%d",&n)!=EOF)
	{
		for(int i=0;i<n;i++)
			scanf("%d",&cat[i]);
		sort(cat,cat+n);
		
		int l=0,r=cat[n-1]-cat[0];
		int midr=(n*(n-1)/2+1)/2;
		while(l<=r)
		{
			int mid=(l+r)>>1;
		
			int rank=0;
			for(int i=0;i<n;i++)
			{
				int temp=binary(cat,i,mid,n);
				if(temp>=0)
					rank+=temp;
				
			}
			
			if(rank>=midr)
				r=mid-1;
			else
				l=mid+1;
					
		}	
		printf("%d\n",r+1);
	}
}