贪心算法+数据结构(大根堆): 将数组和减半的最少操作次数

题目:

给你一个正整数数组 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;
    }
}

有任何问题欢迎在评论区讨论!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值