1046. 最后一块石头的重量

Posted 2021-09-06 易小顺

算法记录

LeetCode 题目：

  有一堆石头，每块石头的重量都是正整数。
  每一回合，从中选出两块 最重的 石头，然后将它们一起粉碎。假设石头的重量分别为 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]，这就是最后剩下那块石头的重量。

二、分析

• 分析题意可知，需要维护一个保持列表最大值的结构，每次从中取出前两个最大值进行差值运算，再将剩余的放入列表中。
• 这里我们就可以对列表进行排序操作，每次的数据插入或者删除都需要进行维护序列。
• 优先队列恰好符合我们的需求。

class Solution {
    public int lastStoneWeight(int[] stones) {
        Queue<Integer> queue = new PriorityQueue<>(new Comparator<Integer>() {
            public int compare(Integer e1, Integer e2) {
                return e2 - e1;
            }
        });   
        for(int i : stones)
            queue.add(i);
        int e1 = 0;
        int e2 = 0;
        while(queue.size() > 1) {
            e1 = queue.poll();
            e2 = queue.poll();
            if(e1 != e2)
                queue.add(Math.abs(e1 - e2));
        }
        if(queue.size() == 1)
            return queue.poll();
        else
            return 0;
    }
}

---

总结

熟悉优先队列的用法和实际结构。