【算法刷题-第7题】执行k次操作后的最大分数-Python

目录

题目

题解

代码

---

题目

给你一个下标从 0 开始的整数数组 nums 和一个整数 k 。你的 起始分数 为 0 。

在一步 操作 中：

1. 选出一个满足 0 <= i < nums.length 的下标 i ，
2. 将你的 分数 增加 nums[i] ，并且
3. 将 nums[i] 替换为 ceil(nums[i] / 3) 。

返回在 恰好 执行 k 次操作后，你可能获得的最大分数。

向上取整函数 ceil(val) 的结果是大于或等于 val 的最小整数。

示例 1：

输入：nums = [10,10,10,10,10], k = 5
输出：50
解释：对数组中每个元素执行一次操作。最后分数是 10 + 10 + 10 + 10 + 10 = 50 。

示例 2：

输入：nums = [1,10,3,3,3], k = 3
输出：17
解释：可以执行下述操作：
第 1 步操作：选中 i = 1 ，nums 变为 [1,4,3,3,3] 。分数增加 10 。
第 2 步操作：选中 i = 1 ，nums 变为 [1,2,3,3,3] 。分数增加 4 。
第 3 步操作：选中 i = 2 ，nums 变为 [1,1,1,3,3] 。分数增加 3 。

题解

1. 最开始我直接才用暴力的方式，每次取最大的值，也就是每次都要sort一次，然后很遗憾超时了

2. 然后我发现，他题意很符合大根堆的性质，也就是父节点大于两个子节点

3. 于是直接构造一个大根堆，然后排序，每次取最大的值，然后取完之后，把新的值放在最上面，然后下沉操作，重新保证根节点最大

代码

class Solution:
    def maxKelements(self, nums: List[int], k: int) -> int:
        maxn = 0
        for i in range(len(nums)):
            nums[i] = -nums[i]   # 将相反的存入，维持大根堆
        heapify(nums)    # 建立一个堆
        
        for i in range(k):
            maxn -= nums[0]   # 相减，因为为负
            heapreplace(nums,-math.ceil(-nums[0]/3))  # 堆顶弹出，加入新数
            
        return maxn