void AdjustUp(HPDataType* arr,int child)
{
int parent = (child - 1) / 2;
while (child > 0)//不需要等于,child只要走到根节点的位置,根节点没有父节点不需要交换
{
//建大堆,>
//建小堆,<
if (arr[child] < arr[parent])
{
Swap(&arr[parent], &arr[child]);
child = parent;
parent = (child - 1) / 2;
}
else
{
break;
}
}
}
void AdjustDown(HPDataType* arr, int parent, int n)
{
int child = parent * 2 + 1;//左孩子
//while (parent < n)
while (child < n)
{
//小堆:找左右孩子中找最小的
//大堆:找左右孩子中找大的
if (child + 1 < n && arr[child] > arr[child + 1])
{
child++;
}
if (arr[child] < arr[parent])
{
Swap(&arr[child], &arr[parent]);
parent = child;
child = parent * 2 + 1;
}
else
{
break;
}
}
}
排降序-----小堆
堆顶跟堆尾交换,最后一位size变得最小10,不用处理了
然后size--,堆顶向下调整,不断交换,直到排成新的小堆
新的小堆的堆顶与堆尾交换,sizew位置又是小堆最小的13;
以此类推,不断循环,小的往后排,形成降序
void HeapSort(int* arr, int n)
{
//建堆
//升序---大堆
//降序----小堆
//向上调整算法建堆
for (int i = 0; i < n; i++)
{
AdjustUp(arr, i);
}
//向下调整算法建堆
//for (int i = (n-1-1)/2; i >= 0; i--)
//{
// AdjustDown(arr, i , n);
//}
//循环将堆顶数据跟最后位置(会变化,每次减少一个数据)的数据进行交换
int end = n - 1;
while (end > 0)
{
Swap(&arr[0], &arr[end]);
AdjustDown(arr, 0, end);
end--;
}
}
排升序-----大堆
堆顶跟堆尾交换,最后一位size变得最大20,不用处理了
然后size--,堆顶向下调整,不断交换,直到排成新的大堆
新的大堆的堆顶与堆尾交换,sizew位置又是大堆最大的19;
以此类推,不断循环,大的往后排,形成降序
void HeapSort(int* arr, int n)
{
//建堆
//升序---大堆
//降序----小堆
//向上调整算法建堆
for (int i = 0; i < n; i++)
{
AdjustUp(arr, i);
}
//向下调整算法建堆
//for (int i = (n-1-1)/2; i >= 0; i--)
//{
// AdjustDown(arr, i , n);
//}
//循环将堆顶数据跟最后位置(会变化,每次减少一个数据)的数据进行交换
int end = n - 1;
while (end > 0)
{
Swap(&arr[0], &arr[end]);
AdjustDown(arr, 0, end);
end--;
}
}