2023-06-07 LeetCode每日一题（老鼠和奶酪）

2023-06-07每日一题

一、题目编号

2611. 老鼠和奶酪

二、题目链接

点击跳转到题目位置

三、题目描述

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

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

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

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

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

提示：

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

四、解题代码

class Solution {
public:
    int miceAndCheese(vector<int>& reward1, vector<int>& reward2, int k) {
        int n = reward1.size();
        vector<int> res(n);
        vector<int> nums[2100];
        for(int i = 0; i < n; ++i){
            res[i] = (reward1[i] - reward2[i]);
        }
        for(int i = 0; i <= 2000; ++i){
            nums[i].clear();
        }
        for(int i = 0; i < n; ++i){
            nums[res[i] + 1000].push_back(i);
        }
        sort(res.begin(), res.end(), greater<int>());
        int m = 0;
        int i = 0;
        int ret = 0;
        while(m < n){
            if(i < n - 1 && res[i] == res[i+1]){
                ++i;
                continue;
            }
            for(int j = 0; j < (int)nums[res[i] + 1000].size(); ++j){
                if(m < k){
                    ret += reward1[ nums[res[i] + 1000][j] ];
                    ++m;  
                } else{
                    ret += reward2[ nums[res[i] + 1000][j] ];
                    ++m;
                }
            }
            ++i;
        }
    return ret;
    }
};

五、解题思路

(1) 最重要的是，计算reward1和reward2的差距，然后将差距按照从大到小排，因为第一只老鼠吃k个，那么就按差距从大到小排的取得前k个，那么最重要的来了，如何从大到小排取前k个呢。

(2) 用res数组存取差距，然后用数字将每个不同的差距所对应的reward数组中的下标记录下来。然后将res数组按照从大到小排序。

(3) 利用循环来取得前k个reward1中的和，接着再计算reward2中的和。

(4) 这是一道典型的关于贪心算法的题目，