func buildHeap(nums []int){
//从底往上构造大顶堆
for i:=len(nums)/2+1;i>=0;i–{//len(nums)/2+1为第一个非叶子节点,减少循环次数,增加效率
chonggou(nums,i,len(nums))//对i为父节点的堆进行构造大顶堆
}//构造大顶堆
for i:=len(nums)-1;i>=0;i–{
nums[0], nums[i] = nums[i], nums[0]
chonggou(nums, 0, i)//首尾交换,本身是大顶堆,交换后使得末尾为堆的最大值,然后对除了刚才交换后的最大值剩余的数字进行重新构造大顶堆
}
}
//认为数组pos为父节点,父节点的左子树为pos2+1,右子树为pos2+2,构造大顶堆
func chonggou(nums []int,pos int,lenth int){
for{
child:=pos*2+1//左子树
if(child+1>=lenth){//超出范围则结束
break
}
if(nums[child+1]>nums[child]){//求更大的子节点
child++
}
if nums[pos]<nums[child]{//若父节点比子节点小则交换,用于构造大顶堆
nums[pos],nums[child] = nums[child],nums[pos]
pos = child//交换后子节点的堆属性有可能受影响,对子节点进行构造大顶堆
}else{//父节点更大,结束
break
}
}
}
堆排序
最新推荐文章于 2022-05-22 00:14:12 发布