对于一个长度为n的数组R(n)={r[0],r[1],…,r[n-1]}。
大顶堆(最大堆):r(i)>=r(2i+1)且r(i)>=r(2i+2),堆顶元素必为最大值,用于升序;
小顶堆(最小堆):r(i)<=r(2i+1)且r(i)<=r(2i+2),堆顶元素必为最小值,用于降序。
堆排序主要包括两个过程:(1)构建堆(从最后一个非叶子节点开始,即序号为n/2-1的节点);(2)交换堆顶元素和最后一个元素的位置。
堆排序的时间复杂度:最好情况、平均情况和最坏情况的时间复杂度都为O(nlogn);
堆排序的空间复杂度:O(1);
堆排序是不稳定的排序算法,适用于n大时较好。
Java代码实现大顶堆(最大堆):
public class MaxHeapDemo
{
//调整该完全二叉树,构建大顶堆
public void adjustMaxHeap(int[] arr,int pos,int length)
{
int temp;
int child;
for(temp = arr[pos];2 * pos + 1 <= length;pos = child)
{
child = 2 * pos + 1;//父结点pos的左子节点
if(child < length && arr[child] < arr[child + 1])
child++;
if(arr[child] > temp)
arr[pos] = arr[child];
else
break;//结束循环
}
arr[pos] = temp;
}
public void constructMaxHeap(int[] arr)
{
int len = arr.length;
//从最后一个非叶子节点开始调整该完全二叉树,构建大顶堆
for(int i = (len/2 - 1);i >= 0;i--)
{
adjustMaxHeap(arr,i,len - 1);
}
//交换堆顶元素和最后一个元素的位置
for(int i = len - 1;i > 0;i--)
{
int tmp = arr[i];
arr[i] = arr[0];
arr[0] = tmp;
//从第一个非叶子节点(根节点)开始调整大顶堆
adjustMaxHeap(arr,0,i - 1);
}
}
public void print(int[] arr)
{
for(int i = 0;i < arr.length;i++)
{
System.out.print(arr[i] + " ");
}
System.out.println();//换行
}
public static void main(String[] args)
{
int[] arr = {4,5,1,6,2,7,3,8};
MaxHeapDemo d = new MaxHeapDemo();
d.constructMaxHeap(arr);
d.print(arr);
}
}