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

2 篇文章 0 订阅
2 篇文章 0 订阅

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

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作差后,结果也还是那个数,所以我们直接出循环后返回出队列后的值,就是最终结果,方便了不少。

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

lhj_loveFang_1105

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值