堆排序
堆根据其排列特点分为大顶堆和小顶堆。顾名思义,大顶堆是指对于任何一个子树而言,父节
点永远大于它的任何一个子节点。那相反,小顶堆是指对于任何一个子树而言,父节点永远小于它
的任何一个子节点。如下如所示:
我们依次对树中的节点进行编号,映射到数组中就是下面这个样子:
堆排序基本思路:首先,我们将n个元素经过比较且交换位置按关键码建成大顶堆,此时堆顶的元
素(即此时树中的根节点)是最大的,然后我们将堆顶的元素与最后一个节点的元素位置进行交换,
并将此时的最后一个元素放出去,然后在按关键码建成大顶堆,然后再将堆顶的元素与最后一个节点
位置的元素进行交换,依次反复,可以得到一个按关键码有序的序列。这个过程就称之为堆排序。
代码展示
package demo4;
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;
//调整为大顶堆
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);
}
}
}