算法之数据结构基础（5）

一、二叉堆

什么是二叉堆？
二叉堆本质上是一种完全二叉树，它有两个类型：最大堆、最小堆。

• 最大堆：最大堆的任何一个饥父节点的值，都大于或等于它左、右节点的值
• 最小堆：最小堆的任何一个父节点的值都小于或等于它的左、右孩子节点的值。

二叉堆的根节点叫做堆顶，最大堆的堆顶是整个堆中的最大元素。相反，最小堆的堆顶是整个堆中的最小元素。

1、二叉堆的自我调整

即把一个不符合堆特性的完全二叉树，调整为一个堆。
以最小堆为例，从插入、删除、构建三个角度分析。

（1）插入

• 当二叉堆插入节点时，每次插入都插入再完全二叉树的最后一个位置。如图，在一个二叉堆中插入一个新的节点0。
  
• 这时，插入的节点0比其父节点5大，若想要符合最小堆的原则，任意一个父节点都小于等于它的左右孩子节点。就得让新节点0进行上浮，和其父节点交换位置。如下图。
  
• 此时0节点的父节点是2，还是比0大，继续上浮。如图。
  
• 发现根节点1还是比0节点大，继续上浮到堆顶，根节点。如图。
  

（2）删除

二叉堆的删除和插入刚好相反，若所删除的节点位于堆顶，那么为了维持完全二叉树的结构，会把最后一个节点补到堆顶的位置。接着再从堆顶比较左右节点，逐步构建成二叉堆的结构。

• 如图，我们要删除下面二叉堆中的节点0。
  
• 删除了0节点，为了维持完全二叉树的结构，会将树底的节点放入堆顶。如下图。
  
• 接下来，为了维持二叉堆的结构，会将根节点9和其左右孩子节点进行比较。取其左右孩子节点最小的节点进行下沉。显然，此处和节点1交换位置。
  
• 继续和9节点的左右孩子节点进行比较，取最小节点交换位置。如图。
  
  这样一来，就对二叉堆进行了重新调整。

（3）构建二叉堆

即把一个无序的完全二叉树调整为二叉堆，本质就是让所有非叶子节点下沉。
如下图是一个完全二叉树，以此为例，进行构建为一个最小堆。

• 首先，从最后一个非叶子节点开始，也就是节点4，对节点4和其左右孩子节点进行比较，没有比4小的。就不做处理。
• 接下来对节点2进行处理，对比其左右孩子节点，没有比2节点小的，不做处理。
• 接着对节点3进行处理，对比其左右孩子节点，发现节点1小于节点3，进行呼唤位置。得到如图结构。
  
• 接着对节点6进行处理，发现节点6的左右孩子节点都小于本身。取最小孩子进行互换位置。

• 互换位置后，继续对本节点6的孩子节点进行对比，发现3节点小，继续下沉。
  
• 这样就将一个乱序的完全二叉树构建成了一个二叉堆。

（4）小结

堆的插入操作和堆的上浮操作，每次交换次数后，都是堆高度的已办，所以堆的插入和上浮时间复杂度为O(logn)。而堆的构建时间复杂度为O(n)

2、二叉堆的代码实现

二叉堆虽然是一个完全二叉树，但其实现方式是顺序存储，而不是链式存储，即数组。在没有指针指向左右节点的情况下，只能以数组下标的方式进行计算。
例如，若父节点的下标是parent，那么它左孩子节点下标为2×parent+2，右孩子下标为2×parent+2。

（1）上浮

/**
* 上浮
* @param array 待调整的堆
*/
public static void up(int[] array) {
   int childIndex = array.length - 1;
   int parentIndex = (childIndex - 1) / 2;
   //temp保存孩子节点的值，用于赋值，替换。
   int temp = array[childIndex];
   while (childIndex > 0 && temp < array[parentIndex]) {
       array[childIndex] = array[parentIndex];
       childIndex = parentIndex;
       parentIndex = (childIndex - 1) / 2;
   }
   array[parentIndex] = temp;
}

（2）下沉

/**
  * 下沉
  * @param array             待调整的堆
  * @param parentIndex       要下沉的父节点
  * @param length            堆的有效大小
  */
 public static void down(int[] array,int parentIndex,int length) {
     //temp保存父节点。
     int temp = array[parentIndex];
     int childIndex = 2 * parentIndex + 1;
     while (childIndex < length) {
         //如果有右孩子节点，且右孩子小于左孩子，则定位到右孩子
         if (childIndex + 1 < length && array[childIndex + 1] < array[childIndex]) {
             childIndex++;
         }
         //如果父节点小于其最小的孩子节点，直接跳出
         if (temp <= array[childIndex]) {
             break;
         }
         //走到这里，证明有孩子节点比父节点小，进行替换赋值。
         array[parentIndex] = array[childIndex];
         parentIndex = childIndex;
         childIndex = 2 * childIndex + 1;
     }
     array[parentIndex] = temp;
 }

（3）构建堆

构建堆，原理就是将所有非叶子节点进行下沉操作。

/**
 * 构建
 * @param array     待构建的二叉树数组
 */
public static void build(int[] array){
    //从最后一个非叶子节点开始，依次做下沉操作
    for (int i = (array.length - 2) / 2; i >= 0; i--) {
        down(array, i, array.length);
    }
}

运行

public static void main(String[] args) {
    int[] arr = new int[]{1, 74, 6, 8, 5, 3, 78, 9,0,8,6,5,4};
    build(arr);
    System.out.println(Arrays.toString(arr));
}

[0, 1, 3, 8, 5, 4, 78, 9, 74, 8, 6, 5, 6]

（4）小结

二叉堆是实现堆排序和优先队列的基础。