public class HeapSort {
public static void main(String[] args) {
int[] arr = new int[]{9,6,8,7,0,1,10,4,2};
heapSort(arr);
System.out.println(Arrays.toString(arr));
}
public static void heapSort(int[] arr){
//开始位置时最后一个节点的父节点
int start = (arr.length-1)/2;
//调整为大顶堆
for (int i=start;i>=0;i--){
maxHeap(arr,arr.length,i);
}
//先把数组中的第0个和堆中的最后一个数交换位置,再把前面的t调整为大顶堆
for (int i=arr.length-1;i>=0;i--){
int temp = arr[i];
arr[i] = arr[0];
arr[0] = temp;
maxHeap(arr,i,0);
}
}
//大顶堆
public static void maxHeap(int[] arr,int size,int index){
//左子节点
int leftNode = 2index+1;
//右子节点
int rightNode = 2index+2;
int max = index;
//和两个节点分别比较,找出最大的节点
if (leftNode<size&&arr[leftNode]>arr[max]){
max=leftNode;
}
if (rightNode<size&&arr[rightNode]>arr[max]){
max=rightNode;
}
//交换位置
if (max!=index){
int temp = arr[index];
arr[index] = arr[max];
arr[max] = temp;
//交换位置后该节点还要和自己的子节点比较,保证是大顶堆
maxHeap(arr,size,max);
}
}
}