题目描述:
有两只老鼠和 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 是最高得分。
实现代码:
class Solution {
public:
int miceAndCheese(vector<int>& reward1, vector<int>& reward2, int k) {
int ans = 0;//初始化得分变量为0
int n = reward1.size();//计算奶酪的数量
vector<int> diffs(n);
for (int i = 0; i < n; i++) {
ans += reward2[i];//先让所有奶酪让小鼠2全部吃掉所得积分
diffs[i] = reward1[i] - reward2[i];//依次计算所有奶酪小鼠1和小鼠2的得分差值
}
sort(diffs.begin(), diffs.end());//按从大到小的规则对差值进行排序
for (int i = 1; i <= k; i++) {
ans += diffs[n - i];//将分数差别较大的几块替换掉
}
return ans;
}
};
由于题干有提到每块奶酪都会被吃掉,所以可以让小鼠2将所有奶酪先吃掉,再根据差值进行替换。