【排序算法】之堆排序的实现

#include<iostream>
using namespace std;

/*
1：找到第一个非叶子节点的位置cur：数组元素个数size/2-1
2：从cur开始，依次将各子树调整为最大堆--->整体成为最大堆
3：堆顶元素与数组最后一个元素交换
4：数组元素减1，重复执行上述操作
*/
void AdjustUp(int *arr, int size)
{
int cur = size / 2 - 1;//指向第一个非叶子节点
while (cur >= 0)
{
int left = cur * 2 + 1;//该节点的左孩子
int max = left;
int right = left + 1;  //该节点的右孩子
//求出左右孩子中较大的节点
if (right < size&&arr[right] > arr[max])
{
max = right;
}
//左右孩子中较大的节点，与根节点比较，根节存放较大值
if (arr[max] > arr[cur])
{
swap(arr[max], arr[cur]);
}
//为下一颗树调整做准备
cur--;
}
}
void HeapSort(int *arr, int size)
{
for (int i = size; i > 1; --i)
{
AdjustUp(arr, i);//调整为最大堆
swap(arr[0], arr[i - 1]);
//1:将堆顶元素与数组最后一个元素交换
//2:然后堆元素个数减1，重新调整为最大堆,重复1操作
}
}
void Print(int *arr, int size)
{
for (int i = 0; i < size; ++i)
{
cout << arr[i] << " ";
}
cout << endl;
}

int main()
{
int arr[] = { 2, 9, 7, 15, 8, 21, 12, 100, 99 };
int size = sizeof(arr) / sizeof(arr[0]);

HeapSort(arr, size);
Print(arr, size);
system("pause");
return 0;
}