题干:
有一堆石头,每块石头的重量都是正整数。
每一回合,从中选出两块 最重的 石头,然后将它们一起粉碎。假设石头的重量分别为 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;}
};