在之前的博客中我们说过其他排序,但是没有说堆排序,因为堆排序运用到了树的知识。今天我们来说说使用堆排来进行排序。
基本介绍
代码实现
package com.xxxx.tree;
public class HeapSort {
public static void main(String[] args) {
// TODO Auto-generated method stub
int[] arr= {1,3,5,7,10};
heapSort(arr);
for(int a:arr) {
System.out.print(a+" ");
}
}
//编写一个堆排序的方法
public static void heapSort(int[] arr) {
//【1】先将一个数组转换为堆结构,但是数组的初始状态实际上就对应了初始的堆的状态,不需要转换
//【4】实现一个循环,每一次都表示一个元素有序,每一次都将参与堆排序的数组最大下标向前提一个
for(int end=arr.length-1;end>0;end--) {
//【2】将对结构不断的调整为大顶堆
adjustHeap(arr,end);
//【3】将堆顶元素(此时表示的是堆中最大的元素)和堆的最后一个叶子节点进行交换,表示这个元素已经有序
int tmp=arr[0];
arr[0]=arr[end];
arr[end]=tmp;
}
}
//将一个数组(二叉树),调整成一个大顶堆
/**
* @param arr 构建大顶堆的数组
* @param end 参与构建大顶堆数组元素的最大下标(相当于堆中的最后一个节点(叶子节点)在数组array中的对应下标)
*/
public static void adjustHeap(int arr[],int end) {
//【1】根据公式计算出堆结构中最后一个父节点的下标
//公式:lastFather=[(start+end)/2]↑-1
int lastFather=(0+end)%2!=0? (0+end)/2 :(0+end)/2-1;
//【5】从最后一个父节点 开始想前-1,使得每一个父节点都能实现左右pk,以下犯上的步骤
for(int father=lastFather;father>=0;father--) {
//【2】根据父节点的下标推算出左右孩子的下标:左孩子:2n+1 右孩子:2n+2
int left=father*2+1;
int right=father*2+2;
//【3】在保证右孩子下标没有越界的情况下,使用右孩子和父节点进行比较,如果右孩子>父节点,进行交换
if(right<=end&&arr[right]>arr[father]) {
int tmp=arr[right];
arr[right]=arr[father];
arr[father]=tmp;
}
//【4】使用左节点和父节点进行交换,如果左孩子>父节点,进行交换
if(arr[left]>arr[father]) {
int tmp=arr[left];
arr[left]=arr[father];
arr[father]=tmp;
}
}
}
}