poj3579 Median （二分-查找第K大的值）

题目链接：https://vjudge.net/problem/POJ-3579

题目大意：给你一组数据，让你从数据中选出a,b两个数，然后求出他俩差的绝对值，让你把所有两数之差求解出来，然后求出这个新的数列的中位数

题目思路：如果求出所有的差值的绝对值再排序这样是绝对不行的，这样的话复杂度会达到O(n^2)肯定会超时。那我们只能选择其他的方法。

                    这里设原数据都存在a数组，一共有n个数

                    1.求出比中位数大的数有n*(n-1)/2/2个

                    2.对a进行升序排序

                    3.二分中位数

                    4.遍历a数组，sum+大于a[i]+mid的所有数

学到的东西：对于时间复杂度较高的问题，可以采用二分的方法降低算法的时间复杂度

代码：

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
const int maxn = 1e5+10;
const int inf = 1e9+7;
int a[maxn];
int n,m;

bool judge(int mid)
{
    int sum=0;
    for(int i=0;i<n;i++){
        sum+=(n-1)-(lower_bound(a+i+1,a+n,a[i]+mid)-a)+1;
    }
    if(sum>m) return true;

    return false;
}
int main()
{
    while(~scanf("%d",&n)){
        for(int i=0;i<n;i++) scanf("%d",&a[i]);
        sort(a,a+n);
        int c=n*(n-1)/2;
        m=c/2;//大于中位数差值的数量

        int l=0,r=inf,mid;
        int ans;
        while(l<=r){
            mid = (l+r)>>1;
            if(judge(mid)) ans=mid,l=mid+1;
            else r=mid-1;
        }
        printf("%d\n",ans);
    }
}