优化堆排序(Java 实例代码)

目录

优化堆排序

[Java 实例代码](#Java 实例代码)

[src/runoob/heap/HeapSort.java 文件代码:](#src/runoob/heap/HeapSort.java 文件代码:)


优化堆排序

上一节的堆排序,我们开辟了额外的空间进行构造堆和对堆进行排序。这一小节,我们进行优化,使用原地堆排序。

对于一个最大堆,首先将开始位置数据和数组末尾数值进行交换,那么数组末尾就是最大元素,然后再对W元素进行 shift down 操作,重新生成最大堆,然后将新生成的最大数和整个数组倒数第二位置进行交换,此时倒数第二位置就是倒数第二大数据,这个过程以此类推。

整个过程可以用如下图表示:

Java 实例代码

源码包下载: Downloadhttps://www.runoob.com/wp-content/uploads/2020/09/runoob-algorithm-HeapSort.zip

src/runoob/heap/HeapSort.java 文件代码:

package runoob.heap;

import runoob.sort.SortTestHelper;

/**

* 原地堆排序

*/

public class HeapSort<T extends Comparable> {

public static void sort(Comparable[] arr) {

int n = arr.length;

// 注意,此时我们的堆是从0开始索引的

// 从(最后一个元素的索引-1)/2开始

// 最后一个元素的索引 = n-1

for (int i = (n - 1 - 1) / 2; i >= 0; i--)

shiftDown(arr, n, i);

for (int i = n - 1; i > 0; i--) {

swap(arr, 0, i);

shiftDown(arr, i, 0);

}

}

// 交换堆中索引为i和j的两个元素

private static void swap(Object[] arr, int i, int j) {

Object t = arr[i];

arr[i] = arr[j];

arr[j] = t;

}

// 原始的shiftDown过程

private static void shiftDown(Comparable[] arr, int n, int k) {

while (2 * k + 1 < n) {

//左孩子节点

int j = 2 * k + 1;

//右孩子节点比左孩子节点大

if (j + 1 < n && arr[j + 1].compareTo(arr[j]) > 0)

j += 1;

//比两孩子节点都大

if (arr[k].compareTo(arr[j]) >= 0) break;

//交换原节点和孩子节点的值

swap(arr, k, j);

k = j;

}

}

// 测试 HeapSort

public static void main(String[] args) {

int N = 100;

Integer[] arr = SortTestHelper.generateRandomArray(N, 0, 100000);

sort(arr);

// 将heapify中的数据逐渐使用extractMax取出来

// 取出来的顺序应该是按照从大到小的顺序取出来的

for (int i = 0; i < N; i++) {

System.out.print(arr[i] + " ");

}

// 确保arr数组是从大到小排列的

for (int i = 1; i < N; i++)

assert arr[i - 1] >= arr[i];

}

}

相关推荐
雷神乐乐7 分钟前
File.separator与File.separatorChar的区别
java·路径分隔符
小刘|11 分钟前
《Java 实现希尔排序:原理剖析与代码详解》
java·算法·排序算法
jjyangyou16 分钟前
物联网核心安全系列——物联网安全需求
物联网·算法·安全·嵌入式·产品经理·硬件·产品设计
逊嘘30 分钟前
【Java语言】抽象类与接口
java·开发语言·jvm
van叶~33 分钟前
算法妙妙屋-------1.递归的深邃回响:二叉树的奇妙剪枝
c++·算法
简简单单做算法33 分钟前
基于Retinex算法的图像去雾matlab仿真
算法·matlab·图像去雾·retinex
morris13137 分钟前
【SpringBoot】Xss的常见攻击方式与防御手段
java·spring boot·xss·csp
云卓SKYDROID1 小时前
除草机器人算法以及技术详解!
算法·机器人·科普·高科技·云卓科技·算法技术
七星静香1 小时前
laravel chunkById 分块查询 使用时的问题
java·前端·laravel
Jacob程序员1 小时前
java导出word文件(手绘)
java·开发语言·word