每日一题 2530. 执行 K 次操作后的最大分数（中等，最大根堆）

eyvr

于 2023-10-18 15:10:39 发布

阅读量237

点赞数

分类专栏：用Python刷力扣文章标签：算法 leetcode python

本文链接：https://blog.csdn.net/qq_46636391/article/details/133906733

版权

用Python刷力扣专栏收录该内容

108 篇文章

订阅专栏

本文介绍了一种使用Pythonheapq模块解决的问题，即在给定列表中始终保持最大的k个元素，并在每个步骤后将最大元素替换为其值除以三。解决方案中涉及heapify和heapreplace函数的应用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在这里插入图片描述
显然每次需要取出最大的元素，用过后将他除以三重新加入数组中，所以只要维护一个最大根堆即可

class Solution:
    def maxKelements(self, nums: List[int], k: int) -> int:
        nums = [-i for i in nums]
        heapq.heapify(nums)
        ans = 0
        for i in range(k):
            ans += -heapq.heapreplace(nums, -ceil(-nums[0]/3))
        
        return ans