二分答案策略--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

解题思路以及关键代码

（二分答案加二分查找数）
二分答案 首先确定答案区间 本题就是排序后最后一个减去第一
主要关键就是每一步选好二分答案p_mid之后，如何判断有多少满足不等式的个数
这里（find1是找a[n]中第一个>x的位置 位置从1开始 find2是找a[n]中第一个==x的位置 位置从1开始 ）仍然可以利用二分
一共就有4种情况 find1找到find2找到 find1找到find2找不到 等
下面是我找样例推得规律。

int p_mid = (p_min + p_max) / 2;  //二分答案 
int count = 0;
for (int i = 1; i < n; i++) //Xj <= Xi + P 其中 j > i  count为满足该不等式的个数 
{	
	int tmp;
	int a1 = find1(a[i]+p_mid, n);
	tmp = a1 - i -1;
	if(a1!=-1) 
		count += tmp;
	else if(a1!=-1&&find2(a[i]+p_mid, n)!=-1) 
		count =count + find2(a[i]+p_mid, n) - i;//没有比 a[i]+p_mid >=的数   等于之前的都满足 
	else
		count =count + n-i; //没有比 a[i]+p_mid > 的数 即所有的 j>i 都满足 
		

注意点：输入数据有10万个，一定要用scanf 而且本题卡常数  稍不留意就会TLE
实测，原本cin过不了，用了scanf速度提高了2倍就过了

全部代码

#include<iostream>
#include<algorithm>
#include<cstdio>
using namespace std;
int a[100005];
int find1(int x,int n) //找a[n]中第一个>x的位置  位置从1开始
{
	int l=1,r=n,ans=-1;
	while(l<=r)
	{
		int mid=(l+r)/2;
		if(a[mid]>x)
		{
			ans = mid;
			r = mid-1;
		}
		else l=mid+1;
	}
	return ans;
}
int find2(int x,int n) //找a[n]中第一个==x的位置 不存在返回-1 位置从1开始
{
	int l=1,r=n,ans=-1;
	while(l<=r)
	{
		int mid=(l+r)/2;
		if(a[mid]==x)
		{
			ans = mid;
			r = mid-1;
		}
		else if(a[mid] > x) r= mid-1;
		else l=mid+1;
	}
	return ans;
}
int main() 
{
	int n;
	while (cin >> n) 
	{
		for (int i = 1; i <= n; i++)
			scanf("%d",&a[i]);
		sort(a+1, a + n + 1);
		int len = (1 + n * (n - 1)) / 2; 	//新数组的长度 
		int p_max = a[n] - a[1];
		int p_min = 0;  //p_max为最大可能的答案  
		int ans = 0;
		while (p_min <= p_max)  
		{
			int p_mid = (p_min + p_max) / 2;  //二分答案 
			int count = 0;
			for (int i = 1; i < n; i++) //Xj <= Xi + P 其中 j > i  count为满足该不等式的个数 
			{	
				int tmp;
				int a1 = find1(a[i]+p_mid, n);
				tmp = a1 - i -1;
				if(a1!=-1) 
					count += tmp;
				else if(a1!=-1&&find2(a[i]+p_mid, n)!=-1) 
					count =count + find2(a[i]+p_mid, n) - i;//没有比 a[i]+p_mid >=的数   等于之前的都满足 
				else
					count =count + n-i; //没有比 a[i]+p_mid > 的数 即所有的 j>i 都满足 
			}
			if (count >= (len+1)/2)  //中位数即为排序之后 (len+1)/2 位置对应的数字
			{
				ans = p_mid;
				p_max = p_mid - 1;
			}
			else 
				p_min = p_mid + 1;			
		}
		cout << ans << endl;
	}
	return 0;
}