05堆排序

1. 基本概念

什么是堆？

1. 是一棵完全二叉树
2. 所有父节点的值大于子节点的值

• 堆调整的heapify操作
• 当前已是一棵完全二叉树，第i个节点需要调整，以i为父节点，找到对应的子节点c1(左子节点)，c2(右子节点)，将c1和c2中最大值与父节点的值交换，依次递归下去。
• 直到i的值大于等于n（n为整个堆的节点数）

堆因为是完全二叉树，所有节点的值可以用数组来保存，假设节点个数为n，用数组int arr[n]保存，从上到下，从左到右，依次保存在数组中。
下标的对应关系为：
parent=(i - 1)/2
c1 = 2i + 1
c2 = 2i + 2

如果整个树都是乱序的，如何构建堆？
从h-1层开始，对每个节点做heapify操作，直到头节点

如何进行堆排序

堆排序（升序）：

1. 先建堆（大根堆）
2. 然后每次交换堆顶元素和最后一个元素，并切断最后一个元素（最大），再进行heapify操作

最好、最坏、平均时间复杂度都是O(nlog(n))
空间复杂度是O(1)

2. 代码实现

#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <string>
#include <vector>
#include <set>
#include <map>
#include <unordered_set>
#include <unordered_map>
#include <list>
#include <algorithm>
#include <sstream>

using namespace std;

/* 交换数组中两个元素的值 */
void swap(int arr[], int i, int j){
	int tmp = arr[i];
	arr[i] = arr[j];
	arr[j] = tmp;
}

/* 从第i个元素递归进行heapify操作 */
void heapify(int tree[], int n, int i){
	if (i >= n) return;

	int c1 = 2 * i + 1; // 左孩子
	int c2 = 2 * i + 2; // 右孩子

	int max_idx = i;
	if (c1 < n && tree[c1] > tree[max_idx]){
		max_idx = c1;
	}

	if (c2 < n && tree[c2] > tree[max_idx]){
		max_idx = c2;
	}

	if (max_idx != i){
		swap(tree, max_idx, i);
		heapify(tree, n, max_idx);
	}
}

/* 建堆 */
void buid_heap(int tree[], int n){
	int last_node = n - 1;
	int parent = (last_node - 1) / 2;

	for (int i = parent; i >= 0; --i){
		heapify(tree, n, i);
	}
}

/* 
堆排序（升序）：
1.先建堆（大根堆）
2.然后每次交换堆顶元素和最后一个元素，并切断最后一个元素（最大）
*/
void heap_sort(int tree[], int n){
	buid_heap(tree, n);

	for (int i = n - 1; i >= 0; --i){
		swap(tree, 0, i);
		heapify(tree, i, 0);
	}
}

void test01(){
	int tree[] = { 4, 10, 3, 5, 1, 2 };
	int n = sizeof(tree) / sizeof(tree[0]);

	heap_sort(tree, n);
	for (int i = 0; i < n; ++i){
		cout << tree[i] << " ";
	}
	cout << endl;
}
int main()
{
	test01();
	system("pause");
	return 0;
}

3. 参考

强烈推荐下面视频，讲得很清楚！
https://www.bilibili.com/video/BV1Eb41147dK?t=1544