package com.algorithm2;


import java.util.Arrays;

/**
 * 二叉堆
 */
public class BinaryHeap {

    /**
     * "上浮"调整
     * @param array  待调整的堆
     */


    public static void upAdjust(int[] array) {
        // 定义孩子索引
        int childIndex = array.length - 1;
        // 定义父亲节点索引
        int parentIndex = (childIndex - 1) / 2;
        // temp保存在插入叶子节点的值，用于最后的赋值
        int temp = array[childIndex];
        // 当孩子索引大于0，并且临时保存插入叶子节点的值小于父亲节点的值时
        while (childIndex > 0 && temp < array[parentIndex])
        {
        // 无须真正交换，单向赋值即可
        array[childIndex] = array[parentIndex];
        // 孩子索引等于父亲节点索引
        childIndex = parentIndex;
        // 父亲节点索引等于（父亲节点索引 - 1 ）的一半
        parentIndex = (parentIndex - 1) / 2;
        }
        // 将temp赋值给孩子节点对应索引的值
        array[childIndex] = temp;
    }

    /**
     * “下沉”调整
     * @param array 待调整的堆
     * @param parentIndex  要“下沉”的父节点
     * @param length 堆的有效大小
     */

     public static void downAdjust(int[] array, int parentIndex,int length) {
        // temp保存父节点的值，用于最后的赋值
         int temp = array[parentIndex];
        // 令孩子节点索引等于父亲节点索引的2倍 + 1
         int childIndex = parentIndex * 2 + 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;
         }
     }

    /**
     * 构建堆
     * @param array 待调整的堆
     */

     public static void buildHeap(int[] array) {
         // 从最后一个非叶子点开始，依次做“下沉"调整
         for (int i = (array.length - 2) / 2; i >= 0; i--) {
             downAdjust(array, i, array.length);
         }
     }


  public static void main(String[] args) {
         int[] array = new int[] {1, 3, 2, 6, 5, 7, 8, 9, 10, 0};
         upAdjust(array);
         System.out.println(Arrays.toString(array));

         array = new int[] {7, 1, 3 ,10, 5, 2, 8, 9, 6};
         buildHeap(array);
         System.out.print(Arrays.toString(array));
  }

}