堆排序C++源代码

1. // HeapSort.cpp : Defines the entry point for the console application.
2. //
4. #include "stdafx.h"
5. #include <stdlib.h>
6. #include <iostream>
7. #include <time.h>
8. using namespace std;
10. //调整大根堆
11. void HeapAdjust(int* pnData, int nStart, int nLen)
12. {
13.     int nMaxChild = 0;
14.     int nTemp = 0;
16.     while ( ( 2 * nStart + 1) < nLen)
17.     {
18.         nMaxChild = 2 * nStart + 1;
19.         if ( (2 * nStart + 2) < nLen)
20.         {
21.             //比较左子树和右子树,记录最大值的Index
22.             if (pnData[2 * nStart + 1] < pnData[2 * nStart + 2])
23.             {
24.                 nMaxChild = 2 * nStart + 2;
25.             }
26.         }
28.         //change data
29.         if (pnData[nStart] < pnData[nMaxChild])
30.         {
31.             //交换nStart与nMaxChild的数据
32.             nTemp = pnData[nStart];
33.             pnData[nStart] = pnData[nMaxChild];
34.             pnData[nMaxChild] = nTemp;
36.             //堆被破坏,需要重新调整
37.             nStart = nMaxChild;
38.         }
39.         else
40.         {
41.             //比较左右孩子均大则堆未破坏,不再需要调整
42.             break;
43.         }
44.     }
45. }
47. //堆排序
48. void HeapSort(int* pnData, int nLen)
49. {
50.     int i = 0;
51.     int nTemp = 0;
53.     //将pnData[0, Len-1]建成大根堆
54.     for (i = nLen / 2  - 1; i >= 0; i--)
55.     {
56.         ::HeapAdjust(pnData, i, nLen);
57.     }
59.     for (i = nLen - 1; i > 0; i--)
60.     {
61.         nTemp = pnData[0];
62.         pnData[0] = pnData[i];
63.         pnData[i] = nTemp;
65.         //将pnData[0...i]重写建成大根堆
66.         ::HeapAdjust(pnData, 0, i);
67.     }
68. }
70. int main()
71. {
72.     int a[10] = {0};
73.     int i = 0;
75.     srand((unsigned)time(NULL));
77.     cout << "the orign data:" << endl;
78.     for (i = 0; i < 10; i++)
79.     {
80.         a[i] = ::rand(); 
81.         cout << a[i] << "/t";
82.     }
84.     cout << endl;
86.     HeapSort(a, 10);
88.     cout << "after sort data" << endl;
89.     for (i = 0; i < 10; i++)
90.     {
91.         cout << a[i] << "/t";
92.     }
94.     cout << endl;
95. }