最小堆的插入删除函数
最小堆的插入删除函数
这里记录最小堆插入删除函数的写法。如果需要最大堆,只需要基本操作的反过来就可以了。
操作基本都有注释。结合注释理解。
Talk is cheap . Show me the code .
vector<int> minHeap;//需要设施哨兵,minHeap[0]=INT_MIN;
void insert(int x) {
minHeap.push_back(x);//将插入值放在堆底
int child=minHeap.size()-1;
while(minHeap[child/2]>minHeap[child]){//上滤操作,调整为最小堆
swap(minHeap[child/2],minHeap[child]);//上滤操作
child/=2;//完全二叉树在数组中表示的特点:child/2=parent
}
}
int Delete(){
int value=minHeap[1];//有哨兵,index为1的位置才是堆顶
swap(minHeap[1],minHeap[minHeap.size()-1]);//交换栈顶和底部元素
minHeap.pop_back();//将目标值弹出
int parent=1,child=0;
for(parent=1;parent*2<minHeap.size();parent=child){
child=parent*2;//完全二叉树在数组中表示的特点:child/2=parent
if(child!=minHeap.size()-1&&minHeap[child+1]<minHeap[child]) child++;
//判断有无右孩子,若有,是左孩子较小还是右孩子较小,再更新child的下标值
if(minHeap[parent]<minHeap[child]) break;//若符合堆的定义,弹出
else swap(minHeap[parent],minHeap[child]);//下滤操作
}
return value;//返回删除的元素值
}
【推荐】还在用 ECharts 开发大屏?试试这款永久免费的开源 BI 工具!
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步