IT数码 购物 网址 头条 软件 日历 阅读 图书馆
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
图片批量下载器
↓批量下载图片,美女图库↓
图片自动播放器
↓图片自动播放器↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁
 
   -> 数据结构与算法 -> PriorityQueue源码分析 -> 正文阅读

[数据结构与算法]PriorityQueue源码分析


前言

主要分析offer(E e) 和 poll() 两个方法,即队列的入队和出队,底层原理是堆排序的上浮和下沉


一、offer(E e)

offer(E e)源码:

    public boolean offer(E e) {
        if (e == null)
            throw new NullPointerException();
        modCount++;
        int i = size;
        if (i >= queue.length)
            grow(i + 1);
        size = i + 1;
        if (i == 0)
            queue[0] = e;
        else
            siftUp(i, e);
        return true;
    }

重点看siftUp函数,代码如下
注意下i,i的含义为新增元素预插入位置的索引,其值为队列的size,即元素e预插入位置为队列尾部

    private void siftUp(int k, E x) {
        if (comparator != null)
            siftUpUsingComparator(k, x);
        else
            siftUpComparable(k, x);
    }

siftUpUsingComparator和siftUpComparable逻辑一样
siftUpUsingComparator用自定义比较器来比较元素大小,主要应用于自定义对象
siftUpComparable主要应用于基本数据类型
这里我们主要看下siftUpComparable的实现,源码如下

    private void siftUpComparable(int k, E x) {
        Comparable<? super E> key = (Comparable<? super E>) x;
        while (k > 0) {
            int parent = (k - 1) >>> 1; // 父节点索引。完全二叉树父子几点的索引关系为: 左子节点索引 = 父节点所以 * 2 + 1
            Object e = queue[parent];
            if (key.compareTo((E) e) >= 0)
                break;
            queue[k] = e;
            k = parent;
        }
        queue[k] = key;
    }

代码的逻辑大概是:

  1. 方法入参:k表示新增元素x的预插入位置,k的初始值为队列size,x的初始值为要插入的元素
  2. 比较新增元素x和其父节点queue[parent]的大小。新增元素的预插入位置索引为k
  3. 如果 x >= queue[parent], 由于是最小堆(父节点的值小于子节点),所以预插入位置正确,执行步骤6
  4. 如果 x < queue[parent], 由于是最小堆(父节点的值小于子节点), 所以预插入位置不对,需要调整预插入位置,元素x上浮
  5. 简单的说,父节点元素queue[parent]和x交换位置
  6. x交换位置后,不一定比其新父节点元素大,所以需要再次判断、调整。将预插入位置k调整为parent。如果k = 0 , 说明上浮到根节点了,执行步骤6;则重复步骤1
  7. 找到了合适的插入位置k,直接赋值:queue[k] = x

二、poll()

poll源码:

    public E poll() {
        if (size == 0)
            return null;
        int s = --size;
        modCount++;
        E result = (E) queue[0];
        E x = (E) queue[s];
        queue[s] = null;
        if (s != 0)
            siftDown(0, x);
        return result;
    }

重点关注siftDown

    private void siftDown(int k, E x) {
        if (comparator != null)
            siftDownUsingComparator(k, x);
        else
            siftDownComparable(k, x);
    }

分析siftDownComparable

    private void siftDownComparable(int k, E x) {
        Comparable<? super E> key = (Comparable<? super E>)x;
        int half = size >>> 1;        // loop while a non-leaf
        while (k < half) {
            int child = (k << 1) + 1; // 左子节点索引
            Object c = queue[child];
            int right = child + 1;    // 右子节点索引
            if (right < size &&
                ((Comparable<? super E>) c).compareTo((E) queue[right]) > 0)
                c = queue[child = right];
            if (key.compareTo((E) c) <= 0)
                break;
            queue[k] = c;
            k = child;
        }
        queue[k] = key;
    }

说明:poll函数弹出的是根节点,调整策略是:预设用最后一个节点填补根节点,然后进行下沉调整
代码的逻辑大概是:

  1. 方法入参:k表示新增元素x的预插入位置,k的初始值为0,x的初始值为队尾元素
  2. 比较k位置的左右子节点大小。child记录较小节点的索引,c记录较小节点的值
  3. 如果x <= c,由于是最小堆(父节点的值小于子节点),表示预插入位置正确,执行步骤5
  4. 如果x > c,把child位置的值(当前比较的3个元素最小的值)赋值给queue(k)(k为当前3个参与比较元素的根节点),同时更新k的值为child。如果k >= half,说明。。。,执行步骤5;否则执行步骤2
  5. 找到了合适的插入位置k,直接赋值:queue[k] = x

总结

疑问:
  1. siftDownComparable方法中,循环中体条件为什么是 k < half, 麻烦哪位大牛给解析下
  数据结构与算法 最新文章
【力扣106】 从中序与后续遍历序列构造二叉
leetcode 322 零钱兑换
哈希的应用:海量数据处理
动态规划|最短Hamilton路径
华为机试_HJ41 称砝码【中等】【menset】【
【C与数据结构】——寒假提高每日练习Day1
基础算法——堆排序
2023王道数据结构线性表--单链表课后习题部
LeetCode 之 反转链表的一部分
【题解】lintcode必刷50题<有效的括号序列
上一篇文章      下一篇文章      查看所有文章
加:2021-12-14 16:12:59  更:2021-12-14 16:14:31 
 
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁

360图书馆 购物 三丰科技 阅读网 日历 万年历 2024年11日历 -2024/11/26 16:53:32-

图片自动播放器
↓图片自动播放器↓
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
图片批量下载器
↓批量下载图片,美女图库↓
  网站联系: qq:121756557 email:121756557@qq.com  IT数码