堆排的实现以及优化

第一种
建堆的时间复杂度O(nlogn)
时间复杂度O(nlongn)
空间复杂度O(n)

#include <iostream>
#include <vector>
#include <algorithm>
#include <cassert>
#include <ctime>

using namespace std;

template <typename Item>
class MaxHeap {
public:
	MaxHeap(int capacity) {
		data = new Item[capacity + 1];
		count = 0;
		this->capacity = capacity;
	}

	~MaxHeap() {
		detele[] data;
	}

	void insert(Item item) {

		assert(count + 1 <= capacity);
		data[count + 1] = item;
		count++;
		shiftup(count);
	}

	Item extractMax() {
		
		assert(count > 0);
		Item ret = data[1];
		swap(data[1], data[count]);
		count--;
		shiftDown(1);

		return ret;
	}
private:
	void shiftup(int k) { //插入时调整堆

		while (k > 1 && data[k / 2] < data[k]) {
			swap(data[k / 2], data[k]);
			k /= 2;
		}
	}

	void shiftDown(int k) { //取出时调整堆

		while (2 * k <= count) {
			
			int j = 2 * k;
			if (j + 1 <= count && data[j + 1] > data[j]) //右节点存在和大小判定
				j++;

			if (data[k] >= data[j]) //父节点大于左右子节点
				break;

			swap(data[k], data[j]);
			k = j;
		}
	}
private:
	int *data;
	int count;
	int capacity;
};

template <typename T>
void heapSort(T arr[], int n) {

	MaxHeap<T> maxheap = MaxHeap<T>(n);
	for (int i = 0; i < n; ++i)
		maxheap.insert(arr[i]);

	for (int i = n - 1; i >= 0; --i)
		arr[i] = maxheap.extractMax();
}

int main() 
{
	srand(time(NULL));

	int *arr = new int[100];
	for (int i = 0; i < 100; ++i)
		arr[i] = rand() % (i + 1);

	heapSort(arr, 100);

	for (int i = 0; i < 100; ++i)
		cout << arr[i] << " ";
	cout << endl;

	delete arr;

	return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值