LeetCode | 最后一块石头的重量

一、题目描述

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

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

如果 x == y，那么两块石头都会被完全粉碎；
如果 x != y，那么重量为 x 的石头将会完全粉碎，而重量为 y 的石头新重量为 y-x。

最后，最多只会剩下一块石头。返回此石头的重量。如果没有石头剩下，就返回 0。

提示：

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

二、解题思路

题目中每次都要选出最大的两块石头粉碎，自然而然想到先建立一个大根堆 然后每次在这个大根堆里取出头两个元素出来进行粉碎，如果两块石头不相等就将粉碎后的新石头重新放进大根堆 这样循环下去直到大根堆里的元素少于两块石头就结束 此时已经没法再粉碎了

三、程序实例

class Solution {
public:
    int lastStoneWeight(vector<int>& stones) {
        int len = stones.size();
        while(len > 1)
        {
            sort(stones.begin(), stones.end());
            int x = stones[len - 2];
            int y = stones[len - 1];
            
            if( x != y )
            {
                stones.erase(stones.end() - 1);
                len = stones.size();
                stones[len - 1] = y - x;
            }
            else
            {
                stones.pop_back();
                stones.pop_back();
                len = stones.size();
            }  
        }
        return (len < 1) ? 0 : stones[0];
    }
};