class Solution {
public int lastStoneWeight(int[] stones) {
PriorityQueue<Integer> queue = new PriorityQueue<>((o1,o2)->o2-o1);
for(int i=0;i<stones.length;i+=1){
queue.add(stones[i]);
}
while(queue.size()>0){
if(queue.size()==1){
return queue.poll();
}
else{
int num1 = queue.poll();
int num2 = queue.poll();
queue.add(Math.abs(num1-num2));
}
}
return 0;
}
}
核心就是使用一个使用优先队列构造一个最大堆
很多时候看到最大的前几个等等,我们很容易想到使用贪心算法,而优先队列常用来解决贪心算法问题。
要记住了出队列用poll(),出栈用pop()