前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >【数据结构】优先级队列(堆)

【数据结构】优先级队列(堆)

作者头像
xxxflower
发布2023-04-16 17:47:27
2630
发布2023-04-16 17:47:27
举报
文章被收录于专栏:《数据结构》《数据结构》

1.优先级队列

1.1概念

队列是一种先进先出的数据结构。但有些情况下,操作的数据可能带有优先级,一般出队列时,可能需要优先级高的的元素先出队列。这种情况下,数据结构提供两个最基本的操作,一个是返回最高优先级对象,一个是添加新的对象,这种数据结构称之为优先级队列(Priority Queue)

2.优先级队列的模拟实现

JDK1.8中的PriorityQueue底层使用了堆的数据结构,而堆实际就是在完全二叉树的基础之上进行了一些元素的调整。 如果有一个关键码的集合K = {k0,k1, k2,…,kn-1},把它的所有元素按完全二叉树的顺序存储方式存储 在一个一维数组中,并满足:Ki <= K2i+1 且 Ki<= K2i+2 (Ki >= K2i+1 且 Ki >= K2i+2) i = 0,1,2…,则称为 小堆(或大堆)。将根节点最大的堆叫做最大堆或大根堆,根节点最小的堆叫做最小堆或小根堆。

在这里插入图片描述
在这里插入图片描述

2.1堆的存储方式

堆是一颗完全二叉树,因此可以层序的规则采用顺序的方式来高效存储。

在这里插入图片描述
在这里插入图片描述

注意:对于非完全二叉树,则不适合使用顺序方式进行存储,因为为了能够还原二叉树,空间中必须要存储空节点,就会导致空间利用率比较低。 将元素存储到数组中后,可以根据二叉树章节的性质5对树进行还原。假设i为节点在数组中的下标,则有:

  • 如果i为0,则i表示的节点为根节点,否则i节点的双亲节点为 (i - 1)/2
  • 如果2 * i + 1 小于节点个数,则节点i的左孩子下标为2 * i + 1,否则没有左孩子
  • 如果2 * i + 2 小于节点个数,则节点i的右孩子下标为2 * i + 2,否则没有右孩子

2.2堆的创建

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
代码语言:javascript
复制
public class TestHeap {
    public int[] elem;
    public int usedSize;//有效的数据个数
    public static final int DEFAULT_SIZE = 10;
    public TestHeap() {
        elem = new int[DEFAULT_SIZE];
    }
    public void initElem(int[] array){
        for(int i = 0;i < elem.length;i++){
            elem[i] = array[i];
            usedSize++;
        }
    }
    public void createHeap(){
        for(int parent = ((usedSize-1)-1)/2;parent >= 0;parent--){
            shiftdown(parent,usedSize);
        }
    }
    //parent:表示每棵子树的节点
    //len表示没课字数调整的结束位置,不能大于len
    public void shiftdown(int parent,int len){
        int  child= parent * 2 +1;
        //孩子的下标必须小于有效长度,保证有左孩子
        while(child < len){
            //child+1 < len保证有右孩子
            if(child+1 < len && elem[child] < elem[child +1]){
                child ++;
            }
            if(elem[child]>elem[parent]){
                int tmp = elem[child];
                elem[child] = elem[parent];
                elem[parent] = tmp;
                parent = child;
                child = 2 *parent + 1;
            }else{
                break;
            }
        }
    }
}
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

这是大堆的情况下。那么我们如何改为小根堆呢?很简单,只需要改变两个符号。

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

2.3建堆的复杂度

在这里插入图片描述
在这里插入图片描述

综上:建堆的时间复杂度为O(n)

2.4堆的插入和删除

想要向堆中插入元素,我们可以先插入到最后一个位置上。在对其进行大根堆调整。

代码语言:javascript
复制
public void offer(int val){
        //如果满了就扩容
        if(isFull()){
            elem = Arrays.copyOf(this.elem,2*this.elem.length);
        }
        elem[usedSize] = val;
        usedSize++;
        //调整为大根堆
        shiftup(usedSize-1);
    }
    public void shiftup(int child){
        int parent = (child-1)/2;
        while(parent >= 0){
            if(elem[child] > elem[parent]){
                int tmp = elem[child];
                elem[child] = elem[parent];
                elem[parent] = tmp;
                child = parent;
                parent = (child -1)/2;
            }else{
                break;
            }
        }
    }
    public boolean isFull(){
        return usedSize == elem.length;
    }
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

