2023-06-07每日一题
一、题目编号
2611. 老鼠和奶酪
二、题目链接
三、题目描述
有两只老鼠和 n 块不同类型的奶酪,每块奶酪都只能被其中一只老鼠吃掉。
下标为 i 处的奶酪被吃掉的得分为:
- 如果第一只老鼠吃掉,则得分为 reward1[i] 。
- 如果第二只老鼠吃掉,则得分为 reward2[i] 。
给你一个正整数数组 reward1 ,一个正整数数组 reward2 ,和一个非负整数 k 。
请你返回第一只老鼠恰好吃掉 k 块奶酪的情况下,最大 得分为多少。
提示:
- 1 <= n == reward1.length == reward2.length <= 105
- 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) 这是一道典型的关于贪心算法的题目,