C++实现堆排序,包括调整堆的过程,以及排序过程。
1、写好从某个结点sink下沉的子函数Heap_build
2、建堆,从最后一个非叶子结点开始,逐个结点调用sink函数,直到完成根节点的sink下沉的过程
3、此时根节点为堆内最大值,让其与最后一个叶子结点调换位置,并从堆中去掉最后一个值(最大值)
4、对新的根节点调用sink函数
5、重复步骤3和4,直到堆内只有一个根节点,排序完成
代码:
#include <iostream>
#include <algorithm>
#include <vector>
#include <time.h>
using namespace std;
//使用指针变量进行交换
void Swap2(int* a, int* b)
{
int temp = *a;
*a = *b;
*b = temp;
}
//使用引用进行交换
void Swap(int& a, int& b)
{
int temp = a;
a = b;
b = temp;
}
//建堆(大顶堆)
void Heap_build(int a[], int root, int length)
{
int lchild = root*2 + 1;
if(lchild < length)
{
int flag = lchild;
int rchild = lchild + 1;
if(rchild < length)
{
if(a[rchild] > a[flag]) //找出左右子结点中的最大值
{
flag = rchild;
}
}
if(a[root] < a[flag])
{
//交换父结点和比父节点大的最大子结点
Swap(a[root], a[flag]);
//从此最大子结点的位置开始递归建堆
Heap_build(a, flag, length);
}
}
}
void Heap_sort(int a[], int len)
{
//从最后一个非叶子结点的父结点开始建堆
for(int i = len/2 - 1; i >= 0; --i)
{
Heap_build(a, i, len);
}
//j表示数组此时的长度,因为len长度已经建过了,从len-1开始
for(int j = len-1; j > 0; --j)
{
//交换首尾元素,将最大值放入到最后位置保存(升序排列)
Swap(a[0], a[j]);
//去除最后位置元素,重新建堆
Heap_build(a, 0, j);
}
}
int main()
{
clock_t Start_time = clock();
int a[10] = {12,45,748,12,56,3,89,4,48,2};
Heap_sort(a, 10);
for(int i = 0; i < 10; ++i)
{
cout << a[i] << " ";
}
clock_t End_time = clock();
cout << endl;
cout << "Total running time is: "
<< (double)(End_time - Start_time) / CLOCKS_PER_SEC * 1000
<< "ms"
<< endl;
return 0;
}