#include <iostream>
using namespace std;
void AdjustHeap(int arr[], int i, int length);
void swap(int arr[], int a, int b);
void HeapSort(int arr[], int length);
int main()
{
int array[10] = {5,2,1,5,3,6,8,9,1,6};
int length = sizeof(array)/sizeof(array[0]);
HeapSort(array, length);
for(int i = 0;i < 10;i++)
{
cout << array[i] << " ";
}
cout << endl;
}
void HeapSort(int arr[], int length)
{
// 根节点下标为0,则最后一个非叶子节点的坐标是(n-1)/2
for(int i = length / 2 - 1; i >= 0; i--)
{
//从最后一个非叶子节点开始调整
AdjustHeap(arr, i, length);
}
for(int j = length - 1; j > 0; j--)
{
//将根节点和未排序的最后一个节点交换
swap(arr, 0, j);
//交换后可能会破坏堆结构,需重新调整堆
AdjustHeap(arr, 0, j);
}
}
void AdjustHeap(int arr[], int i, int length)
{
int temp = arr[i];
//2*i+1为节点的左孩子下标,循环,每次得到左孩子的下标,右孩子下标是2*k+2
for(int k = 2 * i + 1; k < length; k = 2 * k + 1)
{
//比较左右孩子节点的大小
if(k + 1 < length && arr[k] < arr[k + 1])
k++;
//将上面的较大者和根节点比较,交换
if(arr[k] > temp)
{
arr[i] = arr[k];
获得交换者(左右节点较大者)的下标
i = k;
}
else
break;
}
arr[i] = temp;
}
void swap(int arr[], int a, int b)
{
int temp = arr[a];
arr[a] = arr[b];
arr[b] = temp;
}