问题描述:
给你一个正整数数组 nums
。每一次操作中,你可以从 nums
中选择 任意 一个数并将它减小到 恰好 一半。(注意,在后续操作中你可以对减半过的数继续执行操作)
请你返回将 nums
数组和 至少 减少一半的 最少 操作数。
示例一:
输入:nums = [5,19,8,1] 输出:3 解释:初始 nums 的和为 5 + 19 + 8 + 1 = 33 。 以下是将数组和减少至少一半的一种方法: 选择数字 19 并减小为 9.5 。 选择数字 9.5 并减小为 4.75 。 选择数字 8 并减小为 4 。 最终数组为 [5, 4.75, 4, 1] ,和为 5 + 4.75 + 4 + 1 = 14.75 。 nums 的和减小了 33 - 14.75 = 18.25 ,减小的部分超过了初始数组和的一半,18.25 >= 33/2 = 16.5 。 我们需要 3 个操作实现题目要求,所以返回 3 。 可以证明,无法通过少于 3 个操作使数组和减少至少一半。
示例二:
输入:nums = [3,8,20] 输出:3 解释:初始 nums 的和为 3 + 8 + 20 = 31 。 以下是将数组和减少至少一半的一种方法: 选择数字 20 并减小为 10 。 选择数字 10 并减小为 5 。 选择数字 3 并减小为 1.5 。 最终数组为 [1.5, 8, 5] ,和为 1.5 + 8 + 5 = 14.5 。 nums 的和减小了 31 - 14.5 = 16.5 ,减小的部分超过了初始数组和的一半, 16.5 >= 31/2 = 15.5 。 我们需要 3 个操作实现题目要求,所以返回 3 。 可以证明,无法通过少于 3 个操作使数组和减少至少一半。
解题思路:这道题相对来说简单,依旧使用贪心算法,可以每次找出数组中最大的数让后将其减半
再放入数组中进行寻找,在除半,一次就可以使用贪心算法求解了。
由于一直要找数组中的最大值,左移我盟可以使用一个大根堆来求数组最大值,大根堆的第一个元
素就是数组最大值。
注意在减去数组最大值的一半后需要将数组中的最大值删除将最大值的一半放入,从而再次寻找最
大值。
对应的代码如下:
class Solution
{
public:
int halveArray(vector<int>& nums)
{
priority_queue<double> heap;
double sum = 0.0;
for(int x :nums)
{
sum += x;
heap.push(x);
}
sum /= 2.0;
int count = 0;
while(sum > 0)
{
double t = heap.top() / 2.0;
heap.pop();
sum -= t;
heap.push(t);
count++;
}
return count;
}
};
那么如何证明贪心算法就是最优解呢?
我们画图来说明:
1.x之前没使用过:
这是第一种情况下,贪心解可以与最优解交换。
2.x使用过了:
从而可以证明贪心解就是最优解。