堆排序实现

一、堆排序应用

1、海量数据找TopK

2、优先级队列

优先队列首先是一个队列，队列具有先进先出的特性，在优先队列中，给数据加了一个权值(优先级)，此时就不是先进先出了，而是优先级最高的最先出队。可以看出，堆和优先队列非常相似。一个堆就可以看作一个优先级队列。在优先级队列中插入一个元素，就相当于往堆中插入一个元素，从优先级队列中取出优先级最高的元素，就相当于取出堆顶元素。JDK中PriorityQueue就是通过堆化来实现的。

二、java代码实现

package com.tzl;

/**
 * Created by tzl on 2019/5/14.
 */
public class HeapSort {
    public static void main(String[] args) {
        int[] arr = new int[]{2, 8, 4, 5, 9, 6, 8, 1, 7};
        HeapSort.sort(arr);
        for (int i = 0; i < arr.length; i++) {
            System.out.print(arr[i] + "");
        }
    }

    private static void sort(int[] arr) {
        int j = arr.length;
        /**
         * 构建最小堆，从最后一个非叶子结点开始(j/2为层高，再减一则为最后一个非叶子结点),依次处理每个非结点
         */
        for (int i = j / 2-1; i >= 0; i--) {
            buildHeap(arr, i, j);
        }

        /**
         * 将堆顶元素交换到数组尾部后，继续构建最小堆
         */
        while (j > 0) {
            int tmp = arr[j - 1];
            arr[j - 1] = arr[0];
            arr[0] = tmp;
            buildHeap(arr, 0, --j);
        }
    }

    private static void buildHeap(int[] arr, int i, int len) {

        /**
         * 当前结点值
         */
        int temp = arr[i];
        /**
         * 从左叶子结点开始，如果左大于右，则指针j指向右，否则指针仍然在左，比较指针对应值和父结点值，
         * 如果需要交换，则将父结点下沉，下沉后需要继续处理下层结点的子树。
         * 如果未交换，说明这个子树不需要处理，退出循环
         *
         **/
        for (int j = 2 * i + 1; j < len; j = 2 * j + 1) {
            if (j < len - 1 && arr[j] > arr[j + 1]) {
                j++;

            }

            if (temp > arr[j]) {
                arr[i] = arr[j];
                arr[j] = temp;
                i = j;
            } else {
                break;
            }
        }
    }
}