题目描述:
有两只鸭脖和 n
块不同类型的奶酪,每块奶酪都只能被其中一只鸭脖吃掉。
下标为 i
处的奶酪被吃掉的得分为:
- 如果第一只鸭脖吃掉,则得分为
reward1[i]
。 - 如果第二只鸭脖吃掉,则得分为
reward2[i]
。
给你一个正整数数组 reward1
,一个正整数数组 reward2
,和一个非负整数 k
。
请你返回第一只鸭脖恰好吃掉 k
块奶酪的情况下,最大 得分为多少。
题目分析:
每块奶酪都是要被鸭脖吃掉的,取决于被哪只鸭脖吃,第一只鸭脖恰好要吃掉k块奶酪,我们可以先让第二只鸭脖吃掉所有的鸭脖,使用贪心算法,将两只鸭脖分别要吃的奶酪差值计算出来放在一个数组中,将这个数组降序排序,从头选取k个奶酪给第一只鸭脖吃即可,这里直接累加差值更新结果即可。
源代码:
/**
* @param {number[]} reward1
* @param {number[]} reward2
* @param {number} k
* @return {number}
*/
function miceAndCheese(reward1, reward2, k) {
let res = 0;
const diff = new Array();
res += reward2.reduce((a, b) => { return a + b; });
for(let i = 0; i < reward1.length; ++i) {
diff.push(reward1[i] - reward2[i]);
}
diff.sort((a, b) => { return b - a; });
for(let i = 0; i < k; ++i) {
res += diff[i];
}
return res;
};