package com.haiyang.datastructure.sort;
import java.util.Arrays;
/**
* 堆排序原理:首先将待排序的序列创建为大顶堆,每次取堆顶元素
* @author haiYang
* @create 2022-01-18 14:09
*/
public class HeapSort {
//堆排序
public static void heapSort(int[] arr){
int temp=0;
//将序列初始化为大顶堆
for (int i = arr.length/2-1; i >=0 ; i--) {
createBigPileTop(arr,i,arr.length);
}
//进行堆排序
for (int i = arr.length-1; i >0 ; i--) {
//将大顶堆根节点元素与最后一个元素交换
temp = arr[i];
arr[i] = arr[0];
arr[0] = temp;
//将前i个元素创建大顶堆
createBigPileTop(arr,0,i);
}
System.out.println("堆排序结果为:" + Arrays.toString(arr));
}
/**
* 将序列变成大顶堆
*
* @param arr 待初始化的数组
* @param i 最后一个非叶子节点
* @param length 数组的长度
*/
public static void createBigPileTop(int[] arr, int i, int length) {
int temp = arr[i];// 用于保存待调整的元素
//k为待调整节点的左孩子,k = k * 2 + 1代表要调整的子堆
for (int k = i * 2 + 1; k < length; k = k * 2 + 1) {
//判断左右孩子的大小,取值大的索引
if (k + 1 < length && arr[k] < arr[k + 1]) {
k++;
}
//将待调整元素与左右孩子中较大的元素进行比较,用比其大的元素进行替换
if (temp < arr[k]) {
arr[i] = arr[k];
//将子序列继续调整
i = k;
} else {
//如果不满足上诉要求代表当前子序列已经满足大顶堆
break;
}
}
//将最开始的元素放到满足大顶堆的子序列上
arr[i] = temp;
}
public static void main(String[] args) {
int[] ints = {4, 6, 8, 5, 9};
heapSort(ints);
}
}
数据结构之堆排序(java)
于 2022-01-18 14:57:32 首次发布