POJ 3579 Median

傳送門

題目大意是給你一串數字序列,問你這些在序列中所有數字的差的第M/2小的數是多少。M是差值的總數。

直接二分答案,然後再二分搜就行了

代碼

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#define LL __int64
using namespace std;
const int maxn = 1e5 + 1000;
int a[maxn],n,m,ans;

bool check(int val)
{
    int cnt = 0 ;
    for(int i=0;i<n;i++){
        cnt += (n-(lower_bound(a,a+n,a[i]+val)-a));//這裏是數目前這個數是第幾大
    }
    if(cnt > m) return true;
    else return false;
}


int main()
{
    while(~scanf("%d",&n)){
        m = n*(n-1)/4;
        for(int i=0;i<n;i++){
            scanf("%d",&a[i]);
        }
        sort(a,a+n);
        int l = 0, r = a[n-1]-a[0];
        int ans = 0;
        while(r>=l){
            int mid = (l+r)>>1;
            if(check(mid)){
                l = mid + 1;
                ans = mid;
            }else r = mid - 1;
        }

        printf("%d\n",ans);

    }
    return 0;
}



發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章