刷算法时使用PriorityQueue - Go语言中文社区

刷算法时使用PriorityQueue


Queue是一个先进先出(FIFO)的队列,而PriorityQueue特别的地方在于他的出队顺序与元素的优先级有关,对PriorityQueue调用方法remove()/poll(),返回的总是优先级最高的元素。

因此,放入PriorityQueue的元素必须实现接口comparator,PriorityQueue会根据元素的排序顺序决定出队的优先级。

举例说明:对nums2数组降序排列

  PriorityQueue<int[]> maxpq = new PriorityQueue<>(

                new Comparator<int[]>() {

                    public int compare(int[] pair1, int[] pair2) {

                        return pair2[1] - pair1[1];

                    }

                });

        for(int i=0;i<n;i++){

             maxpq.add(new int[]{i,nums2[i]});

        } 

或者是下面这种自定义方式也可以

PriorityQueue<int[]> maxpq=new PriorityQueue<>(

(int[] pair1,int[] pair2)->{

return pair2[1]-pair1[1];//根据第一维的元素降序排列

}

);

for(int i=0;i<n;i++){

maxpq.offer(new int[]{i,nums2[i]});

}

maxpq如下:

3120
111041

版权声明:本文来源CSDN,感谢博主原创文章,遵循 CC 4.0 by-sa 版权协议,转载请附上原文出处链接和本声明。
原文链接:https://blog.csdn.net/qq_41328481/article/details/123402234
站方申明:本站部分内容来自社区用户分享,若涉及侵权,请联系站方删除。
  • 发表于 2023-01-03 21:04:30
  • 阅读 ( 301 )
  • 分类:算法

0 条评论

请先 登录 后评论

官方社群

GO教程

猜你喜欢