一、堆排序简介
堆是一种数据结构,长的和二叉树一个样,所有根结点都大于等于左右子结点的堆叫大顶堆,反之,叫小顶堆,堆是一个完全二叉树。
对于某个结点i,它的父节点是(i-1)/2,它的左孩子节点是2i+1,它的右孩子节点是2i+2。
堆是一种逻辑结构,实际在内存中是以数组的形式存储的。
二、堆排序的代码实现
思路:
(1)根据初始数组构造一个初始堆;
(2)每次交换第一个和最后一个元素,输出最后一个元素(最大值),然后把剩下元素重新调整为大顶堆,再进行交换第一个元素和最后一个元素,再调整大顶堆,重复执行,直到整个数组排序完成。
java代码:
package sort;
import java.util.Arrays;
/*
* 堆排序是不稳定的排序算法
* 时间复杂度:初始建堆的时候是O(n),排序重建堆的时间复杂度是nlog(2^n)
* 所以总的时间复杂度是O(n+nlog2^n) = O(nlog2^n)
*
*/
public class HeapSort {
public static void main(String[] args){
int[] arr = {27,46,12,33,49,27,36,40,42,50,51};
heapSort(arr);
System.out.println(Arrays.toString(arr));
}
public static void heapSort(int[] arr){
//初始建堆
for(int i=arr.length/2-1;i>=0;i--)
{
heapify(arr,i,arr.length-1);//写一个调整堆结构的方法
}
//堆排序的过程
for(int i=arr.length-1;i>0;i--){ //从最后一个节点开始
//交换当前节点和堆顶节点
swap(arr,0,i); //最后沉底的元素在下一次交换时不在堆里边
heapify(arr,0,i-1);
}
}
public static void heapify(int[] arr,int i,int last_index){
int max = i;
if(2*i+1<=last_index && arr[2*i+1]>arr[max]){ //有左孩子
max = 2*i+1; //记左节点为最大值
}
if(2*i+2<=last_index && arr[2*i+2]>arr[max]){ //有右孩子
max = 2*i+2; //记右节点为最大值
}
if(max!=i){ //max!=i,说明左或右节点的值大于父节点(由前两个if语句可知),需要交换
//写一个交换的方法
swap(arr,i,max);
heapify(arr,max,last_index); //递归地对那个节点进行堆排序
}
}
public static void swap(int[] arr,int i,int j){
int temp = arr[i];
arr[i] = arr[j];
arr[j] = temp;
}
}