【牛客网-每日一题3.25】tokitsukaze and Soldier

【题目链接】：传送门

【讲解】：传送门

【题解】

　　1、如果我们确定团队的大小Size，我们可以选择Size名士兵（其s>=Size）

　　2、根据（1）对士兵进行s从大到小排序

　　3、我们要做的是：控制团的大小S，同时确保val最大化，

　　　　所以我们对排序完的士兵依次添加，当s<Size时，依次弹出最小值。利用最小堆，维护区间中的最值问题。

#include<queue>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

typedef long long ll ;
const int N = 1e6 + 10 ; 
typedef struct Node{
    int s,v;
    //s size , v value
}Node;
Node a[N] ;

bool cmp( Node u , Node v ){
    return u.s > v.s ;
}

int main()
{
    int n ; 
    //输入后排序,以s从小到大
    scanf("%d",&n);
    for( int i = 1 ; i <= n ; i++ ){
        scanf("%d%d",&a[i].v,&a[i].s);
    }
    sort( a + 1 , a + 1 + n , cmp );
    
    int Sz = 0 ;
    priority_queue< int , vector<int> , greater<int> > Q ;
    ll ans = 0 , sum = 0 ;
    for( int i = 1 ; i <= n ; i++ ){
        Q.push( a[i].v );
        sum += a[i].v ;
        Sz ++ ;
        while( Sz > a[i].s ){
            //printf(" %d , %lld
",i,sum) ;
            sum -= (int) Q.top() ;
            Q.pop();
            Sz -- ;
        }
        ans = max( ans , sum );
    }
    printf("%lld
",ans);
    return 0 ;
}