力扣 2611. 老鼠和奶酪 排序

题目

有两只老鼠和 n 块不同类型的奶酪，每块奶酪都只能被其中一只老鼠吃掉。

下标为 i 处的奶酪被吃掉的得分为：

如果第一只老鼠吃掉，则得分为 reward1[i] 。
如果第二只老鼠吃掉，则得分为 reward2[i] 。
给你一个正整数数组 reward1 ，一个正整数数组 reward2 ，和一个非负整数 k 。

请你返回第一只老鼠恰好吃掉 k 块奶酪的情况下，最大 得分为多少。

示例

输入：reward1 = [1,1,3,4], reward2 = [4,4,1,1], k = 2
输出：15
解释：这个例子中，第一只老鼠吃掉第 2 和 3 块奶酪（下标从 0 开始），第二只老鼠吃掉第 0 和 1 块奶酪。
总得分为 4 + 4 + 3 + 4 = 15 。
15 是最高得分。

代码解析

首先，假设第2只老鼠吃完了所有的奶酪。

其次，考虑两只老鼠吃每个奶酪的得分差值，可以得到，对于每个奶酪，第一只老鼠比第二只多得的分数，对分数进行升序排序，最后的k个奶酪，就是适合第一只老鼠吃的，加上这k个奶酪的差值，就可以得到最终答案了。

class Solution(object):
    def miceAndCheese(self, reward1, reward2, k):
        """
        :type reward1: List[int]
        :type reward2: List[int]
        :type k: int
        :rtype: int
        """
        res = sum(reward2)
        delt = []
        n = len(reward1)
        for i in range(n):
            delt.append(reward1[i] - reward2[i])

        a = sorted(delt)

        for i in range(k):
            res += a[-(i+1)]

        return res