经典排序 之 堆排

1. 堆排的时间复杂度为O(nlogn), 空间复杂度为O(1)。

2. 先上原汁原味手写的代码

void heapAdjust(vector<int>& array, int len, int idx) {
    int left = idx * 2 + 1;   //左孩子
    int right = left + 1;    //右孩子
    if(left < len || right < len)   //先判断是否存在左右孩子
    {
        int maxidx = idx;   //记录最大值的index
        int maxvalue = array[idx];   //记录对应的最大值
        if(left < len && array[left] > maxvalue)
        {
            maxvalue = array[left];
            maxidx = left;
        }
        if(right < len && array[right] > maxvalue)
        {  
            maxvalue = array[right];
            maxidx = right;
        }
        if(idx != maxidx)  //如果该节点比左右孩子其中一个要小,则交换,并进一步调整交换后的孩子节点
        {
            swap(array[idx], array[maxidx]);
            heapAdjust(array, len, maxidx);
        }
    }
}

void heapInit(vector<int>& array) {
    for(int i = array.size()/2 - 1; i >= 0; i--)  //从末尾的 非叶子节点 开始调整
        heapAdjust(array, array.size(), i);
}
    
void heapSort(vector<int>& array) {
    heapInit(array);  //先初始化堆
    for(int i = array.size() - 1; i >= 0; i--)
    {
        swap(array[0], array[i]);  //将最大值放到树的最后面
        heapAdjust(array, i, 0);  //数组长度一直随着i在变小,然后从根开始调整
    }
}

堆排的思想基于构造一个完全二叉树,以最小堆为例子,步骤为:先初始化堆,然后逐次把根(最大值)放到树的末尾,并从根处重新调整。

注意点:a. 当树的下标为01234时,对于下标为 i 的节点,左右孩子分别 2*i+1, 2*i+2。

b. 注意树长度和相应坐标的关系。

c. 初始化堆时,其实可以只从非叶子节点开始,对应的坐标为(array.size()/2 - 1),不要搞错。如果容易记错,可以直接从末尾array.size()-1开始,这样进入adjust里也不会进行调整,不影响结果。


3. 堆排经常用于 top k 这样的问题上面。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值