大根堆和小根堆

最新推荐文章于 2025-11-15 00:15:00 发布

原创最新推荐文章于 2025-11-15 00:15:00 发布 · 2.2k 阅读

6 ·

CC 4.0 BY-SA版权

文章标签：

#大根堆 #小根堆

java 同时被 2 个专栏收录

10 篇文章

订阅专栏

java面试

2 篇文章

订阅专栏

本文通过一个Java实例展示了如何使用优先队列实现大根堆和小根堆，并演示了这两种堆结构如何进行基本的操作如添加元素和移除元素。通过具体数组操作的例子，读者可以直观地看到堆排序的过程。

大根堆和小根堆在排序和选择第K大的数中经常有用到。
在这里插入图片描述
在Java中是可以直接实现这两个中结构的，使用的优先队列

public class TIMU1 {

    //小根堆
    public PriorityQueue<Integer> minHead = new PriorityQueue<>(new Comparator<Integer>() {
        @Override
        public int compare(Integer o1, Integer o2) {
            return o1 - o2;
        }
    });

    //大根堆
    public PriorityQueue<Integer> maxHead = new PriorityQueue<>(new Comparator<Integer>() {
        @Override
        public int compare(Integer o1, Integer o2) {
            return o2 - o1;
        }
    });

    public static void main(String[] args) {
        TIMU1 ti = new TIMU1();
        int[] arr = {2,9,4,7,3,6};

        for(int i : arr){
            ti.minHead.add(i);
            ti.maxHead.add(i);
        }

        while(!ti.minHead.isEmpty() ){
            System.out.print(ti.minHead.poll() + " ");
        }
        System.out.println();
        while(!ti.maxHead.isEmpty()){
            System.out.print(ti.maxHead.poll() + " ");
        }
    }
}