堆排序的思路:
我们将数组看作一棵树,(顺序存储二叉树),
数组的第n个元素的左子节点下标为 2n+1
数组第n个元素的右子节点下标为 2n+2
数组第n个元素的父节点下标为(n-1)/2
从第一个不为叶子节点开始,判断其和其子节点大小,若子节点比其本身大,则将子节点与其交换。
然后找到该节点上一个节点进行相同操作(在数组中的位置)。
直到找到根节点,即把最大的元素推上了根节点的位置。
将根节点与末尾节点交换,将最大的元素放在数组末尾。下一次不再遍历
图解
代码实现
public class HeapSort {
public static void main(String[] args) {
int arr[]={5,1,3,6,0,2,94,58};
HeapSort(arr);
System.out.println(Arrays.toString(arr));
}
public static void HeapSort(int arr[]){
for(int i=arr.length-1;i>=1;i--){//i代表每次循环的最后一个叶子节点
// 从第一个非叶子节点开始遍历遍历,直到最顶点,将最大的元素放在最上面,每进行一次,求得一次大顶堆树
for(int j=(i-1)/2;j>=0;j--){
maxHeap(arr,j,i);
}
swap(arr,i);
}}
public static void maxHeap(int arr[],int index,int size){//将该节点与其左右子节点比较,将最大的节点拿到最上面,size是该函数对数组操作的范围,除去放在数组末尾已经排好序的节点。
// 对比其左右子节点
int left=2*index+1;
int right=2*index+2;
int max=index;
// 左右节点不能是已经排好顺序的数组
if(left<=size&&arr[left]>arr[max]){
max=left;
}
if(right<=size&&arr[right]>arr[max]){
max=right;
}
if (max!=index){
//找出该节点与其左右子节点最大的点,交换
int temp=arr[index];
arr[index]=arr[max];
arr[max]=temp;
}
}
public static int[] swap(int arr[],int last){//将数组第一个数字换到最后位置,即把大顶堆树的根放到最后一个叶子节点的位置
if(arr==null){return null;}
if(last==0){return arr;}
int temp=arr[0];
arr[0]=arr[last];
arr[last]=temp;
return arr;
}
}
堆排序算法让我明白了Debug的重要性。