POJ 2010 优先队列

题意

有C头牛，选M头牛，每一头牛的分数是第一个数字，花费金钱是第二个数字。有F金钱。问如何选择可以使分数的中位数最大。

题解

网上的二分代码即便AC了，很多想法都还是有问题的。感觉正解应该是用优先队列，先将M头牛按成绩排序，设置两个数组l和r，l数组代表某头牛左边如果要选足够的牛(m/2头牛)，所需要的最小花费，r同理。利用优先队列进行初始化。然后对于存在l和r的区间进行一遍遍历即可。

代码

#include <iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
#include<cmath>
#include<queue>
#include<string>
#include<set>
#include<map>
#define UP(i,l,h) for(int i=l;i<h;i++)
#define DOWN(i,h,l) for(int i=h-1;i>=l;i--)
#define W(a) while(a)
#define INF 0x3f3f3f3f
#define LL long long

using namespace std;

struct Node{
    LL score,money;
    bool operator < (const Node b) const{
        return money<b.money;
    }
};

bool cmp(Node a,Node b){
    return a.score<b.score;
}
Node nodes[100010];
LL l[100010],r[100010];

int main()
{
    int n,c,f;
    W(~scanf("%d%d%d",&n,&c,&f)){
        memset(l,0,sizeof(l));
        memset(r,0,sizeof(r));
        UP(i,0,c){
            scanf("%I64d%I64d",&nodes[i].score,&nodes[i].money);
        }
        sort(nodes,nodes+c,cmp);
        int num=n/2;
        priority_queue<Node> que;
        LL sum=0;
        UP(i,0,num){
            que.push(nodes[i]);
            sum+=nodes[i].money;
        }
        UP(i,num,c){
            l[i]=sum;
            sum+=nodes[i].money;
            que.push(nodes[i]);
            sum-=que.top().money;
            que.pop();
        }
        W(!que.empty()){
            que.pop();
        }
        sum=0;
        UP(i,c-num,c){
            que.push(nodes[i]);
            sum+=nodes[i].money;
        }
        DOWN(i,c-num,0){
            r[i]=sum;
            sum+=nodes[i].money;
            que.push(nodes[i]);
            sum-=que.top().money;
            que.pop();
        }

        int ans=-1;
        UP(i,num,c-num){
            if((l[i]+r[i]+nodes[i].money)<=f&&nodes[i].score>ans)
                ans=nodes[i].score;
        }
        printf("%d\n",ans);
    }
    return 0;
}

/*
1 5 10
30 25
50 21
20 20
5 18
35 10

3 5 70
30 25
50 21
20 20
5 18
35 30
*/