本文中所有的代码全都是大根堆!实现语言是Java
图片来源都是这位大神的,大神的文章也给了我很多启发 数据结构之堆
堆排序 这个视频通俗易懂从什么是堆,什么是堆化,再到实现堆排序讲的很清晰,实现语言是C
如果有一个关键码的集合K = {k0,k1, k2,…,kn-1},把它的所有元素按完全二叉树的顺序存储方式存储在一个一维数组中,并满足:Ki <= K2i+1 且 Ki<=K2i+2 ,则称为小堆(或大堆)。
将根节点最大的堆叫做最大堆或大根堆,根节点最小的堆叫做最小堆或小根堆。
既然知道了堆在逻辑结构上符合完全二叉树的性质,那么堆就符合完全二叉树的性质,这里有两个性质堆接下来的学习很重要!!
在堆中,cur代表当前节点的下标,parent代表其父节点下标,child1,child2代表其两个子节点下标,那么满足
知道了堆是由数组构成的,那么数组如何变成我们想要的堆呢?这里就提出了一种方法堆化!
heapify堆化,一种将数组转换为堆的算法!其他的博文也把它叫做向上调整算法算法、向下调整算法。
向上调整算法是基于向下调整算法的,所以要想明白向上调整必须先明白什么是向下调整,这也是难点!
向下调整算法原理
向下调整算法是基于原本的堆因为插入元素导致不是堆而做出调整的算法
图中的例子也可以看出587是满足条件的小根堆堆
向上调整算法原理
上文说到向上调整算法是基于向下调整算法实现的!为什么这么说呢?因为向上调整的过程是将二叉树分割成最小单元(这里的最小单元是指由一个根节点零到两个的孩子节点构成的最小二叉树),从下往上对每一个最小二叉树做向下调整算法。当对所有的最小二叉树做完向下调整算法后也就完成了堆化!
明白什么是堆化以后我们来动手实现一下,这里用到的是递归的方法
递归这里有点难理解我单独说一下,因为我们是将数组传进来,第一次遍历的i代表的是堆的最顶部的元素,从上到下直到碰到叶子节点就结束了
/**
* 交换两个值,注意这里不能使用位运算来交换,因为会导致数组首个元素为0的情况
*
* @param array 数组
* @param i 第一个值
* @param j 第二个值
*/
public static void swap(int[] array, int i, int j) {
int temp = array[i];
array[i] = array[j];
array[j] = temp;
}
//堆化是由向上调整算法和向下调整算法构成的
/**
* 向下调整算法
*
* @param tree 存储结构上为数组,逻辑结构上为完全二叉树
* @param size 数组长度
* @param i 父节点
*/
public static void heapify(int[] tree, int size, int i) {
//记录父节点方便后续交换
int max = i;
//父节点的两个子节点
int child1 = 2 * i + 1;
int child2 = 2 * i + 2;
//如果左孩子大于父节点那么记录左孩子结点
if (child1 < size && tree[child1] > tree[max]) {
max = child1;
}
//如果右孩子大于父节点或者是左节点(上一个if判true记录了左孩子)那么记录右孩子
if (child2 < size && tree[child2] > tree[max]) {
max = child2;
}
if (max != i) {
swap(tree, max, i);
heapify(tree, size, max);
}
}
/**
* 向上调整算法 将无序数组构建成堆
*
* @param tree 数组
* @param size 数组大小
*/
public static void buildHeap(int[] tree, int size) {
//从下往上调整算法,从最后一个节点开始
int lastNode = size - 1;
//该节点的父节点
int parent = (lastNode - 1) / 2;
for (int i = parent; i >= 0; i--) {
//对每一个最小单元做向下调整算法
heapify(tree, size, i);
}
}
不理解递归也不关系这里有迭代的
public static void heapify(int[] tree, int size, int i) {
//当前节点的左右孩子
int child1 = 2 * i + 1;
int child2 = 2 * i + 2;
//遍历到叶子节点了也就结束了
while (child1 < size) {
int max = i;
//如果左孩子大于父节点那么记录左孩子结点
if (child1 < size && tree[child1] > tree[max]) {
max = child1;
}
//如果右孩子大于父节点或者是左节点(上一个if判true记录了左孩子)那么记录右孩子
if (child2 < size && tree[child2] > tree[max]) {
max = child2;
}
//如果父结点索引是最大值的索引,那已经是大根堆了,则退出循环
if (i == max) {
break;
}
//父结点不是最大值,与孩子中较大的值交换
swap(tree, max, i);
//将索引指向孩子中较大的值的索引
i = max;
//重新计算交换之后的孩子的索引
child1 = 2 * i + 1;
child2 = 2 * i + 2;
}
}
明白堆化(向下调整)以后相信你对堆的感觉越来越接近了
明白了堆的构建过程后,堆排序简直不要太简单,直接讲算法实现原理
这里贴出完整代码
/**
* 堆排序
*
* @author ccy
* @version 1.0
* @date 2021/11/20 15:11
*/
public class HeapSort {
/**
* 交换两个值,注意这里不能使用位运算来交换,因为会导致数组首个元素为0的情况
*
* @param array 数组
* @param i 第一个值
* @param j 第二个值
*/
public static void swap(int[] array, int i, int j) {
int temp = array[i];
array[i] = array[j];
array[j] = temp;
}
//堆化是由向上调整算法和向下调整算法构成的
/**
* 向下调整算法
*
* @param tree 存储结构上为数组,逻辑结构上为完全二叉树
* @param size 数组长度
* @param i 父节点
*/
public static void heapify(int[] tree, int size, int i) {
//记录父节点方便后续交换
int max = i;
//父节点的两个子节点
int child1 = 2 * i + 1;
int child2 = 2 * i + 2;
//如果左孩子大于父节点那么记录左孩子结点
if (child1 < size && tree[child1] > tree[max]) {
max = child1;
}
//如果右孩子大于父节点或者是左节点(上一个if判true记录了左孩子)那么记录右孩子
if (child2 < size && tree[child2] > tree[max]) {
max = child2;
}
if (max != i) {
swap(tree, max, i);
heapify(tree, size, max);
}
}
/**
* 向上调整算法 将无序数组构建成堆
*
* @param tree 数组
* @param size 数组大小
*/
public static void buildHeap(int[] tree, int size) {
//从下往上调整算法,从最后一个节点开始
int lastNode = size - 1;
//该节点的父节点
int parent = (lastNode - 1) / 2;
for (int i = parent; i >= 0; i--) {
//对每一个最小单元做向下调整算法
heapify(tree, size, i);
}
}
/**
* 堆排序
*
* @param tree 待排序数组
* @param size 数组长度
*/
public static void heapSort(int[] tree, int size) {
buildHeap(tree, size);
for (int i = size - 1; i >= 0; i--) {
swap(tree, i, 0);
heapify(tree, i, 0);
}
}
public static void main(String[] args) {
int[] tree = {
2, 5, 3, 1, 10, 4};
heapSort(tree, tree.length);
for (int temp : tree
) {
System.out.print(temp + " ");
}
}
}
文章写到这里,再介绍一个底层由堆实现的队列PriorityQueue,优先级队列,通过重写compare方法将加入元素进行排序,
PriorityQueue 是基于优先堆的一个无界队列,这个优先队列中的元素可以默认自然排序或者通过提供的
Comparator 在队列实例化的时排序。
PriorityQueue 不允许空值,而且不支持 non-comparable(不可比较)的对象,比如用户自定义的类。优先队列要求使用 Java Comparable 和 Comparator 接口给对象排序,并且在排序时会按照优先级处理其中的元素。
PriorityQueue 的大小是不受限制的,但在创建时可以指定初始大小。当我们向优先队列增加元素的时候,队列大小会自动增加。
PriorityQueue 是非线程安全的,所以 Java 提供了 PriorityBlockingQueue(实现 BlockingQueue接口)用于Java 多线程环境。