牛客竞赛-NC50439

牛客竞赛-NC50439 每日一题3.25

牛客竞赛-NC50439_第1张图片
思路:

如果没有每个士兵的人数限制,显而易见的我们会直接按照武力值大小排序,但是每个士兵对人数的加以了限制,如果我们还是按武力值大小加人的话,每次加人和删掉人都会导致队伍能容纳的人数的变化,而且还是跳跃的变化(完全有可能忽大忽小),更关键的是我们并不知道人数限制不满足了我们该删去谁——是删掉限制了我们人数的那个人扩充容量呢还是按现在的限制把多的人都删掉呢?

但是想到这里,你应该能发现(也有可能你一开始就发现了),如果我们已知团队人数最多为k,那么我们的选择是可以贪心的——选 si大于等于 k 的武力值最大的 k个人。

那么我们可以考虑枚举这个k(k的取值并不需要连续,按所有的 si取值就可以),但再枚举k之后时间复杂度已经不允许再每次都挨个求武力值和了,这时我们来考虑一下这个武力值的和是否可以在枚举 k 的过程中就维护出来(你可以考虑从大到小枚举 k,或者从小到大枚举 k,看哪一个可以)。然后我们发现,如果我们从大到小枚举 k,k 每减少一次就相当于需要先把 si​ 满条件的人先加进来,然后把多于 k个的武力值最小的几个人删掉,再加入与删除的操作过程中就可以维护所有人的权值和。换句话说,如果我们每次都加入当前还没有进入队伍的 si最大的人,那么只需要在加入他之后删掉超出了这个si的武力值最小的人就可以了。
而插入与删除的操作需要用一个支持加入元素和删除最小值的数据结构来维护,这个数据结构当然是堆(优先队列)啦!
转自:传送门

#pragma GCC optimize(3,"Ofast","inline")  	//G++
#include
#define mem(a,x) memset(a,x,sizeof(a))
#define debug(x) cout << #x << ": " << x << endl;
#define ios ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define fcout cout<
using namespace std;
typedef long long ll;
//======================================
namespace FastIO{
char print_f[105];void read() {}void print() {putchar('\n');}
template <typename T, typename... T2>
inline void read(T &x, T2 &... oth){x = 0;char ch = getchar();ll f = 1;while (!isdigit(ch)){if (ch == '-')f *= -1;ch = getchar();}while (isdigit(ch)){x = x * 10 + ch - 48;ch = getchar();}x *= f;read(oth...);}
template <typename T, typename... T2>
inline void print(T x, T2... oth){ll p3=-1;if(x<0) putchar('-'),x=-x;do{print_f[++p3] = x%10 + 48;}while(x/=10);while(p3>=0) putchar(print_f[p3--]);putchar(' ');print(oth...);}} // namespace FastIO
using FastIO::print;
using FastIO::read;
//======================================
typedef pair<int,int> pii;
const int inf=0x3f3f3f3f;
const int mod=1e9+7;
const int maxn = 1e6+5;
map<int,vector<int>>mp;
int main() {
#ifndef ONLINE_JUDGE
    freopen("H:\\code\\in.in", "r", stdin);
    freopen("H:\\code\\out.out", "w", stdout);
    clock_t c1 = clock();
#endif
//**************************************
    int n;
    read(n);
    for(int i=1,x,y;i<=n;i++){
        read(x,y);
        mp[-y].push_back(x);
    }
    priority_queue<int,vector<int>,greater<int>>q;
    ll ans=0,sum=0;
    for(auto it:mp){
        int u=-it.first;
        for(auto v:it.second){
            q.push(v);
            sum+=v;
        }
        while(q.size()>u) sum-=q.top(),q.pop();
        ans=max(ans,sum);
    }
    print(ans);
//**************************************
    
#ifndef ONLINE_JUDGE
    cerr << "Time:" << clock() - c1 << "ms" << endl;
#endif
    return 0;
}

你可能感兴趣的:(日常训练,牛客补题)