[路飞] 49——leetcode -1046. 最后一块石头的重量

Offer 驾到，掘友接招！我正在参与2022春招打卡活动，点击查看活动详情。

1046. 最后一块石头的重量

题目分析

有一堆石头，每块石头的重量都是正整数。

每一回合，从中选出两块 最重的 石头，然后将它们一起粉碎。假设石头的重量分别为 x 和 y，且 x <= y。那么粉碎的可能结果如下：

• 如果 x == y，那么两块石头都会被完全粉碎；
• 如果 x != y，那么重量为 x 的石头将会完全粉碎，而重量为 y 的石头新重量为 y-x。 最后，最多只会剩下一块石头。返回此石头的重量。如果没有石头剩下，就返回 0。

思路讲解

我们可以维护⼀个⼤根堆，然后每次取出堆顶的元素，两两相减，将结果再加⼊到堆中，直到堆中的元素⼩于两个

示例

输入：[2,7,4,1,8,1]
输出：1
解释：
先选出 7 和 8，得到 1，所以数组转换为 [2,4,1,1,1]，
再选出 2 和 4，得到 2，所以数组转换为 [2,1,1,1]，
接着是 2 和 1，得到 1，所以数组转换为 [1,1,1]，
最后选出 1 和 1，得到 0，最终数组转换为 [1]，这就是最后剩下那块石头的重量。

代码

/*
 * @lc app=leetcode.cn id=1046 lang=javascript
 *
 * [1046] 最后一块石头的重量
 */

// @lc code=start
/**
 * @param {number[]} stones
 * @return {number}
 */
var lastStoneWeight = function(stones) {
  const maxHeap = new MaxPriorityQueue() // new一个大根堆
  for (let i = 0; i < stones.length; i++) {
    maxHeap.enqueue('x', stones[i]) // 随便存一个数据'x', 数据不重要，需要的是石头的重量，把每个石头的重量扔进去
  }
  while (maxHeap.size() > 1) {
    // 每次取出来一个堆顶的元素，通过优先级取
    const a = maxHeap.dequeue()['priority'] // 拿到第一个最大的元素
    const b = maxHeap.dequeue()['priority'] // 拿到次大的元素
    if (a > b) {
      maxHeap.enqueue('x', a - b)
    }
  }
  return maxHeap.isEmpty() ? 0 : maxHeap.dequeue()['priority']
};
// @lc code=end