ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

Day1 Heap与Priority_queue

2021-06-21 03:01:26  阅读:197  来源: 互联网

标签:queue heap int 调换 Priority swap Heap 节点


Heap

Heap是建立在完全二叉树基础上的一类的特殊的数据结构,主要特征为每个父节点的值都不小于(不大于)其两个子节点的值
Heap的建立过程本质上是两棵相邻的Heap的合并,只需比较相邻两棵Heap的根和其公共父亲的值的大小关系即可,将最大的值调换到其公共父亲处即可。注意当调换后可能造成被调换的子树不满足Heap性质,需要一路检查到底

建立完Heap后只要每次取出根节点并将目前最后一个节点调换至根节点并一路交换到底即可。

复杂度\(O(nlogn)\)

代码如下

#include<iostream>
#include<cstdio>
#define maxn 100050
using namespace std;
int n;
int a[maxn],b[maxn];
void heap_swap(int x,int size){
	while(1){
		int ls=x<<1,rs=x<<1|1;
		if(ls>size)return;
		if(rs>size){
			if(a[x]<a[ls])swap(a[x],a[ls]);
			return;
		}
		if(a[ls]>a[rs]){
			if(a[x]<a[ls]){
				swap(a[x],a[ls]);
				x=ls;
			}
			else return;
		}
		else {
			if(a[x]<a[rs]){
				swap(a[x],a[rs]);
				x=rs;
			}
			else return;
		}
	}
}
void heap_sort(int a[]){
	for(int i=n>>1;i;i--)
		heap_swap(i,n);
	for(int i=n;i;i--){
		b[i]=a[1];
		swap(a[1],a[i]);
		heap_swap(1,i-1);
	}
	for(int i=1;i<=n;i++)a[i]=b[i];
}
int main(){
	cin>>n;
	for(int i=1;i<=n;i++)cin>>a[i];
	heap_sort(a);
	for(int i=1;i<=n;i++){
		cout<<a[i]<<' ';
	}
	cout<<endl;
}

Priority_queue

基于大根堆和小根堆,我们不难写出一个可以在\(O(logn)\)复杂度内维护数集最大最小值的做法——优先队列。

按照上文的思路,最值便是数组的第一位,删除操作即为取出根节点并将目前最后一个节点调换至根节点并一路交换到底,插入操作即为在最后一位后再插入一个节点并一路向上交换。

代码如下:

#include<iostream>
#include<cstdio>
#define maxn 100050
using namespace std;
int n;
void heap_swap(int a[],int x,int size){
	while(1){
		int ls=x<<1,rs=x<<1|1;
		if(ls>size)return;
		if(rs>size){
			if(a[x]<a[ls])swap(a[x],a[ls]);
			return;
		}
		if(a[ls]>a[rs]){
			if(a[x]<a[ls]){
				swap(a[x],a[ls]);
				x=ls;
			}
			else return;
		}
		else {
			if(a[x]<a[rs]){
				swap(a[x],a[rs]);
				x=rs;
			}
			else return;
		}
	}
}
struct priority_queue{
	int size=0;
	int x[maxn];
	int front(){
		return x[1];
	}
	void pop(){
		if(!size)return;
		if(size==1){
			size--;
			return;
		}
		swap(x[1],x[size]);
		size--;
		heap_swap(x,1,size);
	}
	void push(int num){
		x[++size]=num;
		int now=size;
		while(now){
			if(now==1)return;
			if(x[now]>x[now>>1])swap(x[now],x[now>>1]);
			else return;
		}
	}
};
int main(){
    /*
	priority_queue q;
	q.push(2);
	q.push(5);
	q.push(3);
	q.push(1);
	cout<<q.front()<<endl;
	q.pop();
	cout<<q.front()<<endl;
	*/
}

标签:queue,heap,int,调换,Priority,swap,Heap,节点
来源: https://www.cnblogs.com/Fang-Hao/p/14910708.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有