堆常用来实现优先队列,在这种队列中,待删除的元素为优先级最高(最低)的那个。在任何时候,任意优先元素都是可以插入到队列中去的,是计算机科学中一类特殊的数据结构的统称。
一、堆的定义:
如果有一个关键码的集合K={k0,k1,k2,…kn-1},把它的所有元素按完全二叉树的顺序存储方式存储
在一个一维数组中,并满足:Ki<=K2i+1且Ki<=K2i+2(Ki>=K2i+1且Ki>=K2i+2) i =0,1,2…则称为小堆(或大堆)。将根节点最大的堆叫做最大堆或大根堆,根节点最小的堆叫做最小堆或小根堆。
堆的性质:
●堆中某个节点的值总是不大于或不小于其父节点的值;·堆总是一棵完全二叉树。
如下图,图一为大根堆,图二为小根堆。
二、堆的存储
从堆的概念可知,堆是一棵完全二叉树,因此可以层序的规则采用顺序的方式来高效存储。
注意:对于非完全二叉树,则不适合使用顺序方式进行存储,因为为了能够还原二叉树,空间中必须要存储空节点,就会导致空间利用率比较低。
将元素存储到数组中后,可以根据二叉树章节的性质5对树进行还原。假设i为节点在数组中的下标,则有:
- 如果i为0,则i表示的节点为根节点,否则i节点的双亲节点为(i -1)/2;
- 如果2i+1小于节点个数,则节点的左孩子下标为2i+1,否则没有左孩子;
- 如果2i+2小于节点个数,则节点i的右孩子下标为2i+2,否则没有右孩子。
三、支持的基本操作
堆支持以下的基本操作:
- creat:建立一个空堆;
- offer:向堆中插入一个新元素;
- peek:获取当前堆顶元素的值;
- poll:删除堆顶元素。
堆的两种调整方式:
- 向下调整
- 向上调整
四、自定义实现堆的基本操作
1、向下调整:
(1)前提:当前节点的左右子树,都是符合堆的要求,只有根节点不符合。这个时候就可以针对根节点开始进行向下调整了。
(2)方法:以大堆为例
- 先找到左右子树中较大的值;
- 拿着较大的值和根节点进行比较判定是否符合堆的要求;如果不符合,就交换这两个元素;
- 交换完毕之后,再针对对应的子树继续进行向下调整,直到整个树都符合堆的规则为止。
// array 表示用来存储堆的数组.
// size 表示数组上有效元素的个数(有效元素不一定和 array.length 一样)
// index 表示从哪个位置开始进行向下调整
// 以大堆为例
public static void shiftDown(int[] array, int size, int index) {
int parent = index;
int child = 2 * parent + 1;
// 循环条件的含义:
// 如果 child < size, 就意味着 child 是存在的.
// 如果 child >= size, 就说明 parent 已经是叶子节点.
while (child < size