## 题目链接
2611. 老鼠和奶酪 - 力扣(LeetCode)
## 题目描述
有两只老鼠和 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
## 解题思路
### 贪心
不管第一只老鼠是吃0块还是1块还是n块,剩下的奶酪都要被第二只老鼠吃掉,这样才能满足最大的利益(最高分)。所以可以先假设第二只老鼠吃掉了全部奶酪,然后相同下标的奶酪,有的reward1分数高,有的reward2分数高,也可能reward1和reward2分数相等。那么在所有reward1-reward2的差值中找出最大的k个(里面也有可能包含负数)。k个差值的和加上第二只老鼠吃掉全部奶酪的和,就是最高分。
Java代码实现:
public int miceAndCheese(int[] reward1, int[] reward2, int k) {
int ans = 0;
int n = reward1.length;
PriorityQueue<Integer> pq = new PriorityQueue<Integer>();
for (int i = 0; i < n; i++) {
ans += reward2[i];
pq.offer(reward1[i] - reward2[i]);
if (pq.size() > k) {
pq.poll();
}
}
while (!pq.isEmpty()) {
ans += pq.poll();
}
return ans;
}