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

Sample Input

4
1 3 2 4
3
1 10 2

Sample Output

1
8

思路

ans[]=abs(cat[i]-cat[j])当把cat数组进行排序后，绝对值便可去掉，即ans[]=cat[i]-cat[j],若ans中的中位数，即使找到符合cat[i]-cat[j]<=mid的数量为ans数组总数量的一半即（len+1）/2，向下取整。为了减少复杂度，采用两次二分加一次for循环，最先一次的二分是在cat[i]-cat[j]的最小值和最大值中进行二分，然后，此次二分的条件为返回的cat[i]-cat[j]<=mid的数量（写成一个函数），在此函数中，对j进行for循环遍历，对i进行二分求解，二分条件为是否满足cat[i]-cat[j]<=mid。对于每个j值，都要将符合条件的数量进行存储，最终将所有的数量和返回。

代码

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
int f(int, int*, int);
int main()
{
	int n;
	while (scanf("%d",&n)!=EOF)
	{
		int* a = new int[n];
		for (int i = 0; i < n; i++)
			scanf("%d", &a[i]);
		sort(a, a + n);
		int low = 1, high = a[n - 1] - a[0],ans=0;
		while (low <= high)
		{
			int mid = (low + high) /2;
			if (f(mid, a, n) >= (n * (n - 1) / 2 + 1) / 2)
			{
				ans = mid;
				high = mid-1;
			}
			else
			{
				//ans = mid;
				low = mid + 1;
			}
		}
		cout << ans << endl;
	}	
	return 0;
}
int f(int x, int a[], int n)
{
	int ans = 0;
	for (int i = 0; i < n; i++)
	{
		int low = i + 1, high = n-1;
		while (low <= high)
		{
			int mid = (high + low)/2;
			if (a[mid] - a[i] <= x)
				low = mid + 1;
			else
				high = mid - 1;
		}
		ans += high- i;
	}
	return ans;
}