§1
题目描述
每次拿最重的两块石头相撞,重的留下差值,直到剩 0 或 1 块。
示例 = [2,7,4,1,8,1] → 1
§2
思路解析动画文字版
每轮排序能做,但重复排序浪费。
用最大堆维护当前最重石头,每轮弹两次,再把差值压回。
1. 石头进最大堆:把所有石头放进最大堆,堆顶永远是当前最重的那块。每轮取最重的两块对撞。
2. 砸 8 和 7 → 剩 1 压回:取最重两块 8、7:不等 → 碎石 8−7=1 压回堆。堆顶现在是 4。
3. 砸 4 和 2 → 剩 2 压回:取 4、2:剩 4−2=2 压回。
4. 砸 2 和 1 → 剩 1 压回:取 2、1:剩 1 压回。
5. 关键帧 · 砸 1 和 1 → 同归于尽:关键帧:取 1、1 —— 相等!两块同时粉碎,什么都不压回堆。堆只剩 [1]。这是本题唯一的分支判断:等就都没、不等才留差。
6. 少于两块 · 答案 1:堆里少于 2 块,停。剩这一块 → 答案 1。(若石头全部两两抵消、堆空,则返回 0)
一句话记住:用最大堆维护当前最重石头,每轮弹两次,再把差值压回。
边界三连:三种情况先想清楚。
面试追问 1:核心思路。
面试追问 2:复杂度分析。
面试追问 3:Python 取负技巧。
这题学完不要乱跳,下一步去对应专题 /leetcode-animation/ds?k=heap 连刷同类题;卡住时用 AI 答疑问“我的状态定义哪里不稳”。
§3
参考代码
class Solution: def lastStoneWeight(self, stones): import heapq heap = [-x for x in stones] # 取负 → 用最小堆模拟最大堆 heapq.heapify(heap) while len(heap) > 1: a = -heapq.heappop(heap) # 最重 b = -heapq.heappop(heap) # 次重 if a != b: heapq.heappush(heap, -(a - b)) return -heap[0] if heap else 0看懂代码不等于写得出。盖住上面,自己默写一遍试试。卡在哪一行说不清,就点右下角问小欧。
§4
复杂度
- 时间复杂度:O(n log n),最多 n 轮,每轮弹两次压一次各 O(log n)
- 空间复杂度:O(n),堆存所有石头
§5
可套用的代码模板
骨架:最大堆反复取两块最重对撞,不等留差、等则同灭,最后看剩没剩。
Python
# 最后石头骨架 · 最大堆heap = 最大堆(stones)while len(heap) > 1: a = pop_max(heap) # 最重 b = pop_max(heap) # 次重 if a != b: push(heap, a - b) # 不等才留差return heap[0] if heap else 0§6
易错点
✗ 错误写法:用排序、每轮重新排
✓ 正确写法:用最大堆,取最大 O(log n)
每轮全排序是 O(n log n)、总 O(n²log n);堆每次只 O(log n)。
✗ 错误写法:相等时把 0 压回堆
✓ 正确写法:相等就都销毁,不压回
压回 0 会污染堆、还凭空多一块;题意是两块都碎。
✗ 错误写法:忘了最后可能空堆
✓ 正确写法:堆空返回 0
全部两两抵消时一块不剩,要返回 0。
§
面试追问把动画讲成自己的话
追问核心思路是什么?
用最大堆维护当前最重石头,每轮弹两次(最重、次重),不等就把差值压回,直到剩 ≤1 块。
这道题到这就讲完了。动画和文字是同一套思路——别光看,关掉页面自己默写一遍。然后顺着主线继续:
下一题 · 堆 / 优先队列 3/7
→最接近原点的 K 个点
LeetCode 973 · 中等 · 沿着 堆 / 优先队列 继续往下推进
把这道题真正学会,再走
图解算法年卡 ¥99 /年
- ✓本题每步动画的吴师兄语音讲解(全站陆续覆盖)
- ✓小欧带 8 步通关训练——追问到不看答案也能写对、能 30 秒讲给面试官听
- ✓学习报告,记录每道题的掌握程度
299道动画图解
37节语音讲解
28课数据结构
¥0.27折合 / 天
76k+ GitHub Star · 吴师兄开源图解算法,几十万开发者在看的算法讲解
¥99 开通年卡 →
想成体系刷透这类套路?去图解算法专题