在写堆排序之前,有几个注意的点要提前说明一下
大顶堆:父节点的值大于孩子节点
小顶堆:父节点的值小于孩子节点
二叉树:
根据这张图:这就是一个简单的二叉树,其中100是父节点,88是左孩子节点,66是右孩子节点
叶子节点:处于最底层的数据,由他们不能再出现新的分叉,便称之为叶子节点
根节点:是最顶部的根,也就是最顶部(数组第一个元素)
初始时,我们先将数组臆想为二叉树
第二步:
第三步:
了解了大概的思路,就可以开始进行编写代码啦
当然在开始写代码之前,还有一个小小的工作没有完成,任何确定父子节点的下标呢
下来我们经过小小的数学运算:
好啦,现在可以正式进入我们的写代码环节啦
我们先来写它的主体部分
根据之前的步骤,我们首先要做的是找到最后一个非叶子节点
怎么去找呢?我们不知道最后一个非叶子节点是哪个,但是我们知道最后一个元素的下标哇,所以我们用推父节点的那个公式,便可以轻松解决:((len-1)-1)/2,这里我们循环的判断条件是i>=0,为什么呢?因为我们之前已经了解过了,最后一个节点便是根节点,它的下标便是0;
然后调用调整为大顶堆的函数(后面实现)
void Heap(int *arr,int len) {
for (int i = ((len-1)-1)/2; i >=0;i--) {//将数组中的数字想象成大顶堆其中(len-1)是尾元素下标
//((len-1)-1)/2是求最后一个非叶子节点,并且有由下向上,由右往左
Heap_Adjust(arr,len,i,len-1);//len-1表示饱和式救援
}
for (int i = 0; i < len - 1; i++) {//这里我们已经把它调整为了大顶堆,只需要每次将0号下标的元素和最后一个元素进行交换
int tmp = arr[0];
arr[0] = arr[len - 1 - i];
arr[len - 1 - i] = tmp;
Heap_Adjust(arr,len,0,(len-1-i)-1);//交换完成后,重新将其调整为大顶堆形式
}
}
当我们得到大顶堆之后该怎么做呢?这是数据中的最大值便是根节点啦,只需要将根节点和最后一个元素进行交换即可(和冒泡排序有点类似),注意 交换数据完成后,数据已经变了,不是大顶堆了,所以我们再次调用调整大顶堆的函数将其再次调整为大顶堆
好了,现在来实现一下如何调整为大顶堆。
我们知道,调整的顺序是由右向左,由下往上的,所以这里我们传递进来的参数start是需要调整的父节点的下标。
void Heap_Adjust(int* arr,int len,int start,int end) {
int tmp = arr[start];
for (int i = start * 2 + 1; i <= end; i = start * 2 + 1) {
if (i+1<=end&&arr[i+1]>arr[i]) {//i+1<=end表示右孩子的下标合法存在,并且右孩子大于左孩子
i++;//则让i从原来指向左孩子改为指向有孩子
}
if (arr[i] > tmp) {
arr[start] = arr[i];//交换较大的孩子和父亲
start = i;//并且将i重新赋值给start并开始下一轮循环
}
else {
break;
}
}
arr[start]=tmp;
}
首先我们来找到它的左右孩子,i=start*2+1指向左孩子,我们判断(i+1)右孩子是否存在(如果右孩子存在,那么左孩子一定存在)并且右孩子大于左孩子,我们让i+1指向右孩子
如果较大的孩子比父亲大?那么,将较大的孩子赋值给父亲,并且将较大孩子的下标给start让其开始下一轮的判断。如果小,那么直接停止循环
最后将临时量的值赋给arr[start]
最后我们来测试一下
emm,还缺少一个打印函数,哈哈,这个就快快的写了,也写过很多次了
void show(int *arr,int len) {
for (int i = 0; i < len ;i++) {
printf("%3d",arr[i]);
}
}
测试:
int main() {
int arr[] = {1,5,3,4,8,6,7,5,4};
int len = sizeof(arr) / sizeof(arr[0]);
Heap(arr,len);
show(arr,len);
}
测试用例:
好了,堆排序到此结束,有问题欢迎一起讨论解决