开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第32天,点击查看活动详情
1753. 移除石子的最大得分
你正在玩一个单人游戏,面前放置着大小分别为 a、b 和 c 的 三堆 石子。
每回合你都要从两个 不同的非空堆 中取出一颗石子,并在得分上加 1 分。当存在 两个或更多 的空堆时,游戏停止。
给你三个整数 a 、b 和 c ,返回可以得到的 最大分数 。
示例 1:
输入: a = 4, b = 4, c = 6
输出: 7
解释: 石子起始状态是 (4, 4, 6) ,最优的一组操作是:
- 从第一和第二堆取,石子状态现在是 (3, 3, 6)
- 从第一和第三堆取,石子状态现在是 (2, 3, 5)
- 从第一和第三堆取,石子状态现在是 (1, 3, 4)
- 从第一和第三堆取,石子状态现在是 (0, 3, 3)
- 从第二和第三堆取,石子状态现在是 (0, 2, 2)
- 从第二和第三堆取,石子状态现在是 (0, 1, 1)
- 从第二和第三堆取,石子状态现在是 (0, 0, 0)
总分:7 分 。
提示:
1 <= a, b, c <= 105
思路
根据题目要求,若出现两个或两个以上的空堆时,游戏停止,若我们想得到最大得分,则需要尽可能地防止出现两个空堆,如果我们每次都取两个最多的堆的石子,就可以最大限度地避免出现两个空堆,故我们可以采用贪心的思想,每次在最多的两个堆中取石子即可。
具体来说,可以使用最大堆来实现,每次将最大的两个值减一,再重新入堆,当堆中非零的值小于等于一个时停止。
题解
class Solution {
public int maximumScore(int a, int b, int c) {
PriorityQueue<Integer> heap = new PriorityQueue<>((o1, o2) -> (o2 - o1));
heap.offer(a);
heap.offer(b);
heap.offer(c);
int cnt = 0;
while(heap.size() > 1) {
int max = heap.poll();
int mid = heap.poll();
cnt++;
max--;
mid--;
if(max > 0) {
heap.offer(max);
}
if(mid > 0) {
heap.offer(mid);
}
}
return cnt;
}
}