堆的概念
堆是一种完全二叉树:每个结点的值都大于或等于其左右孩子结点的值,称为大顶堆;每个结点的值都小于或等于其左右孩子结点的值,成为小顶堆。
利用简单的公式描述堆的定义:
大顶堆:arr[i] >= arr[2i+1] && arr[i] >= arr[2i+2]
小顶堆:arr[i] <= arr[2i+1] && arr[i] <= arr[2i+2]
堆排序
堆排序(Heap Sort)是利用堆进行排序的方法。
基本思想:
将待排序的序列构造成一个大顶堆。此时,整个序列的最大值就是堆顶的根结点。将它移走(其实就是将其与堆数组的末尾元素交换,此时末尾元素就是最大值),然后将剩余的n-1个序列重新构造成
一个堆,这样就会得到n个元素中的次大值。如此反复执行,便能得到一个有序序列了。
动态图演示
堆排序算法
堆排序的运行时间主要消耗在初始构建堆和重建堆时的反复筛选上。
在构建堆的过程中,完全二叉树从最下层最右边的非终端结点开始构建,将它与其孩子进行比较和有必要的交换,对于每一个非终端结点来说,其实最多进行两次比较和交换。
#include <iostream>
using namespace std;
void Display(int arr[], int size)
{
for (int i = 0; i < size; i++)
cout << arr[i] << " ";
cout << endl;
}
void swap(int arr[], int m, int n)
{
int tmp = arr[m];
arr[m] = arr[n];
arr[n] = tmp;
}
void Adjust(int arr[], int i, int n)
{
int parentNode = i; //父节点
int childNode = 2 * i + 1; //左子节点
while (childNode < n)
{
if (arr[childNode] < arr[childNode + 1] && childNode + 1 < n) //判断左右节点哪个大,大的与父节点比较
{
childNode++;
}
if (arr[parentNode] < arr[childNode])
{
swap(arr, parentNode, childNode);
parentNode = childNode;
}
childNode = childNode * 2 + 1;
}
}
void heapSort(int arr[], int n)
{
for (int i = n / 2 - 1; i >= 0; i--)
Adjust(arr, i, n);
Display(arr, n);
cout << endl;
//调整大顶堆
for (int i = n - 1; i > 0; i--)
{
swap(arr, 0, i); //此时顶点值为最大值,顶点值和第i个值交换
Adjust(arr, 0, i); //重新创建大顶堆(递归调用)
Display(arr, n);
}
}
int main()
{
int arr[]{ 31, 24, 56, 87, 96, 11, 58, 66, 82, 45 };
int n = sizeof(arr) / sizeof(arr[0]);
heapSort(arr, n);
system("pause");
return 0;
}
复杂度分析
最好、最坏和平均时间复杂度是O(nlogn);
最差空间复杂度O(n);
不稳定;