堆排序(C语言实现)
#include<stdio.h>
void swap(int arr[], int i, int j) {//交换数组中的第i个数和第j个数
int temp = arr[i];
arr[i] = arr[j];
arr[j] = temp;
}
//自上而下调整堆
void heapify(int tree[], int n, int i) {//用tree[]数组表示一棵树;n:树的结点数;i: 对第i结点做heapify操作
if (i >= n)
return;
int c1 = 2 * i + 1;//i的第一个子结点
int c2 = 2 * i + 2;//i的另一个子结点
int max = i;//寻找i,c1,c2三个结点中的最大值
if (c1<n && tree[c1]>tree[max]) {//max代表的是下标,不是数值
max = c1;
}
if (c2<n && tree[c2]>tree[max]) {
max = c2;
}
if (max != i) {
swap(tree, max, i);//最大值和第i个数交换
heapify(tree, n, max);
}
}
//构建堆
void bulid_heap(int tree[], int n) {
int last_node = n - 1;//last_node:最后一个结点
int parent = (last_node - 1) / 2;
int i;
for (i = parent; i >= 0; i--) {//从最后一个结点的父结点开始自下而上对每个结点做haepify
heapify(tree, n, i);
}
}
//堆排序
void heap_sort(int tree[], int n) {
bulid_heap(tree, n);//构建大根堆
int i;
for (i = n - 1; i >= 0; i--) {
swap(tree, i, 0);//交换根结点(最大值)和最后一个结点
heapify(tree, i, 0);//砍掉最后一个结点,i代表当前结点的数量,然后从根结点继续做heapify
}
}
int main() {
int tree[] = { 2,5,3,1,10,4 };
int n = 6;
heap_sort(tree, n);
int i;
for (i = 0; i < n; i++) {
printf("%d\n", tree[i]);
}
return 0;
}
运行结果