POJ2010 优先队列

一道被坑了一个下午的题….不得不说方法还是很巧妙的…
题意:有一所奶牛大学要招生,准备招N个,共有C个报名,每个有一个分数和它所需要的学费。现在政府要承担它们的学费,但是最多只能提供F的费用。要求你编程,使得满足费用的条件下招得得N头奶牛的分数的中位数最大。如果F费用没办法招满N头牛,那么输出-1,否则输出中位数。

N<19999,是一个奇数
C< 105
F < 2∗109

先将C头奶牛按照分数从小到大进行排序,利用大顶堆计算出每头牛左侧 N/2头牛的最小学费和,以及每头牛右侧 N/2 头牛的最小学费和。然后从右往左遍历 N/2 C−N/2−1头牛,如果满足费用小于F,那么它就是答案。
  其中:大顶堆是用priority_queue,时间复杂度O(ClogC)。遍历的时间为C。
具体实现的时候则非常巧妙了;
另一种描述:这里写图片描述

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<queue>
#include<cstring>
using namespace std;
struct cow{
    int mark,money;
}cows[100005];
bool cmp(cow a,cow b)
{
    return a.mark<b.mark;
}
bool operator <(cow a,cow b)
{
    return a.money<b.money;
}
int main()
{
    int n,c,f,i,j,k;
    cin>>n>>c>>f;
    for(i=1;i<=c;i++)
        scanf("%d%d",&cows[i].mark,&cows[i].money);//一定要scanf不然会tle
    sort(cows+1,cows+1+c,cmp);
    int upper[100005],lower[100005];//upper[i]表示以第i只奶牛为中位数时大于中位数的奶牛的钱财和,lower同理
    priority_queue<cow>que1;int total=0;
    for(i=1;i<=c;i++){
        lower[i]=que1.size()==n/2?total:0x3f3f3f3f;//这里非常巧妙,这里的que对应的情况是前i-1只奶牛在队列中的情况,刚好就是d第i只奶牛做中位数的情况
        que1.push(cows[i]);
        total+=cows[i].money;
        if(que1.size()>n/2){//如果队列过多,就把堆顶也就是钱数最大的奶牛扔掉
            total-=que1.top().money;que1.pop();
        }
    }
    total=0;priority_queue<cow>que2;
    for(i = c; i>=1; i--){
        upper[i]=que2.size()==n/2?total:0x3f3f3f3f;
        que2.push(cows[i]);
        total+=cows[i].money;
        if(que2.size()>n/2){
            total-=que2.top().money;que2.pop();
        }
    }
    int ans=-1;
    for(i=c-1;i;i--){//遍历所有可能
        if(upper[i]+lower[i]+cows[i].money<=f) {
            ans=cows[i].mark;
            break;
        }
    }
    cout<<ans<<endl;

    return 0;

}
  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值