剩下最後的石頭

Tofaker發表於2020-11-16

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

每一回合,從中選出兩塊 最重的 石頭,然後將它們一起粉碎。假設石頭的重量分別為 x 和 y,且 x <= y。那麼粉碎的可能結果如下:

如果 x == y,那麼兩塊石頭都會被完全粉碎;
如果 x != y,那麼重量為 x 的石頭將會完全粉碎,而重量為 y 的石頭新重量為 y-x。
最後,最多隻會剩下一塊石頭。返回此石頭的重量。如果沒有石頭剩下,就返回 0。

連結:https://leetcode-cn.com/problems/last-stone-weight

思路 :建立大根堆(利用比較器重寫優先順序佇列的規則) ,然後每次依照題目的處理辦法對堆頂和堆第二個元素進行處理,

注意:return queue.size()==1 ? queue.peek() : 0;

class Solution {
public int lastStoneWeight(int[] stones) {
if (stones.length==0) return -1;
Queue queue = new PriorityQueue<>(new Comparator() {
@Override
public int compare(Integer o1, Integer o2) {
return o2-o1;
}
});
for (int i = 0; i < stones.length; i++) {
queue.offer(stones[i]);
}
while (queue.size() > 1){
int tmp = queue.poll();
if (queue.isEmpty()) return tmp;
if (tmp == queue.peek()){
queue.poll();
}else if (tmp > queue.peek()){
queue.offer(tmp-queue.poll());
}
}
return queue.size()==1 ? queue.peek() : 0;
}
}

相關文章