HeapSort,use maximun heap...

1. /*
2. 建议大家到这看:
3. http://topic.csdn.net/u/20081129/21/81fb7e89-0df5-4911-9884-d05680705a3c.html
6. */
8. #include<iostream> 
9. #include<ctime> 
10. using namespace std;
12. const int DefaultSize = 100;
14. class MaxHeap
15. {
16. public:
17.     MaxHeap(int sz);
18.     int Parent(int i) { return i/2;}
19.     int Left(int i) { return 2*i;}
20.     int Right(int i) {return 2*i+1;}
21.     void MaxHeapify(int* arr,int i);
22.     void BuildMaxHeap(int* arr,int arrSize);
23.     void HeapSort(int* arr,int arrSize);
24. private:
25.     //int* maxHeap; 
26.     int currentSize;  //maxHeapSize; 
27.     void Swap(int &i,int &j) {int t=i; i=j; j=t;}
28. };
29. MaxHeap::MaxHeap(int arrSize)
30. {
31.     currentSize = arrSize;
32. }
34. void MaxHeap::MaxHeapify(int *arr,int i)
35. {
36.     int l = Left(i), r = Right(i),largest;   //left,right,largest 
37.     if (l<=currentSize && arr[l]>arr[i]) largest = l;
38.     else largest = i;
39.     if (r<=currentSize && arr[r]>arr[largest]) largest = r;
40.     if (largest != i)
41.     {
42.         Swap(arr[i],arr[largest]);
43.         MaxHeapify(arr,largest);
44.     }
45. }
46. void MaxHeap::BuildMaxHeap(int* arr,int arrSize)
47. {
48.     for (int i=arrSize/2;i>=0;i--)
49.        MaxHeapify(arr,i);
50. }
51. void MaxHeap::HeapSort(int* arr,int arrSize)
52. {
53.     BuildMaxHeap(arr,arrSize);
54.     //一开始我写成for (int i=arrSize/2;i>0;i--),害我对着书画图才发现这个错误,晕.. 
55.     for (int i=arrSize;i>0;i--)  //最后只有一个元素,不用比较了 
56.     {
57.         Swap(arr[0],arr[i]);    //以arr[0]作为最大堆的根 
58.         currentSize--;          //堆的大小减1 
59.         MaxHeapify(arr,0); 
60.     }
61. }
62. int main(void)
63. {
64.     srand(unsigned(time(NULL)));
65.     int Size = DefaultSize;
66.     int *array = new int[Size];
67.     for(int i=0;i<Size;i++)
68.         array[i] = rand()%1000;
69.     cout<<"before sort"<<endl;
70.     for(int i=0;i<Size;i++)
71.     {
72.         cout<<array[i]<<" ";
73.         if((i+1)%10 == 0)cout<<endl;
74.     }
75.     MaxHeap maxHeap(Size-1);   //这里应为Size-1,因为传Size的话就数组越界了.. 
76.     maxHeap.HeapSort(array,Size-1);
77.     cout<<endl<<"After sort"<<endl;
78.     for(int i=0;i<Size;i++)
79.     {
80.         cout<<array[i]<<" ";
81.         if((i+1)%10 == 0)cout<<endl;
82.     }
83.     delete [] array;
84.     return 0;
85. }