堆排序思想及C++实现

图片及代码来自此处

1. 思想

$step1.$ 将无序数组构造成一个大顶堆
$step2.$ 将堆顶数字与堆最后一个元素交换位置，然后将现在的最后一个元素（原堆顶元素，即最大值）与堆断开
$step3.$ 剩下的数字维护大顶堆的性质，重复step 2，直到取完堆中的数字，最终得到一个从小到大排序的序列

时间复杂度O(nlogn)，其中建堆复杂度O(n)，heapify() 复杂度O(logn)，对n个数进行heapify()。$不稳定$。最坏情况O(n²)，最好情况O(n)

注意到堆是一个完全二叉树

2. 代码

// 堆排序入口
void heap_sort(vector<int>& arr, int n){
	int i;
	// 建堆（以大顶堆为例）
	for(int i = n / 2 - 1; i >= 0; --i)  // 从arr[n / 2 - 1](即最后一个节点的父节点)开始维护堆
		heapify(arr, n, i);
	// 排序
	for(int i = n - 1; i > 0; --i){
		swap(arr[i], arr[0]);
		heapify(arr, i, 0);	 // 保持剩余元素满足最大堆的性质，剩余元素有i个，从堆顶开始维护性质，故第三个参数是0
	}
}

/*
* 维护堆的性质
* @param arr 存储堆的数组
* @param n 数组长度
* @param i 待维护的节点下标
*/
void heapify(vector<int>& arr, int n, int i){
	int largest = i;
	int left = i * 2 + 1, right = i * 2 + 2;
	
	if(left < n && arr[largest] < arr[left])
		largest = left;
	if(right < n && arr[largest] < arr[right])
		largest = right;
	if(largest != i){
		swap(arr[largest], arr[i]);  // 将最大的值换到父节点上
		heapify(arr, n, largest);  // swap()可能破坏堆的性质，需要递归调用heapify()来维护剩余堆的性质
	}	
}

int main(){
	int arr[] = {2, 5, 7, 1, 9, 16, 10, 4};
	heap_sort(arr, arr.size());
	for(int num : arr)
		cout << num << " " << endl;
	return 0;
}