1. 基本概念
什么是堆?
- 是一棵完全二叉树
- 所有父节点的值大于子节点的值
- 堆调整的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操作,直到头节点
如何进行堆排序
堆排序(升序):
- 先建堆(大根堆)
- 然后每次交换堆顶元素和最后一个元素,并切断最后一个元素(最大),再进行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