堆的定义与基本操作接口:建堆、插入、删除、排序(C++实现)

堆的定义:

堆是一颗完全二叉树,树中每个节点的值都不小于(或不大于)其左右孩子结点的值。其中,如果父节点的值大于或等于孩子节点的值,则称之为大顶堆。如果父节点的值小于或等于孩子节点的值,则称之为小顶堆。

堆一般用优先队列来实现。

#include <queue>
// 一般声明
priority_queue<int> q1;
/*
	自定义优先级比较函数时的声明方式:

		参数1:队列中的元素类型
		参数2:元素使用的容器
		参数3:比较函数:可以自定义
*/
priority_queue<int, vector<int>, less<>> q2;

在实际解题中,使用STL中的优先队列后,我们就可以忽略堆的内部实现,直接使用其接口即可。

下面我们看看关于堆的内部实现:

对于完全二叉树,可以使用数组来进行存储。这样节点就是按照层序的方式存储于数组中,其中第1个节点存储与1号位,并且第i号位的节点的左孩子就是2i号位,右孩子就是(2i+1)号位。

堆的表示(使用数组):

const int maxn = 100;
int n;			// 堆中元素的个数
int heap[maxn];	// 堆:完全二叉树

堆节点的调整:向下调整,排序时使用。向下调整的依据时,堆中父节点的权值要比子节点大,否则交换之。当然,在交换的过程中,要选择左右子节点中权值较大者,进行交换。交换完成后,继续向下比较,直到无左右孩子,或者,父节点的权值已经比孩子权值都大,则对于此节点的调整完成。

// 向下调整
// low为欲调整节点的下标,high为堆中最后一个元素的下标
void downAdjust(int low, int high)
{
	int i = low;
	int j = 2 * low;
	while (j <= high)
	{
		if (j + 1 <= high && heap[j + 1] > heap[j])
		{
			j = j + 1; // 右孩子比左孩子大
		}

		// 左右孩子中的最大值比父节点的值大:将其与父节点交换
		if (heap[j] > heap[i])
		{
			swap(heap[j], heap[i]);
			i = j;
			j = i * 2;
		}
		else
		{
			break;
		}
	}
}

建堆:

// 建堆:堆中元素已输入,进行排序调整操作
void createHeap()
{
	for (int i = n / 2; i >= 1; --i)
	{
		downAdjust(i, n);
	}
}

删除堆顶元素:

// 删除堆顶元素
void deleteTop()
{
	heap[1] = heap[n--];  // 直接将最后一个元素覆盖到首元素,然后向下调整
	downAdjust(1, n);
}

向上调整:插入元素时使用。在插入元素的过程中,一般先将要插入的元素,放到数组的末尾,然后进行向上比较。向上比较的原理时,如果当前节点比父节点的权值大(或者小,对于小顶堆来说),则与父节点进行交换,然后继续向上比较,直到不比父节点大,或者比较到根节点,则整个比较过程完成。

// 向上调整:用于插入元素时使用
// 一般将low设置为1,因为一直要调整到根节点为止,high表示要调整节点的数组下标
void upAdjust(int low, int high)
{
	int i = high; 
	int j = i / 2;

	while (j >= low)
	{
		// 如果父节点的权值比子节点权值小,则交换元素
		if (heap[j] < heap[i])
		{
			swap(heap[j], heap[i]);
			i = j;
			j = i / 2;
		}
		else
		{
			break;
		}
	}
}

插入元素:先将新元素放在数组末尾,然后向上调整顺序。


// 添加元素
void insert(int x)
{
	heap[n++] = x;  // 将新元素添加到堆的根部,然后重新调整堆的顺序
	upAdjust(1, n);	// 向上调整
}

堆排序:访问堆顶元素,然后将末尾元素覆盖到堆顶,进行向下调整的操作,即重新排序。


// 堆排序
void heapSort()
{
	createHeap();	// 建堆
	for (int i = n; i > 1; i--)
	{
		cout << heap[i] << endl; // 访问堆顶元素
		swap(heap[i], heap[1]);  // 用尾部元素,覆盖堆顶元素
		downAdjust(1, i - 1);    // 向下调整,重新成为一个堆
	}
	cout << heap[1] << endl;  // 访问堆中最后一个元素
}

以上就是关于堆的常用接口。

参考《算法笔记》

谢谢阅读。

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值