SDU_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

题目分析

这道题所采用的思路是答案二分。本题若数据不是很大，可以用O(n2)的复杂度将所需的差值数列算出并排序，取中位数即可。但是本题数据cat[i] <= 1e9 , 3 <= n <= 1e5，采用O(n2)复杂度会超时。

我们考虑这样几个问题：
若已知p在数列中，怎么判断是不是中位数？求p的名次即可。
怎么求p的名次？计算xj-xi<=p的数量，即遍历i求满足条件的j的数量。即用xi+p对xj二分查找，所求数量为j-i（因为每个j都从i+1开始）。
怎么确定p的范围？对于有序数列cat，p一定在0到cat[n-1]之间。每次对p进行二分查找，直到被bound到一个具体的数字。

思考

在做这道题的时候，我一直在思考一个问题：p是在连续整数上取值，怎么确保他一定等于某个xj-xi？后来我发现，p即使等于了某值，但不在边界处的时候，right还是大于left的，故满足条件但循环并未结束，所以可以确保找到的p就是某个xj-xi。

代码

#define _CRT_SECURE_NO_WARNINGS
#define _ ios::sync_with_stdio(false);cin.tie(0);cout.tie(0)

//#include <bits/stdc++.h>
#include <iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>

//快读
inline int read()
{
	int x = 0, f = 1;
	char ch = getchar();
	while (ch < '0' || ch>'9')
	{
		if (ch == '-')
			f = -1;
		ch = getchar();
	}
	while (ch >= '0' && ch <= '9')
		x = x * 10 + ch - '0', ch = getchar();
	return x * f;
}


using namespace std;

int cat[100009];

int main()
{
	int N;
	while (cin >> N)
	{
		memset(cat, 0, sizeof(cat));
		for (int i = 0; i < N; i++)
		{
			cat[i] = read();
		}
		sort(cat, cat + N);

		int len = N * (N - 1) / 2;//新数组长度
		//int left = cat[1] - cat[0];//N=1时候不存在cat[1]，所以左端用0
		int left_value = 0;
		int right_value = cat[N - 1] - cat[0];
		int mid_index = (len + 1) / 2;

		int result;
		while (left_value <= right_value)
		{//二分找p
			int rank = 0;//计算p的排名累计项
			int p = (left_value + right_value) / 2;

			for (int i = 0; i < N; i++)
			{//定p遍历i算p的rank
				int temp_l = 0, temp_r = N - 1, j = -1;
				while (temp_l <= temp_r)
				{//定p+i算最大j
					int temp_mid = (temp_l + temp_r) >> 1;
					if (cat[temp_mid] <= p + cat[i])
					{
						j = temp_mid;
						temp_l = temp_mid + 1;
					}
					else temp_r = temp_mid - 1;
				}

				if (j != -1)
					rank += (j - i);
			}

			if (rank >= mid_index)
			{
				result = p;
				right_value = p - 1;
			}
			else left_value = p + 1;
		}
		cout << result << endl;
	}

	return 0;
}