Leetcode.2611 老鼠和奶酪

题目链接

Leetcode.2611 老鼠和奶酪 Rating : 1663

题目描述

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

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

• 如果第一只老鼠吃掉，则得分为 $reward1[i]$。
• 如果第二只老鼠吃掉，则得分为 $reward2[i]$ 。

给你一个正整数数组 $reward1$ ，一个正整数数组 $reward2$ ，和一个非负整数 $k$ 。

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

示例 1：

输入：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：

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

提示：

• $1<=n==reward1.length==reward2.length<=10^5$
• $1<=reward1[i],reward2[i]<=1000$
• $0<=k<=n$

解法：排序 + 贪心

我们先假设初始的时候，$n$ 块奶酪都被第二只老鼠吃掉了，那么此时的最大的分为 $sum$。

这时考虑奶酪被第一只老鼠吃掉的情况。假设此时第一个奶酪被第一只老鼠吃了，那么此时的得分就变成了 $sum+reward1[0]-reward2[0]$，假设此时第 $i$ 个奶酪被第一只老鼠吃了，那么此时的得分就变成了 $sum+reward1[i-1]-reward2[i-1]$。

我们要选择 $k$ 块奶酪被第一只老鼠吃掉，又由于我们要求的是最大值。所以我们就选择 $k$ 个最大的即可，从这里面选 { r e w a r d 1 [ 0 ] − r e w a r d 2 [ 0 ] , r e w a r d 1 [ 1 ] − r e w a r d 2 [ 1 ] , r e w a r d 1 [ 2 ] − r e w a r d 2 [ 2 ] , . . . , r e w a r d 1 [ n − 1 ] − r e w a r d 2 [ n − 1 ] } \{ reward1[0] - reward2[0],reward1[1] - reward2[1],reward1[2] - reward2[2],...,reward1[n-1] - reward2[n-1]\} {reward1[0]−reward2[0],reward1[1]−reward2[1],reward1[2]−reward2[2],...,reward1[n−1]−reward2[n−1]}。

时间复杂度： $O(n)$

C++代码：

class Solution {
public:
    int miceAndCheese(vector<int>& reward1, vector<int>& reward2, int k) {
        int sum = 0 , n = reward1.size();

        for(auto x:reward2) sum += x;

        vector<int> d(n);
        for(int i = 0;i < n;i++) d[i] = reward1[i] - reward2[i];

        sort(d.begin(),d.end(),greater<int>());

        for(int i = 0;i < k;++i) sum += d[i];

        return sum;
    }
};