堆排序(C++版)

父节点的索引为i,lchid的索引为2 * i + 1,rchild的索引为2 * i + 2

最后一个含有孩子节点的父节点索引为:(v.size() - 1) / 2

堆排序分3步:

① 维护堆的性质

② 建堆(大根堆,升序排;小根堆,降序排)

③ 排序

#include <iostream>
#include <vector>
#include <algorithm>

using std::vector;
using std::cout;
using std::swap;

// 维护大根堆结点的属性【n为要维护根的大小的索引,i为要调整的节点】
void maxHeapify(vector<int>& v,int i,int n)
{
    int maxIdx = i;
    int lchildIdx = 2 * i + 1;
    int rchildIdx = 2 * i + 2;

    if(lchildIdx <= n && v[maxIdx] < v[lchildIdx])
        maxIdx = lchildIdx;
    if(rchildIdx <= n && v[maxIdx] < v[rchildIdx])
        maxIdx = rchildIdx;
    
    if(maxIdx != i)
    {
        swap(v[maxIdx],v[i]);
        maxHeapify(v,maxIdx,n);
    }
}

// 建立大根堆
void buildMaxSort(vector<int>& v)
{
    for(int i = (v.size() - 1) / 2;i >= 0;-- i)
    {
        // 维护结点的属性
        maxHeapify(v,i,v.size() - 1); // 这里传入最后一个节点的索引
    }
}

// 堆排序
void heapSort(vector<int>& v)
{
    // 建立堆
    buildMaxSort(v);
    // 排序
    for(int i = v.size() - 1;i >= 0;-- i)
    {
        swap(v[0],v[i]);
        maxHeapify(v,0,i - 1); // 这里不能传i,因为刚交换的那个节点我们不能去维护它的堆属性了
    }
}


void printVector(const std::vector<int>& v) {
    for (int num : v)
        std::cout << num << " ";
    std::cout << std::endl;
}

int main(int argc, char const *argv[])
{
    vector<int> v = {12, 11, 13, 5, 6, 7};
    heapSort(v);
    printVector(v);
    getchar();
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值