优先队列 POJ2010

题意：

题目链接

　　有一所奶牛大学要招生，准备招N个，共有C个报名，每个有一个分数和它所需要的学费。现在政府要承担它们的学费，但是最多只能提供F的费用。要求你编程，使得满足费用的条件下招得得N头奶牛的分数的中位数最大。如果F费用没办法招满N头牛，那么输出-1，否则输出中位数。

• N<19999,是一个奇数
• C< $10^5$
• F < $2 * 10^9$

解法：

　　先将C头奶牛按照分数从小到大进行排序，利用大顶堆计算出每头牛左侧 $N/2$头牛的最小学费和，以及每头牛右侧 $N/2$ 头牛的最小学费和。然后从右往左遍历 $N/2 ~ C-N/2-1$头牛，如果满足费用小于F，那么它就是答案。
　　其中：大顶堆是用priority_queue，时间复杂度$O(ClogC)$。遍历的时间为 C <script type="math/tex" id="MathJax-Element-275">C</script>。

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <queue>

using namespace std;

long long Lower[100000+5],upper[100000+5];
struct Cow{
   int score;
   int aid;
}cow[100000+5];

const long long INF = 100000000000;

bool cmp(const Cow &c1, const Cow &c2){
    return c1.score<c2.score;
}

int main()
{
    int N,C,F;
    scanf("%d%d%d",&N,&C,&F);
    memset(cow,0,sizeof(cow));
    memset(Lower,0,sizeof(Lower));
    memset(upper,0,sizeof(upper));
    for(int i = 0; i<C; i++)
        scanf("%d%d",&cow[i].score,&cow[i].aid);
    sort(cow,cow+C,cmp);
    priority_queue<int> q1;
    int total = 0;
    for(int i = 0; i<C; i++)
    {
        Lower[i] = q1.size()==N/2? total:INF;
        q1.push(cow[i].aid);
        total += cow[i].aid;
        if(q1.size()>N/2)
        {
            total -= q1.top();
            q1.pop();
        }
    }
    total = 0;
    priority_queue<int> q2;
    for(int i = C-1; i>=0; i--)
    {
        upper[i] = q2.size()==N/2? total:INF;
        q2.push(cow[i].aid);
        total += cow[i].aid;
        if(q2.size()>N/2)
        {
            total -= q2.top();
            q2.pop();
        }
    }
    int ans = -1;
    for(int i = C-N/2-1; i>=N/2; i--)
    {
        if(Lower[i]+cow[i].aid+upper[i] <= F){
            ans = i;
            break;
        }
    }
    if(ans == -1) printf("-1\n");
    else  printf("%d\n",cow[ans].score);
    return 0;
}