不需要额外空间可对数组排序,把维持在待排序的数组内.在建堆时,采用自顶向下堆化的方法.堆中的元素向下移动,找到该元素的左孩子,如果左孩子存在的话,就比较左孩子与右孩子的大小,找出其中较大的孩子,与父元素比较,如果父元素较大,则循环停止,否则两元素交换.如此即可完成自顶向下的堆化.在堆排序时,先用一个for循环调用建堆函数来建堆,然后用while循环来将数组中最大的元素找到,再将其放到最后一个位置上,待排序元素减一,直到待排序元素小于1,排序完成,循环结束.
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <stack>
#include <queue>
#include <malloc.h>
using namespace std;
#define OK 1
#define ERROR -1
#define TRUE 1
#define FALSE 0
typedef int Status;
// 12 5 2 3 7 6 1 4 11 8 10 9
//输出函数
void Print(int a[], int l, int r)
{
int i;
for(i = l; i <= r; i++)
{
printf("%d ", a[i]);
}
printf("\n");
}
自顶向下堆化
void fix_down(int a[], int k, int N)
{
int j;
while(2*k <= N)
{
//取j为k的子节点
j = 2*k;
//如果左孩子小于右孩子,则选右孩子与父节点比较
if(j < N && a[j] < a[j+1])
{
j++;
}
//如果父节点比两个孩子都大,则循环结束
if(a[k] >= a[j])
{
break;
}
//不然就交换
swap(a[k], a[j]);
//堆继续向下走
k = j;
}
}
//堆排序函数
void heap_sort(int a[], int l, int r)
{
int i, N = r-l+1;
int* pq = a + l - 1;
//pq = a + l - 1;
//for循环建堆
for(i = N/2; i >= 1; i--)
{
fix_down(pq, i, N);
}
//while循环使得数组中第一个元素最大,交换第一个元素与最后一个元素,
//使最大元素处于正确的位置上,直到待排序元素个数为0
while(N > 1)
{
swap(pq[1], pq[N]);
fix_down(pq, 1, --N);
}
}
int main()
{
int a[12] = {2, 5, 3, 7, 6, 1, 4, 11, 8, 10, 9, 12};
//堆排序
heap_sort(a, 0, 11);
Print(a, 0, 11);
return 0;
}