堆排序利用了大根堆(或小根堆)堆顶记录的关键字最大(或最小)这一特征,使得在当前无序区中选取最大(或最小)关键字的记录变得简单。
(1)用最大堆排序的基本思想
① 先将初始文件R[1..n]建成一个最大堆,此堆为初始的无序区;
② 再将关键字最大的记录R[1](即堆顶)和无序区的最后一个记录R[n]交换,由此得到新的无序区R[1..n-1]和有序区R[n],且满足R[1..n-1].keys≤R[n].key;
③由于交换后新的根R[1]可能违反堆性质,故应将当前无序区R[1..n-1]调整为最大堆,然后再次将R[1..n-1]中关键字最大的记录R[1]和该区间的最后一个记录R[n-1]交换,由此得到新的无序区R[1..n-2]和有序区R[n-1..n],且仍满足关系R[1..n-2].keys≤R[n-1..n].keys,同样要将R[1..n-2]调整为最大堆……直到无序区只有一个元素为止。
(2)最大堆排序算法的基本操作:
① 初始化操作:将R[1..n]构造为初始堆;
② 每一趟排序的基本操作:将当前无序区的堆顶记录R[1]和该区间的最后一个记录交换,然后将新的无序区调整为堆(亦称重建堆)。
注意:
①只需做n-1趟排序,选出较大的n-1个关键字即可以使得文件递增有序;
②用小根堆排序与利用大根堆类似,只不过其排序结果是递减有序的。堆排序和直接选择排序相反:在任何时刻堆排序中无序区总是在有序区之前,且有序区是在原向量的尾部由后往前逐步扩大至整个向量为止。
#include <iostream>
#include <vector>
#include <iterator>
using namespace std;
template <class Comparable>
void heapSort(vector<Comparable> &data)//排序过程
{
int size = data.size();
for(int i = size/2; i >= 0; --i)
{
greatHeap(data, i, size);
}
for(int j = size-1; j > 0; --j)
{
/*每次将最大的元素(最大堆的第一个元素)放到最后,实现升序排序*/
Comparable temp = data[0];
data[0] = data[j];
data[j] = temp;
/*调整第一个元素,使其重新符合最大堆*/
greatHeap(data, 0, j);
}
}
int leftChild(int i)//返回下标为i的结点的左孩子的下标
{
return i*2+1;
}
template <class Comparable>
void greatHeap(vector<Comparable> &data, int i, int size)//调整数组中下标为i的结点,使其符合最大堆
{
int child = i;//表示下标
Comparable temp;
for(temp = data[i]; leftChild(i) < size; i = child)
{
child = leftChild(i);
if(child != size-1 && data[child] < data[child+1])
{
child++;
}
if(temp > data[child])
{
break;
}
else
{
data[i] = data[child];
}
}
data[i] = temp;
}
int main(void)
{
int ary[] = {1,7,2,6,3,5,4,8,9,0};
int len = sizeof(ary)/sizeof(int);
vector<int> v(ary, ary+len);
heapSort(v);
copy(v.begin(), v.end(), ostream_iterator<int>(cout, " "));
cout << endl;
return 0;
}