一、堆排序应用
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;
}
}
}
}