题目来源:牛客网
给一个数组,一共有n个数。
你能进行最多k次操作。每次操作可以进行以下步骤:
- 选择数组中的一个偶数 a,将其变成a/2。
现在你进行不超过 k 次操作后,让数组中所有数之和尽可能小。请输出这个最小的和。
示例:
代码:
#include<bits/stdc++.h>
#include<iostream>
using namespace std;
int main()
{
priority_queue<long long> q;//优先级队列
int n, k;
long long ans;
cin >> n >> k;
for (int i = 0; i < n; i++)
{
long long x;
cin >> x;
if (x % 2 == 0)
{
//为偶数
q.push(x);
}
else {
ans += x;
}
}
while (k)
{
if (!q.size())//为空
{
break;
}
long long max = q.top();//队列中最大的元素
q.pop();
if (max % 2 == 0)
{
max /= 2;
q.push(max);
k--;
}
else {
ans += max;
}
}
while (q.size())//如果还有元素
{
ans += q.top();
q.pop();
}
cout << ans;
}
思路:
使用优先级队列可以自动排序的能力,可以确保最大的元素一直在队列顶部,从而轻松解决问题
不懂优先级队列的可以看我之前的文章
C++-----stack和queue_c++ queue和stack-CSDN博客