堆排序
package com.bjsxt.HeapSort;
import java.util.Arrays;
//堆排序
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;
//结束位置,数据长度-1
for(int i=start;i>=0;i--){
maxHeap(arr,arr.length,i);
}
//先把数组中的第0个和堆中的最后一个数交换位置,再把前面的处理为大顶堆
for(int i=arr.length-1;i>0;i--){
int temp = arr[0];
arr[0]=arr[i];
arr[i]=temp;
maxHeap(arr,i,0);
}
}
public static void maxHeap(int[] arr,int size,int index){
//左子节点
int leftNode = 2*index+1;
//右子节点
int rightNode = 2*index+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);
}
}
}
测试结果
[0, 1, 2, 4, 6, 7, 8, 9, 10]