public static void heapSort(int[] arr) {
if (arr == null || arr.length < 2) {
return;
}
//以下代码实现了将数组变成大根堆的形式,但是此时只能保证根节点为最大的节点
//此时以数组的形式展现的话是整体按照从大到小排序,但是中间具体排序并不一定按照从大到小
//heapInsert实现的是从叶节点插入一步步向上遍历的过程
for (int i = 0; i < arr.length; i++) {
heapInsert(arr, i);
}
//实现了大根堆之后,此时将大根堆的根节点和数组中的最后一位进行交换,此时实现了将最大值放到最后。
//然后由于此时交换后的根节点肯定不是最大的,所以此时开始进行堆从根节点插入后的排序操作,
//排序结束后重新进行倒数第二位和根节点的位置变换,依次循环。
int size = arr.length;
swap(arr, 0, --size);
while (size > 0) {
heapify(arr, 0, size);
swap(arr, 0, --size);
}
}
//实现从叶节点向上调整堆
//由于对于堆这种完全二叉树结构的数组而言,满足一下的式子
//left = index*2+1,right = index*2+2;
public static void heapInsert(int[] arr, int index) {
while (arr[index] > arr[(index - 1) / 2]) {
swap(arr, index, (index - 1) / 2);
index = (index - 1) / 2;
}
}
//此时为从上向下进行的堆结构重新排序,index为当前节点
public static void heapify(int[] arr, int index, int size) {
int left = index * 2 + 1;
//需要判断该节点是否为叶节点,如果为叶节点,则不需要进行下面的判断
while (left < size) {
//如果当前节点存在子节点,则在当前节点和子节点中选取最大值
//当前语句判断的非常干练,如果当前节点包含右子节点,同时右子节点的值大于左子节点的值,最大值为右子节点
//如果不包含右子节点,或者右子节点数值小于左子节点,则最大值为左子节点的值。
//自己在进行判断的时候没有详细思考这个原理,导致自己代码无法通过。
int largest = left + 1 < size && arr[left + 1] > arr[left] ? left + 1 : left;
largest = arr[largest] > arr[index] ? largest : index;
//如果此时最大值已经是根节点,则可以直接退出,因为已经形成了大根堆的结构
if (largest == index) {
break;
}
//如果不符合,则将根节点和值较大的节点进行交换
swap(arr, largest, index);
//交换结束之后再让值最大的对应子树作为根节点,再进行接下来的判断,直到整棵树的结构变成大根堆
index = largest;
left = index * 2 + 1;
}
}
堆排序代码及注释
最新推荐文章于 2023-08-23 19:55:13 发布