技術標籤：每日一題javaleetcodequeue

1046. 最後一塊石頭的重量

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

每一回合，從中選出兩塊 最重的 石頭，然後將它們一起粉碎。假設石頭的重量分別為 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

方法一：大頂堆

因為每次都要選擇最大的兩個石頭，每輪 pk 如果 x > y 需要把新石頭加入佇列，所以需要佇列一直保持有序。每次新增新石頭都重新排序的時間複雜度為O( nlogn)，使用大頂堆的複雜度為 O(logn)。

public
 
 int lastStoneWeight(int[] stones) {
    PriorityQueue<Integer> queue = new PriorityQueue<>(Collections.reverseOrder());
    for (int x : stones) {
        queue.offer(x);
    }

    while (queue.size() >= 2) {
        int x = queue.poll();
        int y = queue.poll();
        if (x >
 
 y) {
            queue.offer(x - y);
        }
    }
    return queue.isEmpty() ? 0 : queue.poll();
}

執行結果