leetcode 1962. Remove Stones to Minimize the Total（移除石头使总数最小）

最新推荐文章于 2025-12-03 23:35:37 发布

原创最新推荐文章于 2025-12-03 23:35:37 发布 · 153 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法

leetcode 专栏收录该内容

490 篇文章

订阅专栏

通过使用最大堆实现贪婪策略，此算法可在k次操作内将一堆石头的数量减至最少。每次操作从当前最大的石头堆中移除一半数量的石头。

在这里插入图片描述
piles数组表示每堆石头的个数，可以进行k次操作，使石头总数最小。
每次操作可以移除floor(piles[i]/2)个石头，piles[i]可重复移除。
floor是向下取整。

思路：

greedy
每次选最大的石头堆，就能移除最多的石头数。
所以用到max heap。

    public int minStoneSum(int[] piles, int k) {
        PriorityQueue<Integer> maxHeap = new PriorityQueue<>((a,b)->(b-a));
        int res = 0;

        for(int stones : piles) {
            maxHeap.add(stones);
        }

        while(k > 0) {
            int cur = maxHeap.poll();
            cur -= (int)Math.floor(cur/2);
            maxHeap.add(cur);
            k --;
        }

        for(Integer stones : maxHeap) {
            res += stones;
        }
        return res;
    }