在最小堆中进行插入操作,然后进行顺序输出
程序员文章站
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()); //从小到大排序显示
}
推荐阅读