本文以小顶堆为例,介绍了堆的构建、插入和删除操作的实现过程,本文所使用的小顶堆如下:
堆的构建
- 初始数组为:9,3,7,6,5,1,10,2,按照完全二叉树,将数字依次填入vector。
- 填入后,找到最后一个结点的父节点(或者直接从vector.size()/2处)开始调整。
- 以此类推,自底向上调整到根节点。
堆的插入
- 首先在vector末尾插入数字0,vector的大小加1。
- 从0的父节点开始,依据最小堆的定义,自底向上,递归调整。
堆的删除
- 首先将二叉树的最后一个节点替换到根节点,然后令vector的大小减1。
- 从根节点开始,依据最小堆的定义,自顶向下,递归调整。
代码实现
堆的调整
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];
}