堆的定义:
堆是一颗完全二叉树,树中每个节点的值都不小于(或不大于)其左右孩子结点的值。其中,如果父节点的值大于或等于孩子节点的值,则称之为大顶堆。如果父节点的值小于或等于孩子节点的值,则称之为小顶堆。
堆一般用优先队列来实现。
#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; // 访问堆中最后一个元素
}
以上就是关于堆的常用接口。
参考《算法笔记》
谢谢阅读。