北大17年机试中就出现了堆排序的题,今天上课也讲到了这个算法,索性实现一下。
堆排序算法理解起来很容易:
1. 先遍历节点(除叶子之外),不断比较节点与左右子树的大小并进行swap交换,进行一个adjust调整的递归调用,最终堆顶为最大或者最小值;
2. 然后把堆顶元素放到数组最后,对剩下的元素进行上述的调整过程。
堆排序 时间复杂度为nlogn,但是是不稳定的
建堆(初始化+调整堆, 时间复杂度为O(n));
拿堆的根节点和最后一个节点交换(siftdown, 时间复杂度为O(n*log n) ).
不说了,放代码
#include<stdio.h>
#include<algorithm>
using namespace std;
//调整堆
void adjust(int a[], int len, int index) {
int l, r, max;
l = index * 2 + 1; //左子树的节点
r = index * 2 + 2; //右子树的节点
max = index;
if (l < len&&a[l] > a[index]) {
max = l;
}
if (r < len&&a[r] > a[max]) {
max = r;
}
if (max != index) {
swap(a[max], a[index]);
adjust(a, len, max); //递归的调整下面的节点
}
}
//堆排序-建立堆
void heapsort(int a[], int len) {
for (int i = len/2-1 ; i>=0 ; i--) { //先调整使堆顶为最大值
adjust(a, len, i);
}
for (int i = len - 1; i >= 1; i--) { //将堆顶和最后一个节点交换,再调整剩下的节点组成的堆
swap(a[0], a[i]);
adjust(a, i, 0); //i 为除去最后调整好的节点之外的节点长度
}
}
int main() {
int n;
int a[100];
while (scanf("%d", &n) != EOF) {
for (int i = 0; i < n; i++) {
scanf("%d", &a[i]);
}
heapsort(a, n);
for (int i = n-1; i >=0; i--) { //记得最后要倒序输出哦
printf("%d", a[i]);
}
}
}