leetcode 1046. 最后一块石头的重量

题目描述

有一堆石头，每块石头的重量都是正整数。

每一回合，从中选出两块 最重的 石头，然后将它们一起粉碎。假设石头的重量分别为 x 和 y，且 x <= y。那么粉碎的可能结果如下：

如果 x == y，那么两块石头都会被完全粉碎；
如果 x != y，那么重量为 x 的石头将会完全粉碎，而重量为 y 的石头新重量为 y-x。
最后，最多只会剩下一块石头。返回此石头的重量。如果没有石头剩下，就返回 0。

C++

class Solution {
public:
    //用最大堆实现
    int lastStoneWeight(vector<int>& stones) {
        make_heap(stones.begin(),stones.end(),less<int>()) ; //初始化建立最大堆
        
        int temp=0;
        while(1){
               
             if(stones.size()==1){
                  return stones.back();
            }
            if(stones.size()==0){
                return 0;
            }
            pop_heap(stones.begin(),stones.end(),less<int>());  //删除堆顶元素并且自动调整堆
            int s1=stones.back();// 最重堆
            stones.pop_back();
            pop_heap(stones.begin(),stones.end(),less<int>());  //删除堆顶元素并且自动调整堆
            int s2=stones.back();// 第二重堆
            stones.pop_back();
            temp=s1-s2; 
            //若两块石头不是都被完全粉碎
            if(temp){
            stones.push_back(temp); //将temp添加到向量尾部
            push_heap(stones.begin(),stones.end(),less<int>());  //添加到堆并且调整堆
            }
           
        }
        }
     
};