C/C++实现大根堆(数组实现)

思想

利用一个一维数组通过下标来实现一个大根堆,大根堆其实就是一棵完全二叉树
对于任意一个位置 i 上的节点,他的父节点下标为 (i-1)/2
他的左孩子下标为 2i+1 , 他的右孩子下标为 2i+2
对于0位置上的节点,(0-1)/2就是0,因此0位置上的节点的根节点就是他自己
大根堆就是要保证每一棵子树的根节点是这棵树上最大的

建立大根堆

时间复杂度O(N)

void swap(int arr[], int i, int j){
    int temp = arr[i];
    arr[i] = arr[j];
    arr[j] = temp;
}

void bigRootHeap(int arr[], int size){
    for(int i = 0; i < size; i++){
        for(int j = i; j >= 0; j = (j-1)/2){
            if(arr[j] > arr[(j-1)/2]){
                swap(arr, j, (j-1)/2);
            }else{
                 break;
            }
        }
    }
}

堆的调整(heapify)

我们需要在已经建立好的堆中有一个节点发生变化,就需要对原来已经构建好的堆进行调整
如果当前节点变小了,那么就需要找到他的两个孩子,然后与两个孩子中最大的那个节点进行比较,如果当前更新的节点比最大的孩子小则与最大的孩子进行交换

// heapSize 堆的大小
void heapify(int arr[], int loc, int newNum, int heapSize){
    arr[loc] = newNum;
    int L = 2*loc+1; // 左孩子
    while(L < heapSize){
        int bigLoc = L+1 < heapSize && arr[L+1] > arr[L] ? L+1 : L; //最大的孩子的下标
        if(arr[bigLoc] > arr[loc]){
            swap(arr, loc, bigLoc);
        }else{
            break;
        }
        loc = bigLoc;
        L = 2*loc+1;
    }
}

堆的应用

堆实现O(1)时间复杂度查询中位数

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值