求中位数

问题描述

给定一个 N 个数的数组 cat[i]，并用这个数组生成一个新数组 ans[i]。新数组定义为对于任意的 i, j 且 i != j，均有 ans[] = abs(cat[i] - cat[j])，1 <= i < j <= N。试求出这个新数组的中位数，中位数即为排序之后 (len+1)/2 位置对应的数字，’/’ 为下取整。
【输入】多组输入，每次输入一个 N，表示有 N 个数，之后输入一个长度为 N 的序列 cat， cat[i] <= 1e9 , 3 <= n <= 1e5
【输出】输出新数组 ans 的中位数
【样例输入】
4
1 3 2 4
3
1 10 2
【样例输出】
1
8

问题分析

暴力做法：枚举i，j求ans数组，然后取中位数，但是时间复杂度和空间复制度均为O(n2)
二分做法：1、由绝对值的限制可得，ans数组的数值大于等于0，最大为cat数组最大值减最小值，因此可以二分ans数组得一个数p
2、对于第一次得到的数p，判断其是不是中位数，则计算p在数列中的名次
3、如果P从小到大排的名次比中位数小， P比中位数要小；如果P从小到大排的名次比中位数大，P比中位数要大；如果P从小到大排的名次等于中位数，P就是中位数
4、计算p的名次：如果将cat数组从小到大排列可以去绝对值。
那么计算cat【j】− cat【i】 ≤ p的二元组对数即可。
移项可得cat【j】 ≤ cat【i】 + P, i < j
枚举下标i然后计算满足条件的下标j的个数,其中找满足条件的个数时也用到了二分
复杂度为O(log(n)log(n))

代码分析

1、由于测试数据数据量巨大，因此可能会有cin输入超时问题，换成scanf
2、当累计满足不等式的sum小于中位数的位次时，p一定不满足条件，当大于等于中位数的位次时，因为条件中等于的存在，中位数对应的值在数列中可能不止一个，因此得到的sum比对应位次要大，不存在等于位次的情况，则此时满足条件的p为sum>=m中最小的p值
3、findr（）函数中 if(l=n&&a[l-1]<x) ;代表该数组中没有小于被查找数的情况，原先所写a【l】<x是错误情况，因此l==n，最大为a【n-1】，a【n】会导致数组越界

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



int findr(int a[100010],int x,int n,int l1)//算距离找比目标值x小的数据个数 
{
	int l=l1+1,r=n-1,ans=-1;
	while(l<=r)
	{
		int mid=(l+r)>>1;//速度快
		if(a[mid]==x)
		{
			ans=mid;
			l=mid+1;
		}
		else if(a[mid]>x) r=mid-1;  
		else l=mid+1; 
	} 
	if(l==n&&a[l-1]<x) return l-l1-1;//越界 
	else if(l<n&&ans==-1) return l-l1-1;
	else return ans-l1;
	
}

int main()
{
	//std::ios::sync_with_stdio(false);
	int N;
	while(scanf("%d",&N)!=EOF)//超时！！！ 
	{
		int a[100010];
		for(int i=0;i<N;i++)
		{
			scanf("%d",&a[i]);
		}
		sort(a,a+N);
		int cnt=N*(N-1)/2;//新数组个数 
	    int m= (cnt+1)/2;//中位数的位次 
		int l=0;
		int r=a[N-1]-a[0];//用二分法进行中位数值的猜测 
		int sum=0,mid;
		int p=r;
		while(l<=r)
		{
		    sum=0;//每换一个中位值就重新计算 
			mid=(l+r)>>1;
			for(int i=0;i<N;i++)
			{
				sum+=findr(a,a[i]+mid,N,i);
			
			} 
			if(sum<m) l=mid+1;
			else if(sum>=m) 
			{
				r=mid-1;
				p=min(p,mid);
			
			}
		} 
		cout<<p<<endl;
	}
	
}