堆排序

// HeapSort.cpp : Defines the entry point for the console application.
//

#include "stdafx.h"
#include "stdio.h"
#include "iostream"

using namespace std;

void HeapAdjust(int arr[], int start, int end);
void HeapSort(int arr[], int start, int end);


int _tmain(int argc, _TCHAR* argv[])
{
	int arr[] = {0, 7, 8 , 9, 1, 2, 3, 5, 10, 23, 6};
	HeapSort(arr, 1, 10);
	for(int i = 10; i>=1; i--)
		printf("%5d", arr[i]);
	printf("\n");
	system("pause");
	return 0;
}

// Heap Adjust
void HeapAdjust(int arr[], int start, int end)
{
	int rloc = arr[start];
	for( int j = 2 * start; j <= end; j = j * 2 )
	{
		if( j < end && arr[j+1] < arr[j]) // 存在右节点
			j++; // 指向较小的子节点
		if( arr[j] >= rloc) // 不需要再进行移动
			break;
		arr[start] = arr[j];
		start = j;
	}
	arr[start] = rloc;
}

// Heap Sort
// 堆排序的时间复杂度 nlogn, 空间复杂度 O(1)
// 堆排序的最好情况、最差情况的时间复杂度都为O(nlogn)
void HeapSort(int arr[], int start, int end)
{
	// 建立堆，由[n/2]开始逐个向下调整
	for( int i = end/2; i >= start; i = i-1)
		HeapAdjust(arr, i, end);


	// 输出堆元素，每次输出都将堆顶元素与最后一个元素互换，然后进行调整
	// 也可以不这样做，只是把堆顶元素移走，然后将最后一个元素移动到堆顶，进行调整
	// 这里将堆顶的元素与每次堆排序的最后一个互换，为了节省空间,将排好序的元素由原数组的末尾开始排放
	for( int i = start; i <= end; i++)
	{
		int temp = arr[1];
		arr[1] = arr[end - i + 1];
		arr[end - i + 1] = temp;

		// 重新调整堆
		HeapAdjust(arr, 1, end - i);
	}
}