G.王梦涵学长的迷弟
题目描述:
王梦涵学长,有着 n 个迷弟。每个迷弟有着自己的力量值。
为了让迷弟们和自己一样强大,王梦涵学长学习了一种增幅魔法,每次使用魔法可以让一位迷弟的力量值加 1 。
例如,王梦涵学长有3个迷弟,力量分别为 [1,2,3],王梦涵学长使用一次魔法使得1号迷弟的力量加1,变成 [2,2,3] 。
但是魔法不能无限使用,王梦涵学长最多只能使用 k 次魔法。
王梦涵学长十分关照力量值比较低的迷弟。所以他想在使用不超过 k 次魔法后,使得全部迷弟中力量值最低的迷弟,力量值最大。
请你帮帮王梦涵学长计算在上方的原则下,力量值最低的迷弟的力量值。
输入描述:
第一行两个数整数 n (1 <= n <= 2 × 105),k (1 <= k <= 2 × 105),表示迷弟的人数以及使用魔法的最大次数。
接下来一行 n 个数 ai (1 <= ai <= 2 × 105),表示第 i 个迷弟的力量值。
输出描述:
输出一个数,表示在使用不超过 k 次魔法后,迷弟中,力量值最低的迷弟的最大力量值。
示例:
输入:
5 3
1 1 3 3 4
输出:
2
说明:
一种方法为从 [1, 1, 3, 3, 4] 加到 [3, 2, 3, 3, 4],此时最小力量值为 2 。
思路:每次找出队列中的最小元素,加一后有序地放回队列,最后输出最小的元素。正好可以使用 STL 里的优先队列来对元素每次加一后重新排列。
这里插入下优先队列的用法:
定义:
priority_queue<Type, Container, Functional>
Type 就是数据类型.
Container 就是容器类型(Container必须是用数组实现的容器,比如vector,deque等等,但不能用 list。STL里面默认用的是vector.
Functional 就是比较的方式,当需要用自定义的数据类型时才需要传入这三个参数,使用基本数据类型时,只需要传入数据类型,默认是大顶堆。greater< int > 为升序整型,less< int >为降序整型。
所以我们用升序的优先队列来排序,需要先申请一个优先队列,名为 s :
priority_queue<int, vector<int>, greater<int> > s;
AC代码如下:
#include <bits/stdc++.h>
using namespace std;
int n,k;
int main()
{
cin >> n >> k;
//升序的优先队列
priority_queue<int, vector<int>, greater<int> >s;
for (int i = 0; i < n; i++){
int x;
cin >> x;
//放入队列中
s.push(x);
}
while (k--){
//取出队首元素加一后再放入队列中
int m = s.top();
s.pop();
s.push(++m);
}
//第一个元素最小
cout << s.top() << endl;
return 0;
}