每日一题——移除石子的最大得分

38 阅读1分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 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;
    }
}