算法系列—堆排序

堆排序可以分为两个阶段

堆得构造阶段，我们将原始数组重新组织安排进一个堆中

下沉排序阶段，从堆中按顺序取出所有元素并得到排序结果

PS：此处堆排序，数组下标是从1开始，0位置不存储。

#include <iostream>
#include <algorithm>

void sink(int A[], int k, int size);

void HeapSort(int A[], int size)
{
	int N = size - 1;	//N保存数组元素个数，减1是因为A[0]不计入计算
	for(int k=N/2;k>=1;k--)		//构造堆
	{
		sink(A, k, N);
	}
	while (N>1)		//排序
	{
		std::swap(A[1], A[N--]);
		sink(A, 1, N);
	}
}

void sink(int A[], int k, int N)
{
	while(2*k<=N)
	{
		int j = 2 * k;
		if (j < N && A[j] < A[j + 1])
			j++;
		if (A[k] < A[j])
			std::swap(A[k], A[j]);
		k = j;
	}
}


int main()
{
	int A[] = { 0,1,4,6,4,7,4,10,50,88,2,33,56,3,345,66,67,4 };

	HeapSort(A, sizeof(A) / 4);
	for (int i=1;i<sizeof(A) / 4;i++)
	{
		std::cout << A[i] << " ";
	}
}