整体思路:把一个序列调整为大根堆的形式,然后每次取堆顶元素(把堆顶元素与序列最后一个元素互换),再对新的堆顶进行调整,使满足大根堆的条件,继续取堆顶元素,重复上述操作,直到每个元素都被取出,可以得到一个有序的序列。
一、建立大根堆
1.初始数据
初始序列为
在计算时将序列的长度记为8,下标为0处用来存放临时数据,有效数据初始下标为1
对应的完全二叉树为
2.完全二叉树有关知识点
完全二叉树非叶结点下标<=长度/2,这里的长度为8,即下标<=4
完全二叉树结点(下标为i)的左子树和右子树(如果存在)的下标分别为2*i和2*i+1
3.建立过程
从最后一个非叶结点开始,把较大的值往上提
二、代码实现大根堆的建立和调整
代码思路:把建立和调整分为两个函数完成,BuildMaxHeap函数完成大根堆的建立,HeadAdjust函数完成对应下标的调整(调整为大根堆),在BuildMaxHeap函数的for循环中调用HeadAdjust函数,完成每一个非叶结点的调整。
注:一个结点的调整可能会影响到其他结点
这种情况下如果让53和87交换,也会影响下层,所以要继续进行判断
1.代码实现
//建立大根堆
void BuildMaxHeap(int num[], int length) {
for (int i = length / 2; i > 0; i--) {
HeadAdjust(num, i, length);
}
}
// 调整大根堆
void HeadAdjust(int num[], int i, int length) {
num[0] = num[i];//把每个要检查的结点放到第一个
//依次检查每一个非叶节点是否符合
for (int j = 2 * i; j <= length; j *= 2) {
if (j < length && num[j] < num[j + 1]) j++;//如果右子树存在并且左子树的值小于右子树,指向j+1
if (num[0] < num[j]) {//如果不符合大根堆的要求,把结点的值互换
num[i] = num[j];
i = j;//对向上提的结点处重新判断(空结点)
}
else break;//退出本结点的
}
num[i] = num[0];
}
2.代码解析
建立大根堆本质上就是对各个非叶结点进行调整。
以本序列为例,i的初始值为8/2=4,从下标为4的结点检测到根节点
①首先把下标为4的结点值放入到下标为0处
②然后检查该结点值是否符合大根堆要求
进入HeadAdjust函数,j的初始值为i的左子树的下标,即i*2
如果该结点没有右子树j不变,仍指向32
if (j < length && num[j] < num[j + 1]) j++;//使j指向值最大的结点
如果有右子树则进入if判断,如果大于右子树,则j++,即指向右子树,反之j不变
这个时候j指向的是子结点中的最大值,再和父节点进行比较
如果符合大根堆条件就退出本次循环,即完成此结点的调整,开始下一个结点
如果不符合条件,先把j指向结点的值放到父结点中
然后进行下一轮循环
此时j指向空白结点的左子树
j*2=16,超过范围,所以本结点调整完成
把序列下标为0的数据放入
其他结点类似
三、基于大根堆进行排序
把堆顶元素与最后一个元素互换
此时87的位置已经固定,可以看作序列长度减1
继续对二叉树进行调整,得到新的大根堆
然后再取出最大值,重复上述操作,直到length为0
可以得到从小到大的序列
四、堆排序代码实现
void HeapSort(int num[], int length) {
BuildMaxHeap(num, length);
for (int i = length; i >=1; i--) {
//交换首尾元素
int temp = num[1];
num[1] = num[i];
num[i] = temp;
HeadAdjust(num, 1,i-1);
}
}
对长度为8的序列进行遍历,每次遍历都把堆顶元素和序列最后一个元素进行交换,然后让序列的长度减1,再对序列进行调整,直到减为0就可以得到从小到大排序的序列。
五、完整代码和运行结果
//建立大根堆
void BuildMaxHeap(int num[], int length) {
for (int i = length / 2; i > 0; i--) {
HeadAdjust(num, i, length);
}
}
// 调整大根堆
void HeadAdjust(int num[], int i, int length) {
num[0] = num[i];//把每个要检查的结点放到第一个
//依次检查每一个非叶节点是否符合
for (int j = 2 * i; j <= length; j *= 2) {
if (j < length && num[j] < num[j + 1]) j++;//如果右子树存在并且左子树的值小于右子树,指向j+1
if (num[0] < num[j]) {//如果不符合大根堆的要求,把结点的值互换
num[i] = num[j];
i = j;//对向上提的结点处重新判断(空结点)
}
else break;//退出本结点的
}
num[i] = num[0];
}
//基于大根堆进行排序
void HeapSort(int num[], int length) {
BuildMaxHeap(num, length);
for (int i = length; i >=1; i--) {
//交换首尾元素
int temp = num[1];
num[1] = num[i];
num[i] = temp;
HeadAdjust(num, 1,i-1);
}
}
int main() {
int num[9] = { 0,53,17,78,9,45,65,87,32 };
HeapSort(num, 8);
for (int i = 1; i < 9; i++) {
printf("%d\n", num[i]);
}
}