堆的模拟实现(详解)c++

根据前两篇文章(向上调整算法以及向下调整算法),堆的实现就变得巨简单了。

1 创建

创建⼀个⾜够⼤的数组充当堆;

创建⼀个变量 n,⽤来标记堆中元素的个数

cpp 复制代码
const int N = 1e6 + 10;
int n;
int heap[N];

2 插⼊

把新来的元素放在最后⼀个位置,然后从最后⼀个位置开始执⾏⼀次向上调整算法即可。

cpp 复制代码
void push(int x)
{
	heap[++n] = x;
	up(n);
}

时间复杂度:

时间开销在向上调整算法上,时间复杂度为 O(log N) 。

3 删除栈顶元素

将栈顶元素和最后⼀个元素交换,然后 n--,删除最后⼀个元素;

从根节点开始执⾏⼀次向下调整算法即可

cpp 复制代码
void pop()
{
	swap(heap[1], heap[n]);
	--n;
	down(1);
}

时间复杂度:

时间开销在向上调整算法上,时间复杂度为 O(log N) 。

4 堆顶元素

下标为 1 位置的元素,就是堆顶元素

cpp 复制代码
int top() { return heap[1]; }

时间复杂度:

显然是 O(1) 。

5 堆的⼤⼩

n 的值。

cpp 复制代码
int size() { return n; }

时间复杂度:

显然是 O(1) 。

全部代码实现:

cpp 复制代码
#include <iostream>
using namespace std;

const int N = 1e6 + 10;
int n;
int heap[N];

//向上调整算法
void up(int child) //每次和父亲做比较
{
	int parent = child / 2;
	//父节点存在且当前结点值大于父节点的权值
	while (parent >= 1 && heap[child] > heap[parent])
	{
		swap(heap[child], heap[parent]);
		child = parent;
		parent = child / 2;
	}
}

// 向下调整算法
void down(int parent)//拿当前结点和孩子作比较
{
	int child = parent * 2;
	//当孩子存在指向向下调整算法
	while (child <= n) //左孩子都不存在,右孩子一定不存在
	{
		//找最大的孩子
		//存在右孩子并且右孩子大于左孩子,条件满足指向最大的孩子,否则不作为
		if (child + 1 <= n && heap[child + 1] > heap[child]) child++;
		if (heap[parent] >= heap[child]) return; //如果父节点大于孩子节点就不用调整了
		
		swap(heap[parent], heap[child]); //交换父子节点
		parent = child; //父节点向下走
		child = parent * 2; //孩子向下走
	}
}

// 插入元素
void push(int x)
{
	heap[++n] = x;
	up(n);
}

// 删除堆顶元素
void pop()
{
	swap(heap[1], heap[n]);
	--n;
	down(1);
}

// 查询堆顶元素
int top() { return heap[1]; }
// 堆的大小
int size() { return n; }

int main()
{
	//测试堆
	int a[10] = { 1, 41, 23, 10, 11, 2, -1, 99, 14, 0 };

	//将这10个数依次放入堆中
	for (int i = 0; i < 10; ++i)
	{
		push(a[i]);
	}

	//输出降序
	while (size())
	{
		cout << top() << " ";
		pop();
	}

	return 0;
}
  • 在大根堆中,每次拿出的堆顶元素都是最大的,删完之后再拿出的堆顶元素是次大的,所以输出的结果是一个降序的结果
相关推荐
keepDXRcuriosity11 分钟前
深入探索快速排序算法:原理与 C 语言实现
c语言·算法·排序算法
龙湾开发36 分钟前
计算机图形学编程(使用OpenGL和C++)(第2版)学习笔记 05.纹理贴图
c++·笔记·学习·3d·图形渲染·贴图
flying_131443 分钟前
面试常问系列(一)-神经网络参数初始化-之-softmax
深度学习·神经网络·算法·机器学习·面试
yasuniko1 小时前
C++线程库
开发语言·c++
还有几根头发呀1 小时前
深入理解C/C++内存管理:从基础到高级优化实践
c++
蒟蒻小袁2 小时前
力扣面试150题-- 翻转二叉树
算法·leetcode·面试
养一只Trapped_beast2 小时前
【LeetCode】删除排序数组中的重复项 II
算法·leetcode·职场和发展
矢鱼2 小时前
单调栈所有模版(2)
算法
轮到我狗叫了2 小时前
力扣智慧思想小题,目录力扣.跳跃游戏(思想很重要)力扣.跳跃游戏II(还是思想)力扣.分发糖果力扣151.反转字符串中的单词力扣.轮转数组
数据结构·算法·leetcode
zxctsclrjjjcph2 小时前
【递归、搜索和回溯】递归、搜索和回溯介绍及递归类算法例题
开发语言·c++·算法·力扣