算法导论第六章 堆排序C++源码（附图）

#include <iostream> using namespace std; void swap(int array[] , int i , int j) { /*array[i] = array[i]+array[j]; array[j] = array[i]-array[j]; array[i] = array[i]-array[j];*/ int temp; temp = array[i]; array[i] = array[j]; array[j] = temp; } void MaxHeapify(int array[] , int i , int hSize) { int l = 2*i; int r = 2*i+1; int largest; if(l<=hSize && array[l]>array[i]) { largest = l; } else { largest = i; } if(r<=hSize && array[r]>array[largest]) { largest = r; } if(largest != i) { swap(array , i , largest); MaxHeapify(array , largest , hSize); } } void BuildMaxHeap(int array[] , int hSize) { for(int i=hSize/2 ; i>=1 ; i--) { MaxHeapify(array , i , hSize); } } void HeapSort(int array[] , int length) { int hSize = length; BuildMaxHeap(array , hSize); for(int i=length ; i>=2 ; i--) { swap(array , 1 , i); hSize--; MaxHeapify(array , 1 , hSize); } } int main() { int a[99]; for(int i=1 ; i<99 ; i++) { a[i] = rand()%100; } HeapSort(a,99); for(int i=1 ; i<99 ; i++) { cout << a[i] << ' '; } return 0; }

MaxHeapify(A,2):2号节点与它的子节点中较大者换位置，一直到最下一层或子节点都比它小。

 

BuildMaxHeap:从length[A]/2节点开始一直到1号节点，调用MaxHeapify，算法结束时，形成最大堆。最大堆的特点是任意节点都比它的两个子节点的值大。

 

HeapSort:主要算法。直接调用BuildMaxHeap,BuildMaxHeap调用Heapify建立最大堆。将堆顶元素即最大元素与最后一个元素换位置，数组长度-1，这样原堆顶元素就被放到了最后的位置。之后，用Max-Heapify实现重建堆。