leetcode-最后一块石头的重量

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()

发布了48 篇原创文章 · 获赞 0 · 访问量 4343

猜你喜欢

转载自blog.csdn.net/weixin_41327340/article/details/103722990