Java优先级队列PriorityQueue

1.优先级队列概述

PriorityQueue,即优先队列。优先队列的作用是能保证每次取出的元素都是队列中权值最小的(Java的优先队列每次取最小元素,C++的优先队列每次取最大元素)。这里牵涉到了大小关系,元素大小的评判可以通过元素本身的自然顺序(natural ordering),也可以通过构造时传入的比较器(Comparator,类似于C++的仿函数)。

Java中PriorityQueue实现了Queue接口,不允许放入null元素;其通过堆实现,具体说是通过完全二叉树(complete binary tree)实现的小顶堆(任意一个非叶子节点的权值,都不大于其左右子节点的权值),也就意味着可以通过数组来作为PriorityQueue的底层实现。

2.优先级队列的操作方法

栈中的插入和移除数据项的命名一般都是 push、pop,而队列至今没有标准化的命名。插入操作可以用 add、offer等命名,移除操作可以用 poll、remove、等命名,查看队头操作可以用 peek、element等命令
Java中PriorityQueue实现了Queue接口,Queue接口实现了Collection接口,所以PriorityQueue包含了collection接口的add,remove,element方法
在 Java 中,常见的队列操作以及它们的区别如下所示:

操作方法名描述
插入offer向队列插入元素,在一个满的队列中加入一个新项会返回false
插入add向队列插入元素,在一个满的队列中加入一个新项会抛出异常
删除poll从队列中删除第一个元素,在队列为空时返回null
删除remove从队列中删除第一个元素,在队列为空时会抛出异常
查看队头元素peek在队列的头部查询元素,在队列为空时返回null
查看队头元素element从队列中删除第一个元素,在队列为空时会抛出异常

3.优先级队列的应用

3.1.最小K个数

3.1.1.问题描述

设计一个算法,找出数组中最小的k个数。以任意顺序返回这k个数均可。
示例:

输入: arr = [1,3,5,7,2,4,6,8], k = 4
输出: [1,2,3,4]

提示:

0 <= len(arr) <= 100000
0 <= k <= min(100000, len(arr))

3.1.2.解题思路及实现代码

我们用一个小根堆(优先级队列)实时维护数组的前 k 小值。首先将数组中的全部元素压入堆中(优先级队列),然后出队k个元素即可,求得最小的k个元素

class Solution {
    public int[] smallestK(int[] arr, int k) {
        Queue<Integer> queue=new PriorityQueue<Integer>();      //优先级队列
        int len=arr.length;
        for(int i=0;i<len;i++){             //将数组中元素入队
            queue.offer(arr[i]);
        }
        int [] result=new int[k];
        for(int i=0;i<k;i++){               //出队k个元素
            result[i]=queue.poll();
        }
        return result;
    }
}
  • 4
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值