- 简介
堆排序是利用堆这种数据结构进行的排序。堆通常使用一维数组来实现,是一种近似完全二叉树的结构。堆排序分为大根堆和小根堆。堆排序满足这样的一个特性:大根堆父节点的值总是大于子节点,小根堆的父节点值总是小于子节点。
- 思想
我们在拿到一个数组时,首先将它构造成为一个大根堆/小根堆,这个过程我们叫做建堆。然后将根节点元素和最末尾元素交换,此时最末尾成为有序区。除去最末尾元素的剩下元素所构成的树此时已经不叫堆了,因为它不满足堆的特性。所以我们继续进行调整,使得最小(或最大)的值再次移到根节点,随后交换它与此时堆中的末尾节点,有序区+1...如此反复,直至堆的大小为1(无序区中只有一个数字,代表排序完成)。
- 具体代码(升序,数组实现)
首先我们写出堆排序的主体:
void HeapSort(int arr[],int len)
{
int heap_size = len;
BuildHeap(arr,len);
while(heap_size > 1)
{
change(arr,0,--heap_size);
HeapAdjust(arr,0,heap_size);
}
}
上面这个HeapSort函数就是我们的堆排序的主体部分,首先根据实参传入的数组长度决定堆的大小,进行建堆(BuildHeap函数将在下面书写)。建堆函数要完成的功能就是将一个无序的数组调整成为一个大根堆或者小根堆。建堆完成之后,如果heap_size(无序区)中存留的数字大于1,即这个堆中还有超过一个以上的数字那么证明还需要继续排序。进入循环,不论是刚建成一个堆还是已经经过调整,我们首先都需要将arr[0]这个根节点的元素和当前的堆的末尾元素进行交换,交换完成之后将它放入有序区,即heapsize--,然后对交换过后已经有可能被破坏的堆进行重新调整,重复选出数字放入有序区的这个过程。
接下来我们编写建堆函数,此处要求升序排序我们需要建立大根堆。
void BuildHeap(int arr[],int len)
{
int heap_size = len;
for(int i = heap_size/2 - 1;i > 0; i--)
{
HeapAdjust(arr,i,,heap_size);
}
}
建立堆的第一步就是从二叉树的倒数第一个非叶子节点(len/2-1)开始,从下至上对每一个根节点进行调整。
最后编写调整函数。它只需要注意处理上层节点调整之后下层节点同样有可能需要调整的情况,可以使用递归解决。其余只需要实现简单的数据交换即可。
void HeapAdjust(int arr[],int i,int len)
{
int leftchild = 2*i + 1;
int rightchild = 2*i + 2;
int max = i; //代表从i节点开始进行调整,有可能需要继续向下调整
if(leftchild < len && arr[max] < arr[leftchild])
max = leftchild;
if(rightchild < len && arr[max] < arr[rightchild])
max = rightchild;
if(i != max)
{
change(arr,i,max);
HeapAdjust(arr,max,len);
}
}
参数中的i就代表从i节点开始进行调整,经过数据交换后(如果有这个必要的话),将继续下探(因为如果这个和它的父节点交换过的节点有子节点的话,交换之后可能不满足堆的定义,所以需要下探调整。)直至调整完成,此时又是一个新的堆。