poj 2010 Moo University - Financial Aid 优先队列

题目

题目链接：http://poj.org/problem?id=2010

题目来源：《挑战》练习题

简要题意：有CC头牛成绩aia_i，代价bib_i，钱FF。

\qquad \quad \ \ \ \ 求一秩为NN的牛的子集，代价和不超过FF且牛成绩的中位数最大。

数据范围：1⩽N⩽19999;N⩽C⩽1051 \leqslant N \leqslant 19999;\quad N \leqslant C \leqslant 10^5\quad 代价不爆INT

题解

首先需要排序，然后扫两遍处理出从前后面到某位置取NN头的最小代价。

这实际上就是维护一个堆，对于和可以另外保存一个值来搞。

最后再扫一遍判断一下就行了。

实现

堆的功能可以直接用优先队列来代替。

做的时候把一个i写成了l，然后很自信地WA了若干发，今天做题不顺啊。

觉得还是要吸取教训啊，要好好测试，好好思考才行，尤其是贪心，想法题。

要注意N=1N=1的情况，这个时候队列可能为空，要判断。

另一种做法是去二分答案，前后再排序搞。

代码

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <cstring>
#include <stack>
#include <queue>
#include <string>
#include <vector>
#include <set>
#include <map>

#define pb push_back
#define mp make_pair
#define all(x) (x).begin(),(x).end()
#define sz(x) ((int)(x).size())
#define fi first
#define se second
using namespace std;
typedef long long LL;
typedef vector<int> VI;
typedef pair<int,int> PII;
LL powmod(LL a,LL b, LL MOD) {LL res=1;a%=MOD;for(;b;b>>=1){if(b&1)res=res*a%MOD;a=a*a%MOD;}return res;}
// head
const int N = 1E5+5;

PII a
;
int sumf
;
int sumb
;

int main()
{
int n, m, lim;
scanf("%d%d%d", &n, &m, &lim);
n >>= 1;
int l = n+1, r = m-n;
for (int i = 1; i <= m; i++) {
scanf("%d%d", &a[i].fi, &a[i].se);
}
sort(a+1, a+m+1);

if (n > 0) {
priority_queue<int> q;
for (int i = 1; i < l; i++) {
q.push(a[i].se);
sumf[i] = sumf[i-1]+a[i].se;
}
for (int i = l; i <= r; i++) {
sumf[i] = sumf[i-1];
if (a[i].se >= q.top()) continue;
sumf[i] -= q.top() - a[i].se;
q.pop();
q.push(a[i].se);
}
while (!q.empty()) q.pop();

for (int i = m; i > r; i--) {
q.push(a[i].se);
sumb[i] = sumb[i+1]+a[i].se;
}
for (int i = r; i >= l; i--) {
sumb[i] = sumb[i+1];
if (a[i].se >= q.top()) continue;
sumb[i] -= q.top() - a[i].se;
q.pop();
q.push(a[i].se);
}
}

int ans = -1;
for (int i = l; i <= r; i++) {
if (sumf[i-1]+sumb[i+1]+a[i].se <= lim) {
ans = a[i].fi;
}
}
printf("%d\n", ans);
return 0;
}