测试链接:https://leetcode.cn/problems/sort-an-array/
堆结构:
是一颗完全二叉树
分为大根堆和小根堆
大根堆:每一颗子树最大值都在子树的根部
小根堆:每一颗子树最小值都在子树的根部
每一位父亲i的两个孩子的节点位置(若存在)分别为:i*2+1,i*2+2
同理每一个孩子的父亲节点位置为:(i-1)/2
堆的建立与排序:
//建立大根堆
//每一颗子树的根都是该子树的最大值
void heapup(int i,vector<int>&nums)//大的数往上调整
{
while(((i-1)>>1)>=0&&nums[i]>nums[(i-1)>>1])
{
swap(nums[i],nums[(i-1)/2]);
i=(i-1)/2;
}
}
void heapdown(int i,vector<int>&nums,int size)//小的数往下调整
{
int l=(i<<1)+1;
while(l<size){
int best;
if(l+1<size)
{
if(nums[l+1]>nums[l])best=l+1;else best = l;
}else{
best=l;
}
if(nums[i]>=nums[best]){
break;
}else{
swap(nums[i],nums[best]);
i=best;
l=(i<<1)+1;
}
}
}
void heapsort1(vector<int>&nums)//从顶到底建立大根堆并排序
{
int size=nums.size();
for(int i=0;i<size;i++)
{
heapup(i,nums);//先up建立大根堆
}
while(size>1)
{
swap(nums[0],nums[size-1]);
size--;
heapdown(0,nums,size);//后dwon进行堆排序
}
}
标签:大根堆,子树,nums,int,堆排序,结构,best,size
From: https://www.cnblogs.com/benscode/p/18670236