堆排序（heapsort）是一种比较快速的排序方式，它的时间复杂度为O(nlgn)，并且堆排序具有空间原址性，任何时候只需要有限的空间来存储临时数据。我将用c++实现一个堆来简单分析一下。
堆排序的基本思想为：
1、升序排列，保持大堆；降序排列，保持小堆；
2、建立堆之后，将堆顶数据与堆中最后一个数据交换，堆大小减一，然后向下调整；直到堆中只剩下一个有效值；
下面我将简单分析一下：
第一步建立堆：
1、我用vector顺序表表示数组；
2、用仿函数实现大小堆随时切换，实现代码复用；
3、实现向下调整算法，时间复杂度为O（lgn）；
下面是我用某教材中的一个建最小堆的过程图，希望能更直观一些：
为了保证复用性，用仿函数重载了(),下面是复用的向下调整算法：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
void _AdjustDown(int root,int size)
{
Camper camper; //仿函数
int parent = root;
int child = parent * 2 + 1;
while (child <= size) //保证访问不越界
{
if (child < size && camper(_vec[child+1] , _vec[child])) //保证存在右子树、同时判断右子树是否大于或小于左子树
{
child++;
}
if (camper(_vec[child], _vec[parent]))
{
swap(_vec[parent], _vec[child]);
parent = child;
child = parent * 2 + 1;
}
else
{
break;
}
}
}
|
排序算法思想：
1、将堆顶数据与堆中最后一个数据交换；
2、堆大小减一，然后调用向下调整算法；
3、结束条件：堆中剩下一个有效值；
排序算法实现：
|
1
2
3
4
5
6
7
8
9
10
|
void Sort()
{
size_t size = _vec.size(); //数据数量
while (size > 1)
{
swap(_vec[0], _vec[size - 1]);
size--;
_AdjustDown(size);
}
}
|
仿函数的实现：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
template<class T>
struct Greater //大于
{
bool operator ()(const T& l, const T& p)
{
return l > p;
}
};
template<class T>
struct Less //小于
{
bool operator () (const T&l, const T& p)
{
return l < p;
}
};
|
完整的代码实现：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
#include<iostream>
using namespace std;
#include<vector>
template<class T>
struct Greater //大于
{
bool operator ()(const T& l, const T& p)
{
return l > p;
}
};
template<class T>
struct Less //小于
{
bool operator () (const T&l, const T& p)
{
return l < p;
}
};
template<class T,class Camper>
class HeapSort //建大堆
{
public:
HeapSort()
{}
HeapSort(T* arr, size_t n)
{
_vec.reserve(n);
if (arr != NULL)
{
for (size_t i = 0; i < n; i++)
{
_vec.push_back(arr[i]);
}
}
_AdjustDown(_vec.size());
}
void Sort()
{
size_t size = _vec.size(); //数据数量
while (size > 1)
{
swap(_vec[0], _vec[size - 1]);
size--;
_AdjustDown(size);
}
}
void Print()
{
for (size_t i = 0; i < _vec.size(); i++)
{
cout << _vec[i] <<" ";
}
cout << endl;
}
protected:
void _AdjustDown(int size)
{
int parent = (size - 2) / 2;
while (parent >= 0)
{
_AdjustDown(parent, size - 1);
parent--;
}
}
void _AdjustDown(int root,int size)
{
Camper camper; //仿函数
int parent = root;
int child = parent * 2 + 1;
while (child <= size) //保证访问不越界
{
if (child < size && camper(_vec[child+1] , _vec[child])) //保证存在右子树、同时判断右子树是否大于或小于左子树
{
child++;
}
if (camper(_vec[child], _vec[parent]))
{
swap(_vec[parent], _vec[child]);
parent = child;
child = parent * 2 + 1;
}
else
{
break;
}
}
}
private:
vector<T> _vec;
};
|
测试用例代码：
|
1
2
3
4
5
6
7
8
|
void TextSort()
{
int a[] = { 10, 11, 13, 12, 16, 18, 15, 17, 14, 19 };
HeapSort<int,Greater<int>> h(a, sizeof(a) / sizeof(a[0]));
h.Print();
h.Sort();
h.Print();
}
|
感谢阅读，希望能帮助到大家，谢谢大家对本站的支持！
原文链接：http://blog.csdn.net/he_shuai20/article/details/68948132