程序设计Week4作业——C-TT的神秘礼物

题目描述

给定一个 N 个数的数组 cat[i]，并用这个数组生成一个新数组 ans[i]。新数组定义为对于任意的 i, j 且 i != j，均有 ans[] = abs(cat[i] - cat[j])，1 <= i < j <= N。试求出这个新数组的中位数，中位数即为排序之后 (len+1)/2 位置对应的数字，’/’ 为下取整。

Input

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

Output

输出新数组 ans 的中位数

解题思路

暴力必然不可取，于是我们考虑怎么样优化；
题目要求找到一个未知数列的中位数，乍一看无从下手，但仔细一想，我们对这个数列并非完全未知，我们根据已知数列cat[]，可以得到未知数列ans[]的个数，最大值和最小值，显然未知数列ans[] 的中位数p，它的大小必然介于最大值与最小值中间，并且p的位次必然等于个数/2，我们只需要修改最大值和最小值的值（也就是p的取值范围），来修改p的值，然后通过比较p的位次来不断逼近真正的中位数，由此我们考虑使用二分策略：

• 首先将数组cat[]（代码中为a[]）排序，这样保证可以去掉绝对值符号，再计算出真正中位数下标index，以及预估中位数p的取值区间l，r；
• 在外层对p的值进行二分，因为p是我们猜测的中位数，当p的位次小于index时，说明p的值取小了，p取值区间的左端点l调整为p+1，否则，说明p的值大于等于真正的中位数，调整p取值区间的右端点，直到区间不存在（l<=r）；
• 在内层对p的位次进行二分查找，因为ans[]中的每个数都来自于cat[]中的cat[j]-cat[i]，因此我们只需利用二分查找，计算出满足cat[j]-cat[i]<=p的对数，就能确定当前p在ans[]中的位次；

由此，利用两层二分，就能找到真正中位数p，输出即可。

实现代码

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

int a[100010];
int n;

int P_count(int p)
{//计算满足aj-ai<=p的对数 
	int sum=0;
	for(int i=0;i<n-1;++i)
	{
		int l=i+1,r=n-1,k=0;
		while(l<=r)
		{//二分查找满足aj-ai<=p的最大下标j 
			int j=(l+r)/2;
			if(a[j]<=a[i]+p)
			{//j太小 
				k=j;
				l=j+1;
			}
			else
				r=j-1;
		}
		if(k!=0)
			sum+=(k-i);
	}
	return sum;
}

int main()
{
	while(scanf("%d",&n)!=EOF)
	{
		for(int i=0;i<n;++i)
			scanf("%d",&a[i]);
		sort(a,a+n);
		//差数列中中位数的下标（从1开始） 
		int index=(n*(n-1)/2+1)/2;
		//差数列中中位数的取值范围[l,r] 
		int l=0,r=a[n-1]-a[0];
		//估测的中位数p 
		int p,ans;
		while(l<=r)
		{
			p=(r+l)/2;
			int sum=P_count(p);
			if(sum<index)
			{//p小于真正的中位数 
				l=p+1;
			}
			else
			{//p大于等于真正的中位数 
				r=p-1;
				ans=p;
			}
		}
		printf("%d\n",ans);
	}
	return 0;
}

总结

本题是一道二分的题目，但二分的条件给出的并不明显，需要仔细分析题意，找到中位数的值与位次的关系，再利用两层的二分来求解，而从这种隐式的条件中找关系对我而言是比较困难的，需要加强练习。此外，出错的点在于，当计算p的位次时，没有考虑到某个固定i时，所有的cat[j]-cat[i]<=p都不成立的情况，导致计算位次时出错，对应到代码中，就是在计算sum时少写了if(k!=0)的判断语句，导致所有情况都进行了一次sum+=(k-i)，而每次更新i时，k都会重置为0，没有满足条件的情况出现，k依然为0，sum的值会被错误修改，导致最终结果错误。需要在之后的题目多加注意。