大致题意
给 n 个士兵,每个士兵有一个战力 vi ,有一个限制 si (表示如果选择 i ,那么总共选的士兵数目不能超过 si),求一种士兵的组合方式,使得他们的战力和最大,且满足每个士兵的限制要求。输出最大的战力值和。
思路
一开始各种方法尝试了半天,最后也就是个思维题。
首先按照 si 从小到大排序,然后从 n 到 1 将 vi 插入最小堆,如果当前堆中元素个数 siz > si 那么不断弹出堆中的最小值 ,最终剩下的元素和 更新 dp[siz]。
这样的目的是,后面的元素一定可以用在 si 比较小的集合里,但是反过来不行。整体的思路就是,求当前可用元素的前 si大的值,但是用最大堆是不行的,每次弹出前si大会超时,所以不如保留最大元素,退出较小元素。
说的也不是很清楚,看代码体会吧。
代码
#include<bits/stdc++.h>
using namespace std;
#define maxn 1000005
#define maxm 1000006
#define ll long long int
#define INF 0x3f3f3f3f
#define inc(i,l,r) for(int i=l;i<=r;i++)
#define dec(i,r,l) for(int i=r;i>=l;i--)
#define mem(a) memset(a,0,sizeof(a))
#define sqr(x) (x*x)
#define inf (ll)2e18+1
int read(){
int x=0,f=1;char ch=getchar();
while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
while(isdigit(ch))x=x*10+ch-'0',ch=getchar();
return f*x;
}
int n;
struct node{int v,s;
bool operator < (const node &rid)const{
return s < rid.s;
}
}a[maxn];
priority_queue<int>q;
ll dp[maxn];
int main()
{
n=read();
inc(i,1,n){a[i].v=read();a[i].s=read();}
sort(a+1,a+n+1);
ll sum=0;
dec(i,n,1){
sum+=a[i].v;
q.push(-a[i].v);
while(!q.empty()&&q.size()>a[i].s){sum+=q.top();q.pop();}
dp[q.size()]=max(dp[q.size()],sum);
}
ll ans=0;
inc(i,1,n)ans=max(ans,dp[i]);
printf("%lld\n",ans);
return 0;
}