传送门:codeforces 913D
题目大意:
在时间 T内解决 n 道题,每道题都有两个属性 ai和 ti,其中 ti表示做该题需要的时间,ai表示,只有在总解题数 <= ai时才会给结果加一分。问最终的得分最高是多少,需要解决哪几道题?
思路:
这个题有两个限制条件,一个是解题时间,一个是每道题的 ai值。如果是直接枚举所有解题的组合然后挑选最大值,肯定不现实。分析一下可以知道,最优的情况下一定是只做可以得分的题,样例 1 中的就有点多余了。并且每道题对结果的贡献是相同的,都是加一分。所以我们当时间足够的时候,我们尽可能多的多做题;当在 ai满足时,可以用时间更短的题替换时间长的题。为了避免替换后产生 ai不满足题意的情况,我们按照 ai递减的顺序加入。当当前题目的 ai<=做题总数的时候就是最优了。
具体实现:
以结构体存储每道题,按照 ai递减的顺序排序,然后依次加入一个以解题时间为优先值的优先队列。如果有足够的时间做当前题,则加入;如果没有时间做当前题目,则判断是否比队列中时间耗费最大的更优,如果是,则替换。当当前题的 ai <=做题总数的时候就无法再加入新的题目了,替换也没有意义,所以是最优。
注意:
在优先队列使用 Q.top()函数时,要先用 Q.emtpy()判断队列是否为空,否则会出现 runtime error。
代码:
#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
struct node
{
int a,t,pos; //pos为题目编号
} s[2000100];
int cmp(node x,node y)
{ //排序比较函数
if(x.a==y.a) return x.t<y.t;
else return x.a>y.a;
}
bool operator<(node x,node y)
{ //优先队列优先权函数
return x.t<y.t;
}
int main()
{
int i,n,T,sz; //T的含义是剩余的解题时间
node nd;
priority_queue<node> Q;
while(~scanf("%d%d",&n,&T))
{
while(!Q.empty()) Q.pop();
for(i=0;i<n;i++)
{
scanf("%d%d",&s[i].a,&s[i].t);
s[i].pos=i+1;
}
sort(s,s+n,cmp);
for(i=0;i<n;i++)
{
if(Q.size()>=s[i].a) break; //如果不能加入新的题则最优
if(s[i].t<=T)
{ //如果有时间解决当前题则解决
Q.push(s[i]);
T-=s[i].t;
}
else if(!Q.empty()&&s[i].t<Q.top().t)
{ //如果比队首题目更优则替换
T=T+Q.top().t-s[i].t;
Q.pop();
Q.push(s[i]);
}
}
sz=Q.size();
printf("%d\n%d\n",sz,sz);
if(sz>0)
{
nd=Q.top();
printf("%d",nd.pos);
Q.pop();
while(!Q.empty())
{
nd=Q.top();
printf(" %d",nd.pos);
Q.pop();
}
}
printf("\n");
}
return 0;
}