力扣---6364. 老鼠和奶酪

发布时间 2023-04-02 15:40:31作者: Owlwu

有两只老鼠和 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 <= 105
1 <= reward1[i], reward2[i] <= 1000
0 <= k <= n

来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/mice-and-cheese
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。


 

周赛第三题,也算是第一次达到周赛三题成就(上次思路正确,但代码细节上差了一步,没有做出来)

贪心法,见注释:

// 贪心法。
// 保证第一只老鼠吃掉 k 个奶酪的基础上,第二只老鼠吃的尽可能多。
// 可以用一个二维数组存储奶酪数,然后进行排序,这样可以保证两只老鼠吃到的不会是同一块奶酪。
// 排序规则是:按第一只老鼠吃到的奶酪减去第二只老鼠吃到的奶酪的大小进行排序(从大到小排序),这样可以保证第一只老鼠吃到k个奶酪的基础上,对第二只老鼠吃到的影响尽可能少。
class Solution {
    public int miceAndCheese(int[] reward1, int[] reward2, int k) {
        // 如果 k 等于奶酪总数,直接让第一只老鼠吃掉所有奶酪即可。
        if (k == reward1.length) {
            return Arrays.stream(reward1).sum();
        }
        // 存储两只老鼠吃到的奶酪数,保证排序后两只老鼠吃的是同一个。
        int[][] arr = new int[reward1.length][2];
        for (int i = 0; i < reward1.length; i ++) {
            arr[i][0] = reward1[i];
            arr[i][1] = reward2[i];
        }
        // 排序,排序规则见开头注释。
        Arrays.sort(arr, (a, b) -> ((b[0] - b[1]) - (a[0] - a[1])));
        int res = 0;
        // 先让第一只老鼠吃 k 个奶酪。
        for (int i = 0; i < k; i ++) {
            res += arr[i][0];
        }
        // 然后让第二只老鼠吃剩下的奶酪。
        for (int i = k; i < reward1.length; i ++) {
            res += arr[i][1];
        }
        return res;
    }
}