七大排序之堆排序白话详解**@七大排序之堆排序 思路和代码详解 Java代码**
关于堆排序首先要了解的是 :堆 分为最大堆(大根堆)和最小堆(小根堆),大根堆是指:根节点值最大的完全二叉树,树包括树的每一棵子树,它的根节点值总是大于它的左右子树节点值,小根堆同理。
例如:大根堆的逻辑结构如图
需要注意的是大根堆的存储结构并不是树,而是数组,如图:大根堆的存储结构:
当我们得到一个数组,第一步先将数组调整成一个堆结构,然后用sitfDown方法调整堆的元素,从最后一个非叶子节点开始(因为叶子节点不存在子树,不用调整),将大的元素值向上调整 ,调整每一棵子树直到成为一个大根堆。
//传入一个数组,i为待调整节点值的索引,length为数组最大长度
siftDown(int[] arr, int i, int length)
此时堆顶元素就是整个数组的最大值,将堆顶元素和最后一个叶子节点值交换,然后sitfDown方法调整最后一个叶子节点值之前的堆结构继续成为最大堆,继续将剩下的节点值调整到堆顶,将堆顶元素和倒数第二个节点交换,依次交换直到堆顶元素。
此时堆的存储结构就是一个升序的数组。这就是一个堆的排序过程,降序同理,将数组调整成为一个小根堆,然后用siftUp(int[] arr,int i,int length)方法将小的元素放在末尾,依次交换元素值,就得到一个降序的数组。
附上数组升序的堆排序代码:
/**
* 堆排序 时间复杂度:nlogn
* 注意边界条件
*
* @param arr
*/
public static void heapSort(int[] arr) {
//先将数组调整为最大堆
//遍历堆,从最后一个非叶子节点开始siftDowm操作
for (int i = (arr.length - 1) / 2; i >= 0; i--) {
siftDown(arr, i, arr.length);
}
//调整完后 将堆顶元素和最后一个叶子节点交换
for (int i = arr.length - 1; i > 0; i--) {
swap(arr, 0, i);
//每次交换完成,调整最后一个叶子节点之前的堆
siftDown(arr, 0, i);
}
}
/**
* 元素下沉操作
*
* @param arr 需要调整的数组
* @param i 当前要下沉的元素索引
* @param length 下沉的终止位置,数组长度
*/
private static void siftDown(int[] arr, int i, int length) {
//迭代条件:从最后一个非叶子节点往前 逐个调整
while (2 * i + 1 < length) {
//节点i 的左孩子
int j = (i << 1) + 1;
//判断 如果有右孩子,判断左右孩子哪个大哪个交换
if (j + 1 < length && arr[j] < arr[j + 1]) {
j = j + 1;
}
//判断根节点和孩子中元素值大的孩子的大小并交换
if (arr[i] > arr[j]) {
break;
} else {
swap(arr, i, j);
i = j;
}
}
}
private static void swap(int[] arr, int i, int j) {
int temp = arr[i];
arr[i] = arr[j];
arr[j] = temp;
}
如果阅读过程有任何问题,欢迎提问和指出错误!