2208 将数组和减半的最少操作次数(堆)

1. 问题描述:

给你一个正整数数组 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 个操作使数组和减少至少一半。

示例 2:

输入: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 = 16.5 。
我们需要 3 个操作实现题目要求,所以返回 3 。
可以证明,无法通过少于 3 个操作使数组和减少至少一半。
 
提示:

1 <= nums.length <= 10 ^ 5
1 <= nums[i] <= 10 ^ 7

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/minimum-operations-to-halve-array-sum/

2. 思路分析:

因为需要使得操作次数最少,所以每一次我们需要从当前所有数中选择一个最大的数字让其减半,维护所有数字的最大值可以用大根堆,因为使用的是python语言所以可以使用heapq模块对列表进行操作来维护一个堆,而且python没有大根堆,需要在所有数字前加上一个负号,这样堆顶元素的相反数就是当前所有数字的最大值,然后模拟整个过程即可。

3. 代码如下:

import heapq
from typing import List


class Solution:
    def halveArray(self, nums: List[int]) -> int:
        q = list()
        s = 0
        for x in nums:
            s += x
            q.append(-x)
        # 所有元素加上一个负号表示大根堆
        heapq.heapify(q)
        s1 = s
        count = 0
        while True:
            t = heapq.heappop(q) / 2
            # 因为可以多次操作当前的数字所以可以将当前数字的一半加入到堆中
            heapq.heappush(q, t)
            s1 += t
            count += 1
            if s1 * 2 <= s: break
        return count
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值