Java之PriorityQueue的使用解读

 更新时间:2025年09月04日 09:11:46   作者:minh_coo  
本文介绍PriorityQueue的基本概念、常用方法及自定义排序技巧,重点讲解其在TOPK问题中的应用,通过小顶堆或大顶堆高效筛选最大/最小值,适用于大规模数据处理场景

PriorityQueue用法

一、基本概念

PriorityQueue(优先队列),在概念上,默认为小顶堆,元素单调递增排序。也可通过传入Comparator,重写其中的compare方法自定义排序规则;

在实现上,PriorityQueue实现了Queue接口,使用数组来存储数据,按照每层从左到右的顺序存放,因此它不允许存入null值。

二、常用方法总结

方法作用
add();队尾插入元素,调整堆结构,失败时抛异常
offer();队尾插入元素,调整堆结构,失败时抛false
remove();根据value值删除指定元素,调整堆结构,失败时抛异常
poll();删除队头元素,调整堆结构,失败时抛null
element();获取队列头元素
peek();获取队列头元素
clear();清空队列
size();获取队列元素个数
contains();判断队列中是否包含指定元素
isEmpty();判断队列是否为空

三、具体使用

1、实现降序排列(大顶堆)

方式一、lambda表达式

		PriorityQueue<Integer> queue = new PriorityQueue<>(
                (o1, o2) -> o2 - o1
        );

方式二、重写compare方法

		PriorityQueue<Integer> queue = new PriorityQueue<>(new Comparator<Integer>() {
            @Override
            public int compare(Integer o1, Integer o2) {
                return o2 - o1;
            }
        });

2、实现自定义排序

示例一、按字符串的第三位进行降序排列

		PriorityQueue<String> queue = new PriorityQueue<>(
                (o1, o2) -> o2.charAt(2) - o1.charAt(2)
        );

示例二、自定义一个类People,先按名字排序,再按年龄排序,再按身高排序

public class Solution {
    public static void main(String[] args) {
        PriorityQueue<People> queue = new PriorityQueue<>(
                (o1, o2) -> {
                    if (o1.getName().compareTo(o2.getName()) > 0) {
                        return 1;
                    } else if (o1.getName().compareTo(o2.getName()) < 0) {
                        return -1;
                    } else {
                        if (o1.getAge() > o2.getAge()) {
                            return 1;
                        } else if (o1.getAge() < o2.getAge()) {
                            return -1;
                        } else {
                            if (o1.getHeight() - o2.getHeight() > 0) {
                                return 1;
                            } else if (o1.getHeight() - o2.getHeight() < 0) {
                                return -1;
                            } else {
                                return 0;
                            }
                        }
                    }
                }
        );
        People one = new People("one", 12, 45.6f);
        People two = new People("one", 12, 12.3f);
        queue.add(one);
        queue.add(two);
        for (People tmp : queue) {
            System.out.println(tmp);
        }
    }
}

class People {
    private String name;
    private int age;
    private float height;

    public People(String name, int age, float height) {
        this.name = name;
        this.age = age;
        this.height = height;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public int getAge() {
        return age;
    }

    public void setAge(int age) {
        this.age = age;
    }

    public float getHeight() {
        return height;
    }

    public void setHeight(float height) {
        this.height = height;
    }

    @Override
    public String toString() {
        return "people{" +
                "name='" + name + '\'' +
                ", age=" + age +
                ", height=" + height +
                '}';
    }
}

3、解决TOP K问题

问题场景:从十亿个数中取最大/最小的100个数

解决方案:先取100个数构成小顶堆/大顶堆,后续每来一个数都与队头元素进行判断,比它小就直接丢弃,比它大就进队列中,直至访问完毕,最后剩下的即为所求答案。

总结

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

相关文章

  • java中把字符串转成 double的方法

    java中把字符串转成 double的方法

    Java 中可以使用 Double 类中的静态方法 parseDouble() 将一个字符串转换为 double 类型的数值,本文结合实例代码对java字符串转成 double详细讲解,需要的朋友参考下吧
    2023-08-08
  • java 过滤器模式(Filter/Criteria Pattern)详细介绍

    java 过滤器模式(Filter/Criteria Pattern)详细介绍

    这篇文章主要介绍了java 过滤器模式(Filter/Criteria Pattern)详细介绍的相关资料,需要的朋友可以参考下
    2016-10-10
  • 一文掌握Spring中循环依赖与三级缓存

    一文掌握Spring中循环依赖与三级缓存

    这篇文章主要介绍了Spring中循环依赖与三级缓存,Spring通过三级缓存解决了循环依赖,其中一级缓存为单例池,二级缓存为早期曝光对象earlySingletonObjects,三级缓存为早期曝光对象工厂(singletonFactories),本文结合实例代码介绍的非常详细,需要的朋友参考下吧
    2023-09-09
  • MybatisPlus IService接口功能介绍

    MybatisPlus IService接口功能介绍

    MybatisPlus中的IService接口为开发者提供了一系列基础数据库操作方法,如增删改查等,通过泛型支持不同类型的实体和主键,IService的默认实现类为ServiceImpl,它实现了接口中的基本方法,用户可通过继承ServiceImpl来扩展自定义的业务逻辑
    2024-09-09
  • 深入理解Java中的类加载器原理

    深入理解Java中的类加载器原理

    这篇文章主要介绍了深入理解Java中的类加载器原理,类加载器负责加载所有的类,系统为所有被载入内存中的类生成一个java.lang.Class实例,需要的朋友可以参考下
    2024-01-01
  • Java如何自动生成api文档

    Java如何自动生成api文档

    在 Java 开发中,自动生成 API 文档是一项非常实用的功能,这篇为大家介绍了几种常见的 Java 自动生成 API 文档的方式,需要的可以参考一下
    2025-02-02
  • java字符串的大写字母右移实现方法

    java字符串的大写字母右移实现方法

    下面小编就为大家带来一篇java字符串的大写字母右移实现方法。小编觉得听不错的,现在就分享给大家,也给大家做个参考。一起跟随小编过来看看吧
    2017-04-04
  • PageHelper插件实现一对多查询时的分页问题

    PageHelper插件实现一对多查询时的分页问题

    这篇文章主要介绍了PageHelper插件实现一对多查询时的分页问题,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2019-04-04
  • springboot实现敏感字段加密存储解密显示功能

    springboot实现敏感字段加密存储解密显示功能

    这篇文章主要介绍了springboot实现敏感字段加密存储,解密显示,通过mybatis,自定义注解+AOP切面,Base64加解密方式实现功能,本文通过代码实现给大家介绍的非常详细,需要的朋友可以参考下
    2022-02-02
  • java面试常见问题---ConcurrentHashMap

    java面试常见问题---ConcurrentHashMap

    ConcurrentHashMap是由Segment数组结构和HashEntry数组结构组成。Segment的结构和HashMap类似,是一种数组和链表结构,今天给大家普及java面试常见问题---ConcurrentHashMap知识,一起看看吧
    2021-06-06

最新评论