欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

在最小堆中进行插入操作,然后进行顺序输出

程序员文章站 2022-06-01 21:02:02
...

在看本章前需要先了解最小堆的构建以及堆排序:

连接:最小堆:https://blog.csdn.net/weixin_43570155/article/details/103946065
堆排序:https://blog.csdn.net/weixin_43570155/article/details/103946522

在堆中增加一个值:

直接插入到新元素的末尾,在根据句情况判断元素是否需要向上移动

代码:

#include<bits/stdc++.h>
using namespace std;
int h[100];
int n,k;
void swap(int i,int t){
	int temp;
	temp=h[i];
	h[i]=h[t];
	h[t]=temp;
}
int shiftdown(int i){
	int flag=0; //用于标记是否还需要向下调 
	int t=i;//存数字小的下标 
	while(i*2<=n&&flag==0){  //至少存在左子树 
		if(h[i*2]<h[i])    t=i*2;
		if(i*2+1<=n){  //存在右子树 
			if(h[i*2+1]<h[t])   t=i*2+1;
		}
		if(t!=i){   //发现最小的节点编号不是自己,即子节点中存在比父节点小的数 
			swap(i,t);
			i=t;   //更新i点,以便于再次比较下面的点 
		}
		else
		flag=1;   //该点比一下的点都小,不需要向下找了 
	}
}
int shiftup(int i){
	int flag=0;
	if(i==1) return 0;  //在堆顶,则不需要进行向上调整
	while(i!=1&&flag==0){
		if(h[i/2]>h[i]) swap(i/2,i);   //和他的父节点交换位置 
		else  flag=1;
		i=i/2;
	} 
	
}
int deletemin(){
	int t;
	t=h[1];
	h[1]=h[n];  //将堆的最后一个点赋值到堆顶
	n--;
	shiftdown(1);  //向下调整
	return t;      //返回被删除前的最小值 
}
int main(){
	int temp;
	scanf("%d",&n);
	temp=n;
	for(int i=1;i<=n;i++){
		scanf("%d",&h[i]);
	}
	for(int i=n/2;i>0;i--){   //构建最小堆 
		shiftdown(i);
	}
	scanf("%d",&k);   //需要插入的数字的个数 
	for(int i=n+1;i<=n+k;i++){
		scanf("%d",&h[i]);   //插入到末尾 
		shiftup(i);
	}
	for(int i=1;i<=temp;i++)    printf("%d ",deletemin());   //从小到大排序显示 
} 
相关标签: # 神奇的树