POJ 3579 Median 二分

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/LMengi000/article/details/83626502

 POJ  3579  Median 二分

题意:给出N个数,对于存有每两个数的差值的序列求中位数,如果这个序列有偶数个元素,就取中间偏小的作为中位数。

根据高中的排列组合知识C\binom{2}{n}等于n*(n-1)/2,题目让我们求的是中位数,所以还要除于2;

对a进行从小到大排序;

这样从大到小开始找就可以咯!

upper_bound( begin,end,num):从数组的begin位置到end-1位置二分查找第一个大于num的数字,找到返回该数字的地址,不存在则返回end。通过返回的地址减去起始地址begin,得到找到数字在数组中的下标。

#include<stdlib.h>
#include<iostream>
#include<algorithm>
#include<stdio.h>
#include<string.h>
#include<vector>
#include<math.h>
using namespace std;int n;
const int maxn = 100005;
int a[maxn];
long long m=0;
int check(int mid)
{
    long long cnt=0;
    for(int i=0;i<n;i++)
    {
        int p=upper_bound(a+i,a+n,a[i]+mid)-a; //因为排过序 ,所以 x+a[i]就是 差值+a[i] ,可以找到有几个差值在x以下
        cnt+=(p-i-1);
    }
    if(cnt>=m) return 1;
    else return 0;
}
int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        m=(long long)(n*(n-1))/2;
        m=(m+1)/2;
        memset(a,0,sizeof(a));
        for(int i=0;i<n;i++)
            scanf("%d",&a[i]);
        sort(a,a+n);
        int l=0,r=a[n-1]-a[0]+1,res=0;
        while(l<=r)
        {
            int mid=l+(r-l)/2;
            if(check(mid)) {
                res=mid;
                r=mid-1;
            }
            else l=mid+1;
        }
        printf("%d\n",res);
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/LMengi000/article/details/83626502