C++ 堆结构(数组实现)

要说最大堆和最小堆,就得先知道最大树和最小树。

每个结点的值都大于(小于)或等于其子节点(如果有的话)值的树,就叫最大(最小)树。

最大堆(最小堆)是最大(最小)完全树。

由于堆是完全二叉树,所以可以用公式化描述,用一维数组来有效的描述堆结构。

利用二叉树的性质:

如果对一棵有n个结点的完全二叉树的结点按层序编号(从第1层到第[log2n]向下取整+1层,每层从左到右),则对任一结点i(1≤i≤n),有:

(1)如果i=1,则结点i无双亲,是二叉树的根;如果i>1,则其双亲是结点[i/2]向下取整。

(2)如果2i>n,则结点i为叶子结点,无左孩子;否则,其左孩子是结点2i。

(3)如果2i+1>n,则结点i无右孩子;否则,其右孩子是结点2i+1。

这样就就可以将堆中结点移到它的父节点或者其中一个子节点处。

堆中进行的操作主要是:插入,删除,以及初始化。

下面只讨论最大堆。

在最大堆中进行插入操作,例如在下图所示的左边的最大堆中进行插入,插入后结构应该是和右边的图一样的。


插入时,现将新元素从新堆的父节点开始比较,也就是先从2所在结点开始比较,如果小于父节点的值,那么就直接作为孩子结点插入,如果大于,那么就要现将父节点下移,然后再和父节点的父节点比较,一直比较到根节点为止,所以插入过程,是从叶节点到根的一条路径。

最大堆的删除,从最大堆的删除时,该元素从根部移出。删除后,此时堆需要重新构造,以便于仍然为完全二叉树。例如下图:


先将根节点的值20取出后,为了结构仍然是二叉树,所以直接将最后一个结点,也就是保存2的结点去掉,然后将2的值保存下来,然后呢,就从根节点的两个孩子开始比较,看看2 根的左右孩子,这三个元素中,谁最大,那么根节点的值就是谁,然后就将2和空出来的结点的左右孩子(如果有的话)比较,确认子树的根节点的值,这样一步一步确认下去。

用数组初始化堆,数组元素显示按照下面的顺序,一个一个放在结点中


然后就从最后一个父节点开始,就是第五个结点,10所在的位置啦,从那里开始构造以该节点为根的最大堆。构造好后就移到下一个结点,15所在结点,以此类推,一直到根节点。

下面是代码:

文件"maxheap.h"

#include <iostream>
using namespace std;

template <class T>
class MaxHeap
{
private:
	T *heap;
	int CurSize;
	int MaxSize;
public:
	MaxHeap(int maxsize=10)
	{
		MaxSize=maxsize;
		CurSize=0;
		heap=new T[MaxSize+1];
	}

	~MaxHeap()
	{
		delete[]heap;
	}

	int Get_Size() const
	{
		return CurSize;
	}

	T Get_Max()
	{
		if(CurSize==0)
		{
			cout<<"堆为空"<<endl;
			return -9999;
		}
		else
		{
			return heap[1];
		}
	}

	MaxHeap<T> &Insert(const T& x)
	{
		if(CurSize==MaxSize)
		{
			cout<<"堆满,"<<x<<" 插入失败"<<endl;
			return *this;
		}
		//为x寻找应插入位置
		//i从新的叶子结点开始,沿着树向上
		int i=++CurSize;
		while(i!=1 && x>heap[i/2])
		{
			heap[i]=heap[i/2]; //将元素下移
			i/=2; //移向父节点
		}
		
		heap[i]=x;
		cout<<x<<" 插入成功"<<endl;
		return *this;
	}

	MaxHeap<T> &DeleteMax(T& x)
	{
		//将最大元素放入x,并从堆中删除它
		if(CurSize==0)
		{
			x=-9999;
			return *this;
		}
		
		x=heap[1];

		//重构堆
		heap[0]=heap[CurSize--]; //0号位置存放最后一个元素值,然后将该位置删除

		//从根开始,为heap[0]寻找合适位置
		int i=1;
		int ci=2;

		while(ci<=CurSize)
		{
			//ci是较大的孩子的位置
			if(ci<CurSize && heap[ci]<heap[ci+1])
				ci++;

			//判断是否可以放入heap[i]位置
			if(heap[0]>heap[ci])
				break;
			
			//不能
			heap[i]=heap[ci];
			i=ci; // 下移一层
			ci*=2;
		}

		heap[i]=heap[0];
		return *this;
	}

	void Init_heap(T a[],int size,int maxsize)
	{
		delete[]heap;
		heap=new T[maxsize+1];
		CurSize=size;
		MaxSize=maxsize;

		for(int j=1;j<size+1;j++)
			heap[j]=a[j];

		//产生一个最大堆
		for(int i=CurSize/2;i>=1;i--)
		{
			T y=heap[i]; //子树的根

			//寻找放置y的位置
			int c=2*i;
			while(c<=CurSize)
			{
				if(c<CurSize && heap[c]<heap[c+1])
					c++;

				if(y>=heap[c])
					break;

				heap[c/2]=heap[c];
				c*=2;
			}
			heap[c/2]=y;
		}
	}
};
测试文件"main.cpp"

#include "maxheap.h"

#include <iostream>
using namespace std;

int main()
{
	MaxHeap<int> hp;
	int a[11]={-111,5,2,12,3,55,21,7,9,11,9};
	
	cout<<"用数组a来初始化堆:"<<endl;
	for(int i=1;i<11;i++)
		cout<<a[i]<<"  ";
	cout<<endl;

	hp.Init_heap(a,10,15);

	int max;
	cout<<"目前堆中最大值为:"<<hp.Get_Max()<<endl;

	hp.DeleteMax(max);
	cout<<"删除的堆中最大的元素为:"<<max<<endl;
	cout<<"删除后,现在堆中最大的元素为:"<<hp.Get_Max()<<endl;

	cout<<"现在堆的大小为:"<<hp.Get_Size()<<endl;

	cout<<"向堆中插入新元素"<<endl;

	hp.Insert(22);
	hp.Insert(45);
	hp.Insert(214);
	hp.Insert(16);
	hp.Insert(21);
	hp.Insert(121);
	hp.Insert(111);

	cout<<"插入后现在堆的大小为:"<<hp.Get_Size()<<endl;

	cout<<"现在由大到小输出堆中元素"<<endl;

	do
	{
		hp.DeleteMax(max);
		if(max== -9999)
			break;
		cout<<max<<"  ";
	}while(1);

	cout<<endl;

	return 0;
}
测试结果:


在只是需要使用一个优先队列的时候,这种结构是十分有效率的,空间利用率也很高。但是并不适用于所有优先队列的使用,尤其是需要合并两个优先队列或多个长度不相等的队列的时候。


  • 3
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值