題干：

有一堆石頭，每塊石頭的重量都是正整數。

每一回合，從中選出兩塊 最重的 石頭，然后將它們一起粉碎。假設石頭的重量分別為?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]，這就是最后剩下那塊石頭的重量。
?

提示：

1 <= stones.length <= 30
1 <= stones[i] <= 1000

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/last-stone-weight
著作權歸領扣網絡所有。商業轉載請聯系官方授權，非商業轉載請注明出處。

解題報告：

好像從來沒手寫過堆，做一個模板題以備面試使用。

AC代碼：

class Solution { public: //下標從0開始的堆void down(vector<int>& stones, int x) {int t = x;int left = x * 2 + 1;int right = x * 2 + 2;if(left <stones.size() && stones[left] > stones[t]) t = left;if(right < stones.size() && stones[right] > stones[t]) t = right;if(t != x) {swap(stones[t], stones[x]);down(stones, t);}}void up(vector<int>& stones, int x) {int fa = (x-1)/2;if(stones[x] > stones[fa]) {swap(stones[x], stones[fa]);up(stones, fa);}}void push(vector<int>& stones, int x) {stones.push_back(x);up(stones, stones.size()-1);}int pop(vector<int>& stones) {if(stones.size() == 0) return -1;int x = stones[0];swap(stones[0], stones[stones.size()-1]);stones.pop_back();//注意先pop再downdown(stones, 0);return x;}void init(vector<int>& stones) {for(int i = stones.size()/2 - 1; i>=0; i--) down(stones, i);}int lastStoneWeight(vector<int>& stones) {init(stones);while(stones.size() > 1) {int x1 = pop(stones);int x2 = pop(stones);if(x1 != x2) push(stones, abs(x1-x2));}if(stones.size() == 1) return pop(stones);else return 0;} };

總結

以上是生活随笔為你收集整理的【LeetCode1046】最后一块石头的重量（堆heap）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。