第四周作业——TT 的神秘礼物

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 非常想得到那只可爱的猫咪，你能帮帮他吗？

二、输入
多组输入，每次输入一个 N，表示有 N 个数，之后输入一个长度为 N 的序列 cat， cat[i] <= 1e9 , 3 <= n <= 1e5

三、输出
输出新数组 ans 的中位数

四、样例输入输出

Input

4
1 3 2 4
3
1 10 2

Output

1
8

五、解题思路
给定一个数p，通过计算p的名次判断p是否是中位数。
使用二分法，在数据范围的最大值和最小值之间确定p的值。将X排序，计算满足Xj-Xi<=p的二元数组对的个数，移项得Xj<=Xi+p,枚举下标i，使用二分法计算满足条件的下标j的个数。
先根据X中的元素数确定中位数的位置ans。二分查找j的时候，查找到满足条件的最大的j，从而可以确定满足条件的数组的对数。给定一个p，计算满足Xj<=Xi+p的数组对数大于等于ans，则p的取值选择左半区间，否则选择右半区间，从而找出满足数组对数小于ans的最大的p，进而可以确定第一个满足数组对数大于等于中位数的P。

注意：在二分时注意mid的取值是否更新，我在做p的二分查找时，忘记begin=end时程序已经跳出，最后一步未更新mid，在输出时调用的仍是上一步p二分的mid，导致WA。

六、相关代码

#include <algorithm> 
#include <iostream>
#include <stdio.h>
using namespace std;
int n;

int dich2(int *a,int x)  //二分查找小于等于它最大的 
{
	int l=0,r=n-1,ans=-1;
	while(l<=r)
	{
		int mid=(l+r)>>1;
		if(a[mid]<=x)
		{
			ans=mid;
			l=mid+1;
		}
		else r=mid-1;
	}
	return ans;
}

int main()
{
  while(scanf("%d",&n)!=EOF)
  {
	int cat[n];
	for(int i=0;i<n;i++)
	{
		scanf("%d",&cat[i]);
	}
	sort(cat,cat+n);
	
	int ans=(((n*(n-1))/2)+1)/2;
	int begin=0,end=1000000000;
	int mid=0;

	while(begin<=end)
	{
		mid=(begin+end)/2;
		int sum=0;
		for(int j=0;j<n;j++)
		{
			int ssum=0;
			ssum=dich2(cat,mid+cat[j]);  
		//	cout<<mid<<" "<<mid+cat[j]<<" "<<dich2(cat,mid+cat[j])<<" "<<j<<endl;
			if(ssum-j<0) continue;
			sum=sum+ssum-j;
		}
		if(sum>=ans) end=mid-1;
		else begin=mid+1;
	}
	printf("%d\n",end+1);
  }
}