【注】: 此处没有设置哨兵
#pragma once
#include <iostream>
#include <vector>
using namespace std;
// 堆排序步骤:
/*
1. 把数组表示成完全二叉树......画出图(草纸上)
2. 把完全二叉树调成最大堆或者最小堆
3. 取根节点和最后一个叶节点 交换,删除根节点,重新调整为最大堆或者最小堆
*/
// 代码关键
/*
1. 调整堆,从最后一个非叶节点开始: 下标为: 数组长度/2 -1
2. 父节点的两个孩子节点的下表为: 2i+1; 2i+2;
3. HeapAdjust 自上而下(原因是,祖宗影响子孙的堆)
4. HeapSort中的建立堆: 自下而上,从“最后一个非叶节点”开始
5. 注意处理子数组的问题
*/
void HeapAdjust(int arr[], int start, int end);
void HeapSort(int arr[], int N);
void Swap(int *a, int *b);
// 调整数组和子数组为最大/小堆
// start end 代表 可访问到的 子数组的 开始和结尾(使用数组的话,结尾一直是len-1)
// 建立堆---是从最后一个非叶子结点到第一个
// 调整堆---是从当前节点到最后
void HeapAdjust(int arr[], int start, int end)
{
int parent = start;
int max_leaf = 2 * start + 1;
while (max_leaf <= end)
{
// 1. 找到最大的叶子节点
// 取等于,是因为end = len - 1;
if (max_leaf + 1 <= end && arr[max_leaf + 1] > arr[max_leaf])
max_leaf = max_leaf + 1;
// 2. 把最大叶节点和根节点交换
if (arr[max_leaf] > arr[parent])
{
Swap(&arr[max_leaf], &arr[parent]);
// 在调整远古的父节点时,非常可能打乱 子孙的最大堆情况
// 继续调整,打乱了什么,调整什么;
// 即: 谁惹得事,处理谁。
parent = max_leaf;
max_leaf = 2 * parent + 1;
}
else
return;
}
}
void HeapSort(int arr[], int N)
{
// 从最后一个非叶节点开始,依次调用“调整堆”,完成最大堆的构建
// 第一个是 “最后一个非叶节点”
for (int i = N/2-1; i >= 0; i--)
{
// 调整数组和子数组为最大/小堆
HeapAdjust(arr, i, N - 1);
}
// 第N-1次调整堆
// 堆排序 --- 为什么i 不用等于0, 因为第一次堆排序已经排好,0处已经放着最大值
// 只需要调整位置放在最后,然后缩小数组,继续调整堆,取最大值
for (int i = N - 1; i > 0; i--)
{
// 最大堆后,此时最大值放在arr[0]处, 现在要把他放到最后面
Swap(&arr[0], &arr[i]);
HeapAdjust(arr, 0, i - 1);
}
}
// 只是交换数值,下标并没有交换
void Swap(int *a, int *b)
{
int tmp = *a;
*a = *b;
*b = tmp;
}