C++ 实现最大堆 及 最大堆排序

最大堆参考资料:堆和堆排序
实现一个最大堆,堆的建立可以使用数组或者一个一个添加,可以通过提取堆顶元素实现降序排序,而原位排序则可以实现升序排序,这里使用原位排序。
MaxHeap.h

#pragma once
#include<iostream>
using namespace std;

class MaxHeap
{
private:
	int maxsize;//记录堆允许的最大容量
	int count;//记录堆目前已插入元素个数
	int* a;//堆使用数组保存,记录数组指针
	void shiftUp(int n);//向上交换元素,保持堆特性,不需向外暴露
	void shiftDown(int n);//向下交换元素
public:
	MaxHeap(int size);//建立空堆,元素通过后期一一添加
	MaxHeap(int*pa, int n);//根据已有数组建立堆
	void add(int value);//插入元素
	int extractTop();//提取堆顶元素
	void sort();/*对数组排序,但是会提取出堆内所有元素,
	              也就是这个堆理论上已经空了,但是堆的数组存储的元素已经有序*/
};

MaxHeap.cpp

#include"MaxHeap.h"

MaxHeap::MaxHeap(int size)
{
	count = 0;
	maxsize = size;
	a = new int[size];//申请堆空间
}

void MaxHeap::add(int value)
{
	if (count >= maxsize)
	{
		cout << "堆已满,操作失败!" << endl;
		return;
	}
	a[count] = value;
	count++;//注意增加堆已用容量
	shiftUp(count - 1);//保持堆的特性
	for (int i = 0; i < count; i++)
		cout << a[i] << " ";
	cout << endl;
}
void MaxHeap::shiftUp(int n)
{
	if ((n>0)&&(a[n] > a[(n - 1) / 2]))//递归出口,当子节点不大于父节点或者n等于0时,结束递归
	{
		int temp = a[n];
		a[n] = a[(n - 1) / 2];
		a[(n - 1) / 2] = temp;
		shiftUp((n - 1) / 2);
	}
}
int MaxHeap::extractTop()
{
	int val = a[0];
	a[0] = a[count - 1];
	count--;
	shiftDown(0);
	return val;
}
void MaxHeap::shiftDown(int n)
{
	if (n * 2 + 2 < count)//一般情况:左右子节点都存在
	{
		int temp;
		if ((a[n * 2 + 1] >= a[n * 2 + 2]) && (a[n] < a[n * 2 + 1]))
		{
			temp = a[n];
			a[n] = a[n * 2 + 1];
			a[n * 2 + 1] = temp;
			shiftDown(n * 2 + 1);
		}
		if ((a[n * 2 + 1] < a[n * 2 + 2]) && (a[n] < a[n * 2 + 2]))
		{
			temp = a[n];
			a[n] = a[n * 2 + 2];
			a[n * 2 + 2] = temp;
			shiftDown(n * 2 + 2);
		}
	}
	if ((n * 2 + 1 == count - 1)&& a[n] < a[n * 2 + 1])//只存在左子节点
	{
		int temp;
		temp = a[n];
		a[n] = a[n * 2 + 1];
		a[n * 2 + 1] = temp;
		shiftDown(n * 2 + 1);
	}
}

MaxHeap::MaxHeap(int*pa, int n)
{
	a = pa;
	maxsize = n;
	count = 0;
	for (int i = 0; i < n; i++)
		add(a[i]);/*注意,这里是原位传递,最大堆是在原有数组空间上建立的,
				传递入堆的a[i]会对i位置及i之前的位置重新排序,但是不会影响i后面的数据;
				注意理论上的堆与实际的数据存储之间的关系*/
}

void MaxHeap::sort()
{
	cout << "排序:" << endl;
	int n = count;
	for (int i = n - 1; i >= 0; i--)
	{
		a[i] = extractTop();/*在提取堆顶元素的方法中,对i个元素在理论上的堆内已经被清空,因此可以用于原位存储排序*/
	}
	for (int i = 0; i < n; i++)
	{
		cout << a[i] << " ";
	}
	cout << endl;
}

Main.cpp(客户程序验证)

#include"MaxHeap.h"
#include<time.h>//用于产生随机数
int main()
{
	srand((unsigned)time(NULL));
	MaxHeap h(10);
	for (int i = 0; i < 10; i++)
	{
		int temp = rand() % 100;//产生100以内的随机数
		cout << temp << endl;
		h.add(temp);
	}
	h.sort();
	system("pause");
	return 0;
}

运行结果

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值