【LeetCode:1962. 移除石子使总数最小 | 堆 + 贪心】

算法题

算法刷题专栏 | 面试必备算法 | 面试高频算法
越难的东西,越要努力坚持，因为它具有很高的价值，算法就是这样✨
作者简介：硕风和炜，CSDN-Java领域新星创作者，保研|国家奖学金|高中学习JAVA|大学完善JAVA开发技术栈|面试刷题|面经八股文|经验分享|好用的网站工具分享
恭喜你发现一枚宝藏博主,赶快收入囊中吧
人生如棋，我愿为卒，行动虽慢，可谁曾见我后退一步？

算法题

题目链接

• 1962. 移除石子使总数最小

⛲ 题目描述

给你一个整数数组 piles ，数组 下标从 0 开始 ，其中 piles[i] 表示第 i 堆石子中的石子数量。另给你一个整数 k ，请你执行下述操作 恰好 k 次：

选出任一石子堆 piles[i] ，并从中 移除 floor(piles[i] / 2) 颗石子。
注意：你可以对 同一堆 石子多次执行此操作。

返回执行 k 次操作后，剩下石子的 最小 总数。

floor(x) 为 小于 或 等于 x 的 最大 整数。（即，对 x 向下取整）。

示例 1：

输入：piles = [5,4,9], k = 2
输出：12
解释：可能的执行情景如下：

• 对第 2 堆石子执行移除操作，石子分布情况变成 [5,4,5] 。
• 对第 0 堆石子执行移除操作，石子分布情况变成 [3,4,5] 。
  剩下石子的总数为 12 。
  示例 2：

输入：piles = [4,3,6,7], k = 3
输出：12
解释：可能的执行情景如下：

• 对第 2 堆石子执行移除操作，石子分布情况变成 [4,3,3,7] 。
• 对第 3 堆石子执行移除操作，石子分布情况变成 [4,3,3,4] 。
• 对第 0 堆石子执行移除操作，石子分布情况变成 [2,3,3,4] 。
  剩下石子的总数为 12 。

提示：

1 <= piles.length <= 105
1 <= piles[i] <= 104
1 <= k <= 105

求解思路&实现代码&运行结果

---

⚡ 堆 + 贪心

求解思路

1. 我们每次合并所有石头堆中石头最多的石头，合并k次，统计最终的结果返回。
2. 这个过程维护的状态放入大根堆中，每次找到最多的石头个数。
3. 实现代码如下所示：

实现代码

class Solution {
    public int minStoneSum(int[] piles, int k) {
        PriorityQueue<Integer> queue = new PriorityQueue<>((a, b) -> (b - a));
        int ans = 0;
        for (int v : piles)
            queue.add(v);
        while (k-- > 0) {
            int t = queue.poll();
            queue.add((t + 1) / 2);
        }
        while (!queue.isEmpty()) {
            ans += queue.poll();
        }
        return ans;
    }
}

运行结果

---

共勉

最后，我想和大家分享一句一直激励我的座右铭，希望可以与大家共勉！