一道被坑了一个下午的题….不得不说方法还是很巧妙的…
题意:有一所奶牛大学要招生,准备招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;
}