java中用优先队列解决最后一块石头重量问题

问题如下:
在这里插入图片描述
遇到这种选出最重最大最小的问题,首先我们就想到可以用优先队列来解决。

public int lastStoneWeight(int[] stones) {
       
        PriorityQueue<Integer> queue = new PriorityQueue<>((o1, o2) -> o2-o1);
        for(int i:stones) {
            queue.offer(i);
        }
        while(queue.size() >= 2) {
            Integer x = queue.poll();
            
            Integer y = queue.poll();
            
            queue.offer(x-y);
            
        }
        
        
     return queue.poll();
    }

我们可以实例化一个优先队列,由于标准库中的优先队列默认为小堆,就是最先出队列的是最小的,此时呢我们就自己定义优先级,让最大的先出队列,操作就是这段代码:(o1, o2) -> o2-o1
创建好后,把数组中的值入队列。由于要一次出两个数,循环条件就设为,队列中的个数最少要为2,然后出两次队列就把队列中最大的两个数找出来了,最后把两数之差再加入队列中,此处有点小机灵,提上说,如果队列最后为空,就返回0,否则就返回剩下的那个数,我们这里不管两个数的大小关系,直接把第一个数和第二个数的差加入到队列里,加入两个数不相等,则就是题上意思把两数的差加入到队列里,假如两个数相等,则把0加入到队列里了,假如前面队列里的数都出队列完了,我们可以去试试,结果队列里就只剩下一个0了,直接返回这个队列出队列后的值,就是0,假如原本队列里还剩一个数,那和0作差后,结果也还是那个数,所以我们直接出循环后返回出队列后的值,就是最终结果,方便了不少。

猜你喜欢

转载自blog.csdn.net/AIJXB/article/details/105977875