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个升序链表