經典排序算法——堆排序

對於一個int數組，請編寫一個堆排序算法，對數組元素排序。

給定一個int數組A及數組的大小n，請返回排序後的數組。

測試樣例：

[1,2,3,5,2,3],6

[1,2,2,3,3,5]

class HeapSort {
public:
	int* heapSort(int* A, int n) {
        
		BuildMaxHeap(A, n);//首先將數組A構建成大頂堆
        
		for (int i = n - 1; i >= 1; i--)
		{
            swap(A[0],A[i]);//將堆頂最大值與未排序的最後一個數交換
			MaxHeapFix(A, 0, i-1);//將A[0,i-1]重新調整爲大頂堆
		}
		return A;
	}
    
    //構建大頂堆
	void BuildMaxHeap(int *A, int n)
	{
		for (int i = n / 2 - 1; i >= 0; i--)
			MaxHeapFix(A, i, n-1);
	}
    
    //重新調整爲大頂堆
	void MaxHeapFix(int *A, int i, int n)
	{
		int key = A[i];
		for(int j=2*i;j<n;j*=2)
		{
			if (j<n && A[j]<A[j+1])//找出左右子樹中較大
				++j;
			if (key>=A[j])//根節點key與左右子樹中較大值比較，若key較大，則直接跳出本次循環；若key較小，則進行交換
				break;
			A[i] = A[j];
			i=j;
		}
		A[i] = key;
	}
};