[牛客每日一题] (贪心+小顶堆)NC50439 tokitsukaze

tokitsukaze and Soldier (nowcoder.com)https://ac.nowcoder.com/acm/problem/50439

题目描述

在一个游戏中，tokitsukaze需要在n个士兵中选出一些士兵组成一个团去打副本。
第i个士兵的战力为v[i]，团的战力是团内所有士兵的战力之和。
但是这些士兵有特殊的要求：如果选了第i个士兵，这个士兵希望团的人数不超过s[i]。(如果不选第i个士兵，就没有这个限制。)
tokitsukaze想知道，团的战力最大为多少。

输入描述:

第一行包含一个正整数n(1≤n≤10^5)。
接下来n行，每行包括2个正整数v,s(1≤v≤10^9,1≤s≤n)。

输出描述:

输出一个正整数，表示团的最大战力。

示例1

输入

2
1 2
2 2

输出

3

示例2

输入

3
1 3
2 3
100 1

输出

100

思路:

若是士兵没那么多讲究, 那便可以按照战力排序, 由大到小取所需人数或者求最大战力和

但由于他们比较挑剔, 每次我们取容忍度不小于s[i]的所有满足条件的士兵, 若是他们的人数超过了s[i], 就去除战力最小的

即在满足一部分人的需求的情况下, 去除多余的"弱者"

比较好维护且每次可以对最值进行计算/删除的数据结构, 那么就可以选择堆

法一:手写小根堆

#include<iostream>
#include<queue>
#include<algorithm>
using namespace std;

const int N = 1e5+10;
int treapSize;
int tp[N];
struct soldier{
    int v, s;
    bool operator < (soldier& y){
        return s > y.s;
    }
}so[N];

void up(int u)
{
    while(u > 1 && tp[u >> 1] > tp[u])
    {
        swap(tp[u >> 1], tp[u]);
        u >>= 1;
    }
}
void down(int u)
{
    while((u<<1) <= treapSize)
    {
        int t = u<<1;
        if(t+1 <= treapSize && tp[t+1] < tp[t]) t ++;
        if(tp[t] < tp[u]){
            swap(tp[t], tp[u]);
            u = t;
        }else break;
    }
}
void add(int x)
{
    tp[++treapSize] = x;
    up(treapSize);
}
void pop()
{
    swap(tp[1], tp[treapSize--]);
    down(1);
}
int main()
{
    int n; 
    scanf("%d", &n);
    for(int i = 0; i < n; i ++) scanf("%d%d", &so[i].v, &so[i].s);
    sort(so, so+n);
    
    long long res = 0, t = 0;
    for(int i = 0; i < n; i ++)
    {
        t += so[i].v;
        add(so[i].v);
        
        while(treapSize > so[i].s)
        {
            t -= tp[1];
            pop();
        }
        
        res = max(res, t);
    }
    printf("%lld", res);
    return 0;
}

法二:优先队列

priority_queue<int, vector<int>, greater<int> >

#include<iostream>
#include<queue>
#include<algorithm>
using namespace std;

const int N = 1e5+10;
struct solider{
	int v;
	int s;
	bool operator < (const solider& y){
		return s > y.s;
	}
}so[N]; 

int main()
{
	priority_queue<int, vector<int>, greater<int> > q;//小根堆 
	
	int n;
	cin >> n;
	
	for(int i = 0; i < n ; i ++)
		scanf("%d%d", &so[i].v, &so[i].s);
	
	sort(so, so+n);
	long long res = 0, tmp = 0;
	
	for(int i = 0; i < n; i ++)
	{
		tmp += so[i].v;
		q.push(so[i].v);
		
		while(q.size() > so[i].s)
		{
			tmp -= q.top();
			q.pop();
		}
		
		res = max(res, tmp);
	}
	cout << res;
	return 0;
}