堆排序是利用堆的性质进行的一种选择排序。下面先讨论一下堆。
1.堆
堆实际上是一棵完全二叉树,其任何一非叶节点满足性质:
Key[i]<=key[2i+1]&&Key[i]<=key[2i+2]或者Key[i]>=Key[2i+1]&&key>=key[2i+2]
即任何一非叶节点的关键字不大于或者不小于其左右孩子节点的关键字。
堆分为大顶堆和小顶堆,满足Key[i]>=Key[2i+1]&&key>=key[2i+2]称为大顶堆,满足 Key[i]<=key[2i+1]&&Key[i]<=key[2i+2]称为小顶堆。由上述性质可知大顶堆的堆顶的关键字肯定是所有关键字中最大的,小顶堆的堆顶的关键字是所有关键字中最小的。
2.堆排序的思想
利用大顶堆(小顶堆)堆顶记录的是最大关键字(最小关键字)这一特性,使得每次从无序中选择最大记录(最小记录)变得简单。
其基本思想为(大顶堆):
1)将初始待排序关键字序列(R1,R2....Rn)构建成大顶堆,此堆为初始的无须区;
2)将堆顶元素R[1]与最后一个元素R[n]交换,此时得到新的无序区(R1,R2,......Rn-1)和新的有序区(Rn),且满足R[1,2...n-1]<=R[n];
3)由于交换后新的堆顶R[1]可能违反堆的性质,因此需要对当前无序区(R1,R2,......Rn-1)调整为新堆,然后再次将R[1]与无序区最后一个元素交换,得到新的无序区(R1,R2....Rn-2)和新的有序区(Rn-1,Rn)。不断重复此过程直到有序区的元素个数为n-1,则整个排序过程完成。
操作过程如下:
1)初始化堆:将R[1..n]构造为堆;
2)将当前无序区的堆顶元素R[1]同该区间的最后一个记录交换,然后将新的无序区调整为新的堆。
因此对于堆排序,最重要的两个操作就是构造初始堆和调整堆,其实构造初始堆事实上也是调整堆的过程,只不过构造初始堆是对所有的非叶节点都进行调整。
下面举例说明:
给定一个整形数组a[]={16,7,3,20,17,8},对其进行堆排序。
首先根据该数组元素构建一个完全二叉树,得到
,这样就得到了初始堆。
此时3位于堆顶不满堆的性质,则需调整继续调整
public class Heap <D extends Comparable> {
private java.util.ArrayList<D> list=new java.util.ArrayList<D>();
public Heap()//默认构造方法
{
}
public Heap(D[] objects)//将数组赋值添加到堆中
{
for(int i=0;i<objects.length;i++)
add(objects[i]);
}
public void add(D newObject)//给堆添加一个新节点
{
list.add(newObject);//添加到链表末尾
int currentIndex=list.size()-1;
while(currentIndex>0)
{
int parantIndex=(currentIndex-1)/2;//不管currentIndex为左节点还是右节点,整数除法向下取整得到父节点
//swap if the current object is greater than the last node
if(list.get(currentIndex).compareTo(list.get(parantIndex))>0)//比父节点大就与父节点交换
{
D temp=list.get(currentIndex);
list.set(currentIndex, list.get(parantIndex));
list.set(parantIndex, temp);
}
else
break;
currentIndex=parantIndex;//当前节点变为自己的父节点,继续比较
}
}
public D remove()//从堆中删除根节点(经常需要删除堆中最大的元素,即根节点)
{
if(list.size()==0)
return null;
D removedObject=list.get(0);//根节点在ArrayList中的首元素
list.set(0, list.get(list.size()-1));//将末元素放在根节点位置(覆盖了原来的根节点值)
list.remove(list.size()-1);//删除末节点,相当于是从根节点往下删除
int currentIndex=0;
while(currentIndex<list.size())
{
int leftChildIndex=2*currentIndex+1;
int rightChildIndex=2*currentIndex+2;
if(leftChildIndex>=list.size()) break;
int maxIndex=leftChildIndex;
if(rightChildIndex<list.size())
{
if(list.get(maxIndex).compareTo(list.get(rightChildIndex))<0)
{
maxIndex=rightChildIndex;
}
}
//如果当前节点比孩子小就要交换
if(list.get(currentIndex).compareTo(list.get(maxIndex))<0)
{D temp=list.get(maxIndex);
list.set(maxIndex, list.get(currentIndex));
list.set(currentIndex, temp);
currentIndex=maxIndex;
}
else
break;
}
return removedObject;
}
public int getSize(){
return list.size();
}
}
public class HeapSort {
public static <E extends Comparable> void heapSort(E[] list)
{
Heap<E> heap=new Heap<E>();
for(int i=0;i<list.length;i++)
heap.add(list[i]);
for(int i=list.length-1;i>=0;i--)
list[i]=heap.remove();
}
public static void main(String[] args) {
Integer[] list={2,3,2,5,6,1,-2,3,14,12};
heapSort(list);
for(int i=0;i<list.length;i++)
System.out.print(list[i]+" ");
}
}