题解 HDU - 6698 Coins (想法+优先队列)
题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=6698
题意
给你一个 n n n对物品,每对物品可以花费1元得到 a i ai ai的价值,或者花费2元得到 a i + b i ai+bi ai+bi的价值。
求输出从 1 1 1到 2 n 2n 2n 元时每次获取能的最高价值。
数据范围: 1 ≤ n ≤ 1 0 5 ; 1 ≤ a i , b i ≤ 1 0 4 1 \le n \le 10^5;1 \le ai,bi \le 10^4 1≤n≤105;1≤ai,bi≤104
思路
这题显然不能够DP,因为其需要的空间和时间都太大了,
那么一定时递推,从前面已经算出了的状态向后转移。
经过发现当我为 n n n时,我从 n − 2 n-2 n−2的状态转移过来,一定是最优的
那么就很简单了,我们动态维护两类优先队列,每次贪心的取就好了。
代码
#include <bits/stdc++.h>
using namespace std;
#define rep(i,j,k) for(int i = (int)j;i <= (int)k;i ++)
#define debug(x) cerr<<#x<<":"<<x<<endl
#define pb push_back
typedef long long ll;
typedef pair<int,int> pi;
const int MAXN = (int)2e6+7;
int a[MAXN],b[MAXN],c[MAXN];
priority_queue<pi> se[2],sb[2];
int ans[MAXN];
int mark[2][MAXN];
int main()
{
int T;
scanf("%d",&T);
while (T --) {
rep(i,0,1) {
while (!se[i].empty()) se[i].pop();
while (!sb[i].empty()) sb[i].pop();
}
int N;
scanf("%d",&N);
rep(i,0,5*N) mark[0][i] = mark[1][i] = ans[i] = a[i] = b[i] = 0;
rep(i,1,N) {
scanf("%d %d",&a[i],&b[i]);
c[i] = a[i]+b[i];
se[0].push(make_pair(a[i],i));
se[1].push(make_pair(a[i],i));
sb[0].push(make_pair(c[i],i));
sb[1].push(make_pair(c[i],i));
}
int f = 0,id,now1,now2;
pi tmp;
//1
tmp = se[1].top();
ans[1] = tmp.first;
mark[1][tmp.second] = 1;
se[1].push(make_pair(b[tmp.second],tmp.second+N));
rep(i,2,2*N) {
now1 = 0,now2 = 0;
pi s1,s2,s3;
while (!se[f].empty() && mark[f][se[f].top().second] == 1) se[f].pop();
while (!sb[f].empty() && mark[f][sb[f].top().second] == 1) sb[f].pop();
if (!se[f].empty()) {
s1 = se[f].top();se[f].pop();
while (!se[f].empty() && mark[f][se[f].top().second] == 1) se[f].pop();
if (se[f].empty()) {
se[f].push(s1);
}else {
s2 = se[f].top();se[f].pop();
now1 = (s1.first+s2.first);
}
}
if (!sb[f].empty()) {
s3 = sb[f].top();sb[f].pop();
now2 = s3.first;
}
if (now1 >= now2) {
ans[i] = ans[i-2] + now1;
int id1 = s1.second;
int id2 = s2.second;
mark[f][id1] = 1;
mark[f][id2] = 1;
se[f].push(make_pair(b[id1],id1+N));
se[f].push(make_pair(b[id2],id2+N));
if (now2) {
sb[f].push(s3);
}
}else {
ans[i] = ans[i-2] + now2;
mark[f][s3.second] = 1;
if (now1) {
se[f].push(s1);
se[f].push(s2);
}
}
f = f^1;
}
rep(i,1,2*N-1) printf("%d ",ans[i]);
printf("%d\n",ans[2*N]);
}
}