1046. 最后一块石头的重量

最新推荐文章于 2024-03-04 11:00:49 发布

原创最新推荐文章于 2024-03-04 11:00:49 发布 · 180 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#优先级队列 #leetcode #数据结构

数据结构与算法专栏收录该内容

186 篇文章

订阅专栏

该博客介绍了一个利用大顶堆解决石头碰撞问题的算法。每回合选取两块最重的石头进行碰撞，根据重量差更新堆。最终返回剩下石头的重量，如果没有石头则返回0。该算法涉及数据结构和动态规划思想。

题目描述：有一堆石头，每块石头的重量都是正整数。
每一回合，从中选出两块最重的石头，然后将它们一起粉碎。假设石头的重量分别为 x 和 y，且 x <= y。那么粉碎的可能结果如下：
如果 x == y，那么两块石头都会被完全粉碎；
如果 x != y，那么重量为 x 的石头将会完全粉碎，而重量为 y 的石头新重量为 y-x。
最后，最多只会剩下一块石头。返回此石头的重量。如果没有石头剩下，就返回 0。
解题思路：用大顶推存储所有的石头，每次取出最大的两块做碰撞，如果有剩余的则将差值加入队列中，直到只剩下不大于1块石头，没有石头则返回零，代码如下：

class Solution:
    def lastStoneWeight(self, stones: List[int]) -> int:
        from queue import PriorityQueue
        q = PriorityQueue()
        for s in stones:
            q.put(-s)
        while(q.qsize() > 1):
            cur1 = q.get()
            cur2 = q.get()
            if cur1 != cur2:
                q.put(cur1-cur2)
        return -q.get() if q.qsize() else 0