堆的构建、插入和删除

本文以小顶堆为例,介绍了堆的构建、插入和删除操作的实现过程,本文所使用的小顶堆如下:
在这里插入图片描述

堆的构建

  1. 初始数组为:9,3,7,6,5,1,10,2,按照完全二叉树,将数字依次填入vector。
  2. 填入后,找到最后一个结点的父节点(或者直接从vector.size()/2处)开始调整。
  3. 以此类推,自底向上调整到根节点。

堆的插入

  1. 首先在vector末尾插入数字0,vector的大小加1。
  2. 从0的父节点开始,依据最小堆的定义,自底向上,递归调整。
    在这里插入图片描述

堆的删除

  1. 首先将二叉树的最后一个节点替换到根节点,然后令vector的大小减1。
  2. 从根节点开始,依据最小堆的定义,自顶向下,递归调整。
    在这里插入图片描述
    在这里插入图片描述

代码实现

堆的调整

nums是一个vector,i是当前调整的节点,size是vector的大小。

        //调整堆的结构
    void tune(vector<int>& nums,int i,int size){
        //i的左子节点的索引
        int leftnode=2*i;
        //i的右子节点的索引
        int rightnode=2*i+1;
        int smallestnode=i;
        //寻找到最小值所在的索引
        if(leftnode<size&&nums[leftnode]<nums[smallestnode]){
            smallestnode=leftnode;
        }
        if(rightnode<size&&nums[rightnode]<nums[smallestnode]){
            smallestnode=rightnode;
        }
        //如果最小值的索引不是i,则交换,之后继续调整下一层
        if(smallestnode!=i){
            swap(nums[smallestnode],nums[i]);
            tune(nums,smallestnode,size);
        }
    }

堆的构建和插入

    //建立堆和在堆中插入元素,自底向上递归
    void insert(vector<int>& nums,int size){
        for(int i=size/2;i>=0;i--){
            tune(nums,i,size);
        }
    }

堆的删除

    //删除堆中的元素,自顶向下递归
    void erase(vector<int>& nums,int size){
        tune(nums,0,size);
    }
    int findKthLargest(vector<int>& nums, int k) {
        insert(nums,k);
        for(int i=k;i<nums.size();i++){
            if(nums[i]>nums[0]){
                swap(nums[0],nums[i]);
                erase(nums,k);
            }
        }
        return nums[0];
    }
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值