牛客练习赛50 C tokitsukaze and Soldier （思维 + 堆）

大致题意

给 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;
}