堆排序

北大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]);
		}
	}
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值