其他分享
首页 > 其他分享> > tokitsukaze and Soldier

tokitsukaze and Soldier

作者:互联网

tokitsukaze and Soldier
链接: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

该题是一个小顶堆的应用问题。
我们只需要对可能发生的情况进行枚举比较最终便可得到最大值。
首先,我们先建立一个包含(long long)v[i]和s[i]的结构体b[i],然后按照s[i]的大小对结构体进行从大到小排序,排完序后用一个循环对v[i]求和,求和过程中当堆中元素个数大于其对应的s[i]时把堆顶弹出直到堆中元素个数小于等于对应的s[i],接下来取n1和n2中较大的值赋与n2,再进行循环,循环完后的n2便是最大值。

#include<bits/stdc++.h>
#include<queue>
using namespace std;
priority_queue<long long, vector<long long>, greater<long long> > xiaodui;//优先队列实现小堆
struct junren{
	long long v;
	int s;
};
int cmp(const void *a,const void *b){
	struct junren *c=(struct junren*)a;
	struct junren *d=(struct junren*)b;
	if(c->s!=d->s)
	return d->s-c->s; 
}
int main(){
	int a,i;
	scanf("%d",&a);
	struct junren b[a]; 
	for(i=0;i<a;i++){
		scanf("%lld %d",&b[i].v,&b[i].s);
	}
	qsort(b,a,sizeof(b[0]),cmp);//按结构体b[i]中的s[i]从大小进行排序
	long long n1=0,n2=0;
	for(i=0;i<a;i++){
		n1+=b[i].v;
		xiaodui.push(b[i].v);
		while(xiaodui.size()>b[i].s){
			n1-=xiaodui.top();
			xiaodui.pop();
		}
		n2=max(n1,n2);
	}
	printf("%lld",n2);
}

标签:junren,struct,Soldier,long,tokitsukaze,xiaodui,n1,n2
来源: https://blog.csdn.net/weixin_52855047/article/details/116803698