每日1题 -老鼠和奶酪

最新推荐文章于 2024-06-12 10:00:23 发布

Destiny坠明

最新推荐文章于 2024-06-12 10:00:23 发布

阅读量291

点赞数

分类专栏： LeetCode算法文章标签： leetcode 算法动态规划 python

本文链接：https://blog.csdn.net/qq_55818063/article/details/131085489

版权

LeetCode算法专栏收录该内容

24 篇文章 2 订阅

订阅专栏

class Solution:
    def miceAndCheese(self, reward1: List[int], reward2: List[int], k: int) -> int:
        ans = 0
        n = len(reward1)
        diffs = [reward1[i] - reward2[i] for i in range(n)]
        ans += sum(reward2)
        diffs.sort()
        for i in range(1, k+1):
            ans += diffs[n - i]
        return ans

class Solution:
    def miceAndCheese(self, reward1: List[int], reward2: List[int], k: int) -> int:
        ans = 0
        n = len(reward1)
        pq = []
        for i in range(n):
            ans += reward2[i]
            heappush(pq, reward1[i] - reward2[i])
            if len(pq) > k:
                heappop(pq)
        while pq:
            ans += heappop(pq)
        return ans