PriorityQueue(优先队列)

hot100的常用JAVA API：PriorityQueue(优先队列)

---

一、PriorityQueue是什么？

翻译过来就是优先队列，本质是一个堆， 默认情况下堆顶每次都保留最小值，每插入一个元素，仍动态维护堆顶为最小值。

二、常用函数

1.初始化

代码如下（示例）：使用默认的初始容量（11）创建一个 PriorityQueue，并根据其自然顺序对元素进行排序。

PriorityQueue<Integer> Q = new PriorityQueue<>();

2.修改变成大顶堆(默认是小顶堆)

这里的知识点包括：

• PriorityQueue 默认是小根堆，大根堆需要重写比较器。
• 可以在 new PriorityQueue<>() 中的参数部分加入比较器。
• 具体写法是：(v1, v2) -> v2 - v1。
• Queue 类的输入是 offer() 方法，弹出是 poll() 方法。

//1.lambda表达式写法
Queue<Integer> queueA = new PriorityQueue<>((v1, v2) -> v2 - v1);
//2.匿名内部类写法
Queue<Integer> queueA = new PriorityQueue<>(new Comparator<Integer>() {

			@Override
			public int compare(Integer o1, Integer o2) {
				// TODO 自动生成的方法存根
				return o2-o1;
			}
});

2.队列中插入元素

Q.add(a);
Q.offer(E e);

3.获取元素(最小值或者最大值)

Q.peek();//获取最小值(默认)或者获取最大值
Q.poll();//获取并且删除最小值(默认)或者获取最大值

4.判断是否包含某个元素、移除指定元素、获取元素个数

Q.contains(Object o) // 如果包含指定元素返回true
Q.remove(Object o) // 移除指定元素
Q.size() // 返回元素个数

5.程序实例

5.1实现小顶堆并获取最小的元素

//     输入:   3 5 7 2 1 -1 10 -2 -3 52
public class Queue01 {
    public static void main(String[] args) {
//        Queue<Integer> queueA = new PriorityQueue<>((v1, v2) -> v2 - v1);
//        queueA.offer(1);
//        queueA.offer(2);
//        queueA.offer(3);
//        System.out.println(queueA.poll());
//        queueA.size();

        Scanner in = new Scanner(new InputStreamReader(System.in));
        PriorityQueue<Integer> Q = new PriorityQueue<>();
        int a;
        for(int i = 0; i < 10; i++){
            a = in.nextInt();
            Q.add(a);
        }
//        Q.poll();
        System.out.print(Q.peek()+" ");// 输出当前队列的最小元素 -3

    }
}

5.2实现大顶堆并获取最大的元素

//输入: 3 5 7 2 1 -1 10 -2 -3 52 
public class Queue01 {
    public static void main(String[] args) {
//        Queue<Integer> queueA = new PriorityQueue<>((v1, v2) -> v2 - v1);
//        queueA.offer(1);
//        queueA.offer(2);
//        queueA.offer(3);
//        System.out.println(queueA.poll());
//        queueA.size();

        Scanner in = new Scanner(new InputStreamReader(System.in));
        PriorityQueue<Integer> Q = new PriorityQueue<>(new Comparator<Integer>() {
            @Override
            public int compare(Integer o1, Integer o2) {
                return o2 - o1;
            }
        });
        int a;
//        3 5 7 2 1 -1 10 -2 -3 52
        for(int i = 0; i < 10; i++){
            a = in.nextInt();
            Q.add(a);
        }
        Q.poll();
        System.out.print(Q.peek()+" ");// 输出当前队列的最大元素 10

    }
}

总结

以上就是对优先队列的常用的一些API的简单介绍，方便大家在刷力扣的时候使用。具体底层数据结构可以参考这篇文章:
Java堆结构PriorityQueue完全解析
可以拿力扣中的题目进行验证自己是否掌握优先队列:
1.前 K 个高频元素
2.合并K个升序链表