一、认识堆
我们先来堆是一种什么样的数据结构。
堆在逻辑上是一个完全二叉树,但物理上是保存在数组中的。并且堆还需要满足任意节点的值大于或小于其子树中节点的值。如果根节点的值小于子树的值,我们称为小堆。如果根节点的值大于子树的值,称为大堆。
堆的基本作用就是帮我们快速找到集合中的最值。
二、向下调整
我们在建堆的时候,是需要取借助向下调整的。
1、过程(以小堆为例):
(1)设定根节点为当前需要调整的节点-parent。
(2)找到当前节点的左右子树的值(通过下标来查找)。
(3)比较左右子树的值。将较小的节点设为child。
(4)比较child与parent的值的大小。如果child<parent,将两者进行交换,否则就不交换。
(5)由于child的位置始终是子节点,所以需要将child视为parent重新循环上面的步骤。直到parent没有子节点。
2、图解:
3、代码实现:
//size:数组中的被作为堆元素的个数
//index:需要调整的元素的下标
public static void shiftDown(int[] array,int size,int index) {
int parent = index;//第一步
int child = (2 * parent) + 1;//第二步
//保证下标的合法性
while(child < parent) {
//第三步
if(child + 1 < size && array[child + 1] < array[child]) {
child = child + 1;
}
//代码走到这里,child就指向的是parent的子节点中的较小值。
//第四步
if(array[child] < array[parent]) {
int tmp = array[child];
array[child] = array[parent];
array[parent] = tmp;
}else{
//child的值>parent的值,不用交换。
break;
}
//第五步
parent = child;
child = (parent * 2) + 1;
}
}
三、建堆:
如果给出一组数组,它还不是一个堆,我们可以通过代码的实现给构建一个堆。可以利用向下调整,从倒数第一个非叶子节点开始,从后向前依次向下调整,一直到根节点就可以构成堆了。
public static void creatHeap(int[] array,int size) {
for (int i = (size - 1 -1) /2; i >=0 ; i--) {
shiftDown(array,size,1);
}
}
四、优先级队列
优先队列的实现方法其实有很多,但常见的还是使用堆来构建。
1、入队列:(以大堆为例)
步骤:
(1)按尾插法放入元素。
(2)将堆中有效元素数加一。
(3)与父节点比较,父节点的值大于该节点,说明满足堆的性质。否则进行向上调整。
图解:
代码实现:
public void offer(int x) {
array[size] = x;
size++;
shiftUp(array,size-1);
}
private static void shiftUp(int[] array,int index) {
int child = index;
int parent = (child -1) /2;
while (child > 0) {
if(array[parent] < array[child]) {
int tmp = array[parent];
array[parent] = array[child];
array[child] = tmp;
}else {
break;
}
child = parent;
parent = (child - 1) /2;
}
}
2、出队列:
关于出队列,可能会想到直接取堆顶元素,最后将每一个元素都向前移动。但是这样不仅操作麻烦,而且很容易破坏堆的结构。
我们可以数组的最后一个元素替换堆顶元素,最后用一次向下转型就可以了。
3、返回队首元素:
直接取堆顶元素就可以了。