学习堆排序之前,先简单介绍一下什么是堆。
但是要了解堆,还得先连接二叉树。
二叉树,就是每个节点最多有两个子节点的树形的数据结构。二叉树又有完全二叉树、满二叉树等等,细的就不讲了,可以搜一下资料,还是很多讲这个的。
二叉堆,它是属于完全二叉树,同时它又满足堆的性质(子节点的值总是小于它的父节点(也可以是大于))。也就是说,根节点肯定是最大(也可能是最小)的那个元素。
堆排序就是根据这个特点,把数组看成是一个二叉堆,每次计算出一个最大或最小值然后进行排序的。
那么完全二叉树它怎么跟数组关联上呢?其实完全二叉树它有一个特点,就是它完全可以用数组来表示。详细的这里就不讲了,但是有几点需要记住。
知道这三点之后,我们就可以根据完全二叉树的性质,把它编程一个二叉堆,然后进行排序了。
关于满二叉树和完全二叉树我还是简单画个图吧,帮助理解。
这种就叫满二叉树,左右完全平衡,非叶子节点都有两个子节点,叶子节点都在一层。
完全二叉树就是说,满二叉树的基础上,比如说要加一个节点,那么只能先在D下面加,然后依次往后。要删除一个节点,只能从右下角开始,先把G删掉,再删F,再删E这样。
OK二叉树讲的有点多了,下面进入整体,开始将堆排序。
依然是有这样一个数组:
那么我们先把它看成一个完全二叉树,开始把它构建成一个最大堆,第一步,找到最后一个非叶子节点,用上面的公式,数组长度为8,最后一个非叶子节点就是 8 ÷ 2 - 1 = 3,那么最后一个非叶子节点就找到了。最后一个非叶子节点前面的元素,都是非叶子节点,我们用蓝色标记出来:
然后遍历 14,14 的两个子节点下标为 2 × 2 + 1 = 5,2 × 2 + 2 = 6,所以 14 需要与 324 和 1 进行对比,14 比 324 小,交换位置:
这样最大的数就跑到了数组的末尾,第一轮的排序也就结束了。
那么第二轮,我们将已排好序的大数排除到构建最大堆的范围之外,只用前面7个数去构件最大堆,构建完成之后,再把最大数挪到下标6的位置,以此类推,即可完成整个数组的排序。
@Test
public void heapSort() {
int[] nums = new int[]{26, -3, 14, -15, 0, 324, 1, 22};
sort(nums);
System.out.println(Arrays.toString(nums));
}
private void sort(int[] nums) {
// 如果数组长度小于2,直接返回
if (nums.length < 2) {
return;
}
// 遍历
for (int i = 0; i < nums.length; i++) {
// 构建最大堆,nums.length - i 是为了排除掉以排好序的元素
int lastParentNode = (nums.length - i) / 2 - 1;
// 遍历所有非叶子节点
for (int parentNode = lastParentNode; parentNode >= 0; parentNode--) {
// 如果该节点小于它的左子节点,那么交换位置
if (nums[parentNode] < nums[parentNode * 2 + 1]) {
swap(nums, parentNode, parentNode * 2 + 1);
}
// 如果该节点存在右子节点,并且该节点小于它的右子节点,那么交换位置
if (parentNode * 2 + 2 < nums.length - i && nums[parentNode] < nums[parentNode * 2 + 2]) {
swap(nums, parentNode, parentNode * 2 + 2);
}
}
// 构建最大堆完成,把最大的数与堆的最后一个元素交换位置
swap(nums, 0, nums.length - i - 1);
}
}
private void swap(int[] nums, int i, int j) {
int temp = nums[i];
nums[i] = nums[j];
nums[j] = temp;
}