1046. 最后一块石头的重量

链接：https://leetcode-cn.com/problems/last-stone-weight/

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

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

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

 

提示：

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

一、不断排序，用greater<int>()使向量从大到小排序，因为每次比\必有一个0，因此每次循环后len-1，使用--len是使得len为1时不进入循环，之间输出

另注意stones[0]为0时不要len--，这样可能会出现len为0，然后在while中len变为-1，导致死循环

class Solution {
public:
    int lastStoneWeight(vector<int>& stones) {
        int len = stones.size();

        sort(stones.begin(), stones.end(), greater<int>());
        while(--len)
        {
            stones[0] = abs(stones[0] - stones[1]);
            stones[1] = 0;
            sort(stones.begin(), stones.end(), greater<int>());
        }

        return stones[0];
    }
};

二、优先队列

要注意若出现两个0，都不加到优先队列中，而在输出时要根据队列的大小进行输出，若为0（如输入 2，2），输出0

class Solution {
public:
    int lastStoneWeight(vector<int>& stones) {
        priority_queue<int> pq;
        int len = stones.size();
        for(int i = 0; i < len; ++i)
        {
            pq.push(stones[i]);
        }

        while(pq.size() > 1)
        {
            int x = pq.top(); pq.pop();
            int y = pq.top(); pq.pop();
            int ans = abs(x - y);
            if(ans)
                pq.push(ans);
        }
        if(pq.size())
            return pq.top();
        return 0;
    }
};