优先级队列的简单实现

自动调整

namespace my{
	template<class T,class Cont=vector<T>,class Comper=less<T>>//提供类型、容器、比较方法(默认大堆)
	class priority_queue{
		public:
			priority_queue():c(){
				
			}
			template<class InputIterator>
			priority_queue(InputIterator first,InputIterator last):c(first,last){
				make_heap(c.begin(),c.end(),comp); //建立堆
			}
		public:
			bool empty(){
				return 0==c.size();
			}
			size_t size(){
				return c.size();
			}
		public:
			void push(const T &x){
				c.push_back(x);
				push_heap(c.begin(),c.end(),comp);//调整堆
			}
			void pop(){
				pop_heap(c.begin(),c.end(),comp);
				c.pop_back();
			}
			T top(){
				return c.front();
			}
		private:
			Cont c;  //容器
			Comper comp; //比较仿函数
	};
};
int main(){
	vector<int> v{3,2,7,60,4,1,9,8,5};
	my::priority_queue<int> pq(v.begin(),v.end());

	cout<<endl;
	pq.push(100);
	pq.push(200);
	cout<<pq.top();
	pq.pop();
	cout<<pq.top();
	return 0;
}

手动调整

namespace my{
	template<class T,class Cont=vector<T>,class Comper=less<T>>//提供类型、容器、比较方法(默认大堆)
	class priority_queue{
		public:
			priority_queue()
			{}
			template<class InputIterator>
			priority_queue(InputIterator first,InputIterator last):c(first,last){
				//手动调整堆
				int n=c.size();
				int start=n/2-1; //找到二叉树的最后一个分支
				while(start>=0){
					_AdjustDown(start,n);
					start--;
				}
			}
		public:
			void show_heap()const{
				for(int i=0;i<c.size();i++)
					cout<<c[i]<<" ";
				cout<<endl;
			}
		protected:
			//向下调整
			void _AdjustDown(int start,int n){
				int i=start;
				int j=2*i+1;//左子树
				while(j<n){
					if(j+1<n && c[j]<c[j+1]) //找到左右子树最大值
						j++;
					if(c[j]>c[i]){ 
						T tmp=c[i];
						c[i]=c[j];
						c[j]=tmp;
						//向下追踪
						i=j;
						j=2*i+1;
					}//交换
					else break;
				}
			}
		private:
			Cont c;  //容器
			Comper comp; //比较仿函数
	};
};
int main(){
	vector<int> v{3,2,7,60,4,1,9,8,5};
	my::priority_queue<int> pq(v.begin(),v.end());
	pq.show_heap();
	return 0;
}

结果看还不错~

namespace my{
	template<class T,class Cont=vector<T>,class Comper=less<T>>//提供类型、容器、比较方法(默认大堆)
	class priority_queue{
		public:
			priority_queue()
			{}
			template<class InputIterator>
			priority_queue(InputIterator first,InputIterator last):c(first,last){
				//手动调整堆
				int n=c.size();
				int start=n/2-1; //找到二叉树的最后一个分支
				while(start>=0){
					_AdjustDown(start,n);
					start--;
				}
			}
		public:
			void show_heap()const{
				for(int i=0;i<c.size();i++)
					cout<<c[i]<<" ";
				cout<<endl;
			}
			void push(const T &x){
				c.push_back(x);
				_AdjustUp(c.size()-1);
			}
			void pop(){
				std::swap(c[0],c[c.size()-1]);
				c.pop_back();
				_AdjustDown(0,c.size());
			}
		protected:
			//向下调整
			void _AdjustDown(int start,int n){
				int i=start;
				int j=2*i+1;//左子树
				while(j<n){
					if(j+1<n && c[j]<c[j+1]) //找到左右子树最大值
					//if(comp(c[i],c[j]))
						j++;
					if(c[j]>c[i]){ 
						T tmp=c[i];
						c[i]=c[j];
						c[j]=tmp;
						//向下追踪
						i=j;
						j=2*i+1;
					}//交换
					else break;
				}
			}
			//向上调整
			void _AdjustUp(int start){
				int j=start;//子
				int i=(j-1)/2;//父
				while(i>=0){
					if(c[j]>c[i]){
						T tmp=c[i];
						c[i]=c[j];
						c[j]=tmp;
						
						j=i;
						i=(j-1)/2;
					}
					else break;
				}
			}
			
		private:
			Cont c;  //容器
			Comper comp; //比较仿函数
	};
};
int main(){
	vector<int> v{3,2,7,60,4,1,9,8,5};
	my::priority_queue<int> pq(v.begin(),v.end());
	//my::priority_queue<int,vector<int>,generate<int>> pq(v.begin(),v.end());//小堆
	pq.show_heap();
	pq.push(15);
	pq.pop();
	pq.show_heap();
	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值