堆的逻辑结构是一颗完全二叉树;物理结构是一个数组。
void AdjustDown1(int* a, int root, int n)//让根形成左右两个大堆(大堆是指结点为子树的所有结点的最大值)(小堆则为最小值)
{
int parent = root;
int child = parent * 2 + 1;
while (child < n)
{
if (child + 1 < n && a[child + 1] > a[child])//排除只有一个左孩子的情况和左孩子大于有孩子
{
child += 1;
}
if (a[child] > a[parent])
{
Swap(&a[child], &a[parent]);
parent = child;
child = parent * 2 + 1;
}
else
{
break;
}
}
}
void HeapSort(int* a, int n)
{
for (int i = (n - 1 - 1) / 2;i >= 0;i--)//生成堆,从无左右子树的结点开始生成大堆
{
AdjustDown1(a, i, n);
}
int end = n - 1;//数组最后一个元素
while (end > 0)
{
Swap(&a[0], &a[end]);//交换第一个和最后一个元素的值
AdjustDown1(a, 0, end);//以最大结点为根
--end;
}
}
int main()
{
int a[] = { 3,5,2,7,8,6,1,9,4,0 };
HeapSort(a, sizeof(a) / sizeof(a[0]));
for (int i = 0;i < sizeof(a) / sizeof(a[0]);i++)
{
cout << a[i] << endl;
}
}