最大堆和最小堆排序

原理参考趣学数据结构

代码

#include<stdio.h>
#include<stdlib.h>
int r[] = { -1,1,4,590,4,2,8,7,5,89,67,5,2,1,67,86,54 };//存储要排序的数,第一个元素不存储元素赋值为-1
int length = sizeof(r) / sizeof(int);//待排序的数的个数
void swap(int i, int j) {//交换二个变量的值int temp = r[i];r[i] = r[j];r[j] = temp;
}
void sink(int k, int n,int flag) {//flag=0表示最大堆，否则最小堆,实现最大（小）堆,即所有根节点的值大于等于（小于等于）该根节点下的左右节点int j = 0;while (2 * k <= n) {//从倒数第二层有左孩子的结点开始往上检查是否满足每个子树是否满足最大（小）堆的条件j = 2 * k;if (flag == 0) {//最大堆的更新条件if (j<n&&r[j + 1]>r[j])j++;//找到当前根节点下最大的值if (r[k] < r[j])swap(k, j);else//该根节点下已经满足最大堆的条件则退出检查break;}else if(flag==1) {//最小堆的更新条件if (j<n&&r[j + 1]<r[j])j++;//找到当前根节点下最小的值if (r[k] > r[j])swap(k, j);else//该根节点下已经满足最小堆的条件则退出检查break;}k = j;//更新了当前根结点的值则继续向下检查是否满足最大（小）堆的条件
}
}
void print() {for (int i = 1; i<=length-1; i++) {printf("%d ", r[i]);}
}
void heapSort(int n,int flag) {//最大堆或最小堆排序,flag=0表示最大堆排序，否则最小堆排序，这里的最小堆因为没有开辟新的空间所以是逆序的，而最大堆的排序结果是升序的//初始化堆for (int i = n / 2; i > 0; i--) {sink(i, n,flag);}while (n > 1) {print();printf("\n");swap(1, n);//当前第n个元素为第n大(小）元素--n;//选出了最大（小）的元素，则少一个元素sink(1, n, flag);//只改变了根节点的值，只需要对第一个结点进行下沉的更新}
}int main() {printf("初始的顺序\n--------------\n");print();printf("\n最小堆排序后结果\n--------------\n");heapSort(length-1, 1);print();printf("\n--------------\n");printf("初始的顺序\n--------------\n");print();printf("\n最大堆排序后结果\n--------------\n");heapSort(length - 1, 0);print();printf("\n--------------\n");system("pause");return 0;
}