构建大小顶堆

Posted 2021-03-05 1996yrb

　　假设堆中存int类型的数据，直接一波PriorityQueue<Integer> queue = new PriorityQueue<>();即可完成小顶堆的构建，直接A了，因为默认的用new创建优先队列对象为小顶堆。

但是如何完成大顶堆的构建呢？

先介绍一种简洁的写法，利用lambda表达式来操作，PriorityQueue<Integer> queue = new PriorityQueue<>((x,y) -> y - x);即可A了。

还有一种写法如下所示：

static Comparator<Integer> cmp = new Comparator<Integer>() {
      public int compare(Integer e1, Integer e2) {   //注意e1和e2的顺序比较
        return e2 - e1;
      }
};

通过传入一个比较器，重新定义一下比较规则，再通过构造方法创建对象即可创建一个大顶堆。
PriorityQueue<Integer> queue = new PriorityQueue<>(cmp);

　Comparator<Object> cmp = new Comparator<Object>() {
　　　public int compare(Object o1, Object o2) {  //不同写法产生不同的比价方法
　　　//升序
　　　return o1-o2;
　　　//降序
　　　return o2-o1;
　　　}
　};