其次,我们来看堆的删除: 堆的删除一定删除的是堆顶元素。(删除中间元素无意义)

在这里插入图片描述
在这里插入图片描述
代码语言:javascript
复制
public int pop(){
        if(isEmpty()){
            throw new isEmptyExpection("堆空异常!");
        }
        int tmp = elem[0];
        elem[0] = elem[usedSize-1];
        elem[usedSize-1] = tmp;
        usedSize--;
        shiftdown(0,usedSize);
        return tmp;
    }
    public boolean isEmpty(){
        return usedSize == 0;
    }

3.常用接口介绍

Java集合框架中提供了PriorityQueue和PriorityBlockingQueue两种类型的优先级队列,PriorityQueue是线程不安全的,PriorityBlockingQueue是线程安全的。

在这里插入图片描述
在这里插入图片描述

关于PriorityQueue的使用要注意:

  1. 使用时必须导入PriorityQueue所在的包,即:
  2. PriorityQueue中放置的元素必须要能够比较大小,不能插入无法比较大小的对象,否则会抛出ClassCastException异常
  3. 不能插入null对象,否则会抛出NullPointerException
  4. 没有容量限制,可以插入任意多个元素,其内部可以自动扩容
  5. 插入和删除元素的时间复杂度为
  6. PriorityQueue底层使用了堆数据结构
  7. PriorityQueue默认情况下是小堆—即每次获取到的元素都是最小的元素 import java.util.PriorityQueue;
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

默认情况下,PriorityQueue队列是小堆,如果需要大堆需要用户提供比较器;

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

优先级队列的扩容说明:

  1. 如果容量小于64时,是按照oldCapacity的2倍方式扩容的
  2. 如果容量大于等于64,是按照oldCapacity的1.5倍方式扩容的
  3. 如果容量超过MAX_ARRAY_SIZE,按照MAX_ARRAY_SIZE来进行扩容

4.Top-K问题

题目oj:求数据集合中前k个最大的元素或者最小的元素。 思路一:采用快排,然后写一个循环输出前k项最大/最小的值

代码语言:javascript
复制
public int[] smalllestK(int[] arr,int k){
        Arrays.sort(arr);
        int[] tmp = new int[k];
        for(int i = 0; i< k;i++){
            tmp[i] = arr[i];
        }
        return tmp;
    }

思路二:建一个小根堆,取出数组当中的每个元素,存放到小根堆当中,弹出k个元素,存放到数组当中,返回即可。

代码语言:javascript
复制
public int[] smallestK1(int[] arr,int k){
        //1.建一个堆
        PriorityQueue<Integer> minHeap = new PriorityQueue<>();
        //2.取出数组中的每个元素,存放到小根堆中
        for(int i = 0;i < k;i++){
            minHeap.offer(arr[i]);
        }
        //3.弹出k个元素,存放到数组当中。
        int[] tmp = new int[k];
        for(int i = 0;i < k;i++){
            tmp[i] = arr[i];
        }
        return tmp;
    }

思路三:

Top-K问题:求数据集合中前k个最大的元素或者最小的元素,一般情况下数据量都比较大。

本文参与?腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2023-01-17,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客?前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与?腾讯云自媒体分享计划? ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 1.优先级队列
    • 1.1概念
    • 2.优先级队列的模拟实现
      • 2.1堆的存储方式
        • 2.2堆的创建
          • 2.3建堆的复杂度
            • 2.4堆的插入和删除
            • 3.常用接口介绍
            • 4.Top-K问题
            相关产品与服务
            对象存储
            对象存储(Cloud Object Storage,COS)是由腾讯云推出的无目录层次结构、无数据格式限制,可容纳海量数据且支持 HTTP/HTTPS 协议访问的分布式存储服务。腾讯云 COS 的存储桶空间无容量上限,无需分区管理,适用于 CDN 数据分发、数据万象处理或大数据计算与分析的数据湖等多种场景。
            领券
            问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档
            http://www.vxiaotou.com