堆排序原理详解与java实现

以前一直听到堆排序这个词，只知道其排序效率很高，可以达到O(nlogn)的时间复杂度，最坏情况也是如此（这点与快速排序不同，快排最坏情况下为O(n²)）。但对其一直保持着一种敬畏的态度，没有去深究他，今天蹦着学习的态度，参考图书馆的书，并用代码实现，在这里对其进行一番总结。

堆(heap)

一开始听到堆这个词，以为是动态内存分配里面的内存区“堆”，但今天才发现其实这两者完全没有关系。
这里的堆是一个表(list)，记为array，并满足：
$$array[k]\ge array[2k+1]且array[k]\ge array[2k+2]$$
为什么有这样的定义呢，其实这与完全二叉树有关。

如上图，当左边的完全二叉树按照红色的序号存储到一个数组中，就是一个堆。不难发现，在该树中，标号为k的节点的左子节点标号为2k+1，右子节点为2k+2，所以其实堆其实就是指其所对应的逻辑结构——完全二叉树——的任一父节点都大于等于其子节点。这样的堆叫大根堆（或大顶堆），类似地，右边的堆叫小根堆，其任一父节点都小于等于其子节点。我们这里只讨论大根堆，搞定了大根堆，小根堆其实也是差不多的。

堆排序

堆排序就是基于堆这一数据结构的一种排序方法，属于选择排序的一种。

堆排序有以下几个步骤：
(1) 将待排序的序列构建成一个堆
(2) 此时堆顶一定是最大的数，将其与序列的未排序部分的最后一个值交换位置（序列分成两部分，前半部分是未排序的，后半部分是排好序的）
(3) 此时树根的值可能不符合堆的定义，需要将其调整为堆，方法是不断与较大的子节点交换位置，直到满足定义位置
(4) 重复(2)(3)直到排好序为止

例子图解点这里

代码实现

public static void heapSort(int[] arr) {
	int lastUnsorted; // 记录最后一个未排序的位置
	buildHeap(arr);
	for (lastUnsorted = arr.length - 1; lastUnsorted > 0; lastUnsorted--) {
		// 交换最后一个未排序的值与堆顶的值
		int temp = arr[lastUnsorted];
		arr[lastUnsorted] = arr[0];
		arr[0] = temp;
		heapAdjust(arr, 0, lastUnsorted - 1);
	}
	
}

/*
 * 建立大根堆
 */
public static void buildHeap(int[] arr) {
	// 从最后一个非叶子节点开始，将其与其子树调整成大根堆
	for (int i = arr.length / 2 - 1; i >= 0; i--) {
		heapAdjust(arr, i, arr.length - 1);
	}
}

/*
 * arr[low+1...high]满足大根堆的定义
 * 调整arr使arr[low...high]成为大根堆
 */
public static void heapAdjust(int[] arr, int low, int high) {
	int temp = arr[low];
	// 使用i记录左右子节点较大值的下标
	for(int i = 2 * low + 1; i <= high; i = i * 2 + 1) {
		if (i < high && arr[i] < arr[i + 1]) { // 若右子节点更大，则i++
			i++;
		}
		if (temp >= arr[i]) { // 完成
			break;
		} else {
			arr[low] = arr[i];
			low = i;
		}
	}
	arr[low] = temp;
}

无注释简短版

public class HeapSort {
	
	public static void main(String[] args) {
		int arr[] = {6, 5, 3, 1, 8, 7, 9, 0, 6, 5, 3, 1, 8, 7, 10, 20};
		for(int i = arr.length / 2 - 1; i >= 0; i--) {
			heapAdjust(arr, i, arr.length - 1);
		}
		for(int i = arr.length - 1; i > 0; i--) {
			int temp = arr[i];
			arr[i] = arr[0];
			arr[0] = temp;
			heapAdjust(arr, 0, i - 1);
		}
		for (int i = 0; i < arr.length; i++) {
			System.out.print(arr[i] + ",");
		}
	}
	
	public static void heapAdjust(int[] arr, int low, int high) {
		int temp = arr[low];
		for(int i = low * 2 + 1; i <= high; i = i * 2 + 1) {
			if(i < high && arr[i + 1] > arr[i]) {
				i++;
			}
			if(temp > arr[i]) break;
			arr[low] = arr[i];
			low = i;
		}
		arr[low] = temp;
	}
}

补充

构建一个堆：
从最后一个非叶子节点，从右往左，从下往上，调整，使得以该节点作为根节点的子树满足堆的定义。
向堆插入数据：
插入到最后一个位置，然后往上面调整。
删除堆顶元素：
将堆的最后一个元素放到堆顶，并往下调整。