题目:
给你一个正整数数组 nums
。每一次操作中,你可以从 nums
中选择 任意 一个数并将它减小到 恰好 一半。(注意,在后续操作中你可以对减半过的数继续执行操作)
请你返回将 nums
数组和 至少 减少一半的 最少 操作数。
示例 1:
输入: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 个操作使数组和减少至少一半。
分析:
根据题目描述,每一次操作,都会将数组中的一个数减半。要使得数组和至少减少一半的操作次数最少,那么每一次操作都应该选择当前数组中的最大值进行减半。这就是所谓贪心思想(在每一个操作步骤都选取该步骤的最优解)。
然而,我没如果只是用数组维护数据,每一次都要遍历一边数组来找到最大值,因此,我们在这里用大根堆的数据结构来维护数据,这样就能很简单的拿出数据中的最大值。
完整代码如下:
//java代码
class Solution {
public int halveArray(int[] nums) {
//设置大根堆,方便读取最大数字
PriorityQueue<Double> head = new PriorityQueue<>((a,b) -> b.compareTo(a));
double sum = 0.0;//定义数组和
for(int i :nums){
//将sums数组里面的数字i放进大根堆里面
head.offer((double)i);
sum+=i;
}
sum /= 2;//将数组和变成原来的一半
int count = 0;//count用来计数;
while(sum >0){
//找出最大的值将其减半,然后在计算新的数组和看其是否小于等于原数组和的一半
double t = head.poll() /2.0;
sum -= t;
count++;
//将最大值操作完以后仍要将其放回大顶堆
head.offer(t);
}
return count;
}
}
有任何问题欢迎在评论区讨论!