自己动手写数据结构:使用堆实现PriorityQueue类模板(vector)

重要概念:堆(heap)——是其元素具有键并且满足以下“堆性质”的完全二叉树:从根到叶节点的任何路径上的键都是非增的。堆用于实现优先级队列,因为其允许O(lg N)次插入和删除,这是由于push()和pop()函数是通过遍历穿过堆的从根到叶节点的路径实现的,这种路径并不比树的高度长,它至多是(lg N)。

#ifndef MYPRIORITYQUEUE_H
#define MYPRIORITYQUEUE_H

#include<algorithm> 
#include <vector>
using namespace std;

template <class T>
class PriorityQueue
{
public:
	PriorityQueue() {}
	PriorityQueue(const PriorityQueue& q):val(q.val) {}
	~PriorityQueue() {}
	PriorityQueue& operator = (const PriorityQueue& q)
	{
		val.clear();
		val = q.val;
	}

	int size() const {return val.size();}
	bool empty() const {return val.empty();}
	const T& top() const {return val.front();}
	void push(const T&);
	void pop();
protected:
private:
	vector<T> val;
	void heapifyDown(); //从堆顶往下重新存储元素,若堆顶元素小于子元素则交换,若两个子元素均满足则与较大者交换
	void heapifyUp();   //从堆底往上重新存储元素,若堆底元素大于父元素则交换,直到堆顶
};

template <class T>
void PriorityQueue<T>::heapifyDown()
{
	int len = val.size();
	for (int i=0; i<len;)
	{
		int j = i*2 + 1;
		if (j<len && val[i]<val[j])
		{
			if (j+1<len && val[j]<val[j+1])
			{
				swap(val[i], val[j+1]);
			}
			else
				swap(val[i], val[j]);
			i = j;
		}
		else
		{
			if (++j<len && val[i]<val[j])
			{
				swap(val[i], val[j]);
				i = j;
			}
			else
				break;
		}
	}

}

template <class T>
void PriorityQueue<T>::heapifyUp()
{
	int len = val.size();
	for (int i = len-1; i > 0;)
	{
		int j = (i-1)/2;
		if (val[i] > val[j])
		{
			swap(val[i], val[j]);
			i = j;
		}
		else
			break;
	}
}

template <class T>
void PriorityQueue<T>::push(const T& t)
{
	val.push_back(t);
	heapifyUp();
}

template <class T>
void PriorityQueue<T>::pop()
{
	val.front() = val.back();
	val.pop_back();           //删除堆顶元素,用堆底元素替代
	heapifyDown();
}

#endif
测试函数:
int _tmain(int argc, _TCHAR* argv[])
{
	typedef PriorityQueue<int> PQ;
	PQ q1;
	q1.push(44);
	q1.push(66);
	q1.push(33);
	q1.push(88);
	q1.push(77);
	q1.push(55);
	q1.push(22);
	while (!q1.empty())
	{
		cout<<q1.top()<<" ";
		q1.pop();
	}
	return 0;
}

存储数组:88 77 55 44 66 33 22
输出结果:88 77 66 55 44 33 22

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值