williamr@2
|
1 |
/*
|
williamr@2
|
2 |
*
|
williamr@2
|
3 |
*
|
williamr@2
|
4 |
* Copyright (c) 1994
|
williamr@2
|
5 |
* Hewlett-Packard Company
|
williamr@2
|
6 |
*
|
williamr@2
|
7 |
* Copyright (c) 1996,1997
|
williamr@2
|
8 |
* Silicon Graphics Computer Systems, Inc.
|
williamr@2
|
9 |
*
|
williamr@2
|
10 |
* Copyright (c) 1997
|
williamr@2
|
11 |
* Moscow Center for SPARC Technology
|
williamr@2
|
12 |
*
|
williamr@2
|
13 |
* Copyright (c) 1999
|
williamr@2
|
14 |
* Boris Fomitchev
|
williamr@2
|
15 |
*
|
williamr@2
|
16 |
* This material is provided "as is", with absolutely no warranty expressed
|
williamr@2
|
17 |
* or implied. Any use is at your own risk.
|
williamr@2
|
18 |
*
|
williamr@2
|
19 |
* Permission to use or copy this software for any purpose is hereby granted
|
williamr@2
|
20 |
* without fee, provided the above notices are retained on all copies.
|
williamr@2
|
21 |
* Permission to modify the code and to distribute modified code is granted,
|
williamr@2
|
22 |
* provided the above notices are retained, and a notice that the code was
|
williamr@2
|
23 |
* modified is included with the above copyright notice.
|
williamr@2
|
24 |
*
|
williamr@2
|
25 |
*/
|
williamr@2
|
26 |
#ifndef _STLP_HEAP_C
|
williamr@2
|
27 |
#define _STLP_HEAP_C
|
williamr@2
|
28 |
|
williamr@2
|
29 |
#ifndef _STLP_INTERNAL_HEAP_H
|
williamr@2
|
30 |
# include <stl/_heap.h>
|
williamr@2
|
31 |
#endif
|
williamr@2
|
32 |
|
williamr@2
|
33 |
#ifndef _STLP_INTERNAL_ITERATOR_BASE_H
|
williamr@2
|
34 |
# include <stl/_iterator_base.h>
|
williamr@2
|
35 |
#endif
|
williamr@2
|
36 |
|
williamr@2
|
37 |
_STLP_BEGIN_NAMESPACE
|
williamr@2
|
38 |
|
williamr@2
|
39 |
template <class _RandomAccessIterator, class _Distance, class _Tp>
|
williamr@2
|
40 |
_STLP_INLINE_LOOP
|
williamr@2
|
41 |
void
|
williamr@2
|
42 |
__push_heap(_RandomAccessIterator __first,
|
williamr@2
|
43 |
_Distance __holeIndex, _Distance __topIndex, _Tp __val)
|
williamr@2
|
44 |
{
|
williamr@2
|
45 |
_Distance __parent = (__holeIndex - 1) / 2;
|
williamr@2
|
46 |
while (__holeIndex > __topIndex && *(__first + __parent) < __val) {
|
williamr@2
|
47 |
*(__first + __holeIndex) = *(__first + __parent);
|
williamr@2
|
48 |
__holeIndex = __parent;
|
williamr@2
|
49 |
__parent = (__holeIndex - 1) / 2;
|
williamr@2
|
50 |
}
|
williamr@2
|
51 |
*(__first + __holeIndex) = __val;
|
williamr@2
|
52 |
}
|
williamr@2
|
53 |
|
williamr@2
|
54 |
template <class _RandomAccessIterator, class _Distance, class _Tp>
|
williamr@2
|
55 |
inline void
|
williamr@2
|
56 |
__push_heap_aux(_RandomAccessIterator __first,
|
williamr@2
|
57 |
_RandomAccessIterator __last, _Distance*, _Tp*)
|
williamr@2
|
58 |
{
|
williamr@2
|
59 |
__push_heap(__first, _Distance((__last - __first) - 1), _Distance(0),
|
williamr@2
|
60 |
_Tp(*(__last - 1)));
|
williamr@2
|
61 |
}
|
williamr@2
|
62 |
|
williamr@2
|
63 |
template <class _RandomAccessIterator>
|
williamr@2
|
64 |
void
|
williamr@2
|
65 |
push_heap(_RandomAccessIterator __first, _RandomAccessIterator __last)
|
williamr@2
|
66 |
{
|
williamr@2
|
67 |
__push_heap_aux(__first, __last,
|
williamr@2
|
68 |
_STLP_DISTANCE_TYPE(__first, _RandomAccessIterator), _STLP_VALUE_TYPE(__first, _RandomAccessIterator));
|
williamr@2
|
69 |
}
|
williamr@2
|
70 |
|
williamr@2
|
71 |
|
williamr@2
|
72 |
template <class _RandomAccessIterator, class _Distance, class _Tp,
|
williamr@2
|
73 |
class _Compare>
|
williamr@2
|
74 |
_STLP_INLINE_LOOP
|
williamr@2
|
75 |
void
|
williamr@2
|
76 |
__push_heap(_RandomAccessIterator __first, _Distance __holeIndex,
|
williamr@2
|
77 |
_Distance __topIndex, _Tp __val, _Compare __comp)
|
williamr@2
|
78 |
{
|
williamr@2
|
79 |
_Distance __parent = (__holeIndex - 1) / 2;
|
williamr@2
|
80 |
while (__holeIndex > __topIndex && __comp(*(__first + __parent), __val)) {
|
williamr@2
|
81 |
*(__first + __holeIndex) = *(__first + __parent);
|
williamr@2
|
82 |
__holeIndex = __parent;
|
williamr@2
|
83 |
__parent = (__holeIndex - 1) / 2;
|
williamr@2
|
84 |
}
|
williamr@2
|
85 |
*(__first + __holeIndex) = __val;
|
williamr@2
|
86 |
}
|
williamr@2
|
87 |
|
williamr@2
|
88 |
template <class _RandomAccessIterator, class _Compare,
|
williamr@2
|
89 |
class _Distance, class _Tp>
|
williamr@2
|
90 |
inline void
|
williamr@2
|
91 |
__push_heap_aux(_RandomAccessIterator __first,
|
williamr@2
|
92 |
_RandomAccessIterator __last, _Compare __comp,
|
williamr@2
|
93 |
_Distance*, _Tp*)
|
williamr@2
|
94 |
{
|
williamr@2
|
95 |
__push_heap(__first, _Distance((__last - __first) - 1), _Distance(0),
|
williamr@2
|
96 |
_Tp(*(__last - 1)), __comp);
|
williamr@2
|
97 |
}
|
williamr@2
|
98 |
|
williamr@2
|
99 |
template <class _RandomAccessIterator, class _Compare>
|
williamr@2
|
100 |
void
|
williamr@2
|
101 |
push_heap(_RandomAccessIterator __first, _RandomAccessIterator __last,
|
williamr@2
|
102 |
_Compare __comp)
|
williamr@2
|
103 |
{
|
williamr@2
|
104 |
__push_heap_aux(__first, __last, __comp,
|
williamr@2
|
105 |
_STLP_DISTANCE_TYPE(__first, _RandomAccessIterator), _STLP_VALUE_TYPE(__first, _RandomAccessIterator));
|
williamr@2
|
106 |
}
|
williamr@2
|
107 |
|
williamr@2
|
108 |
template <class _RandomAccessIterator, class _Distance, class _Tp>
|
williamr@2
|
109 |
void
|
williamr@2
|
110 |
__adjust_heap(_RandomAccessIterator __first, _Distance __holeIndex,
|
williamr@2
|
111 |
_Distance __len, _Tp __val) {
|
williamr@2
|
112 |
_Distance __topIndex = __holeIndex;
|
williamr@2
|
113 |
_Distance __secondChild = 2 * __holeIndex + 2;
|
williamr@2
|
114 |
while (__secondChild < __len) {
|
williamr@2
|
115 |
if (*(__first + __secondChild) < *(__first + (__secondChild - 1)))
|
williamr@2
|
116 |
__secondChild--;
|
williamr@2
|
117 |
*(__first + __holeIndex) = *(__first + __secondChild);
|
williamr@2
|
118 |
__holeIndex = __secondChild;
|
williamr@2
|
119 |
__secondChild = 2 * (__secondChild + 1);
|
williamr@2
|
120 |
}
|
williamr@2
|
121 |
if (__secondChild == __len) {
|
williamr@2
|
122 |
*(__first + __holeIndex) = *(__first + (__secondChild - 1));
|
williamr@2
|
123 |
__holeIndex = __secondChild - 1;
|
williamr@2
|
124 |
}
|
williamr@2
|
125 |
__push_heap(__first, __holeIndex, __topIndex, __val);
|
williamr@2
|
126 |
}
|
williamr@2
|
127 |
|
williamr@2
|
128 |
|
williamr@2
|
129 |
template <class _RandomAccessIterator, class _Tp>
|
williamr@2
|
130 |
inline void
|
williamr@2
|
131 |
__pop_heap_aux(_RandomAccessIterator __first, _RandomAccessIterator __last, _Tp*) {
|
williamr@2
|
132 |
__pop_heap(__first, __last - 1, __last - 1,
|
williamr@2
|
133 |
_Tp(*(__last - 1)), _STLP_DISTANCE_TYPE(__first, _RandomAccessIterator));
|
williamr@2
|
134 |
}
|
williamr@2
|
135 |
|
williamr@2
|
136 |
template <class _RandomAccessIterator>
|
williamr@2
|
137 |
void pop_heap(_RandomAccessIterator __first,
|
williamr@2
|
138 |
_RandomAccessIterator __last) {
|
williamr@2
|
139 |
__pop_heap_aux(__first, __last, _STLP_VALUE_TYPE(__first, _RandomAccessIterator));
|
williamr@2
|
140 |
}
|
williamr@2
|
141 |
|
williamr@2
|
142 |
template <class _RandomAccessIterator, class _Distance,
|
williamr@2
|
143 |
class _Tp, class _Compare>
|
williamr@2
|
144 |
void
|
williamr@2
|
145 |
__adjust_heap(_RandomAccessIterator __first, _Distance __holeIndex,
|
williamr@2
|
146 |
_Distance __len, _Tp __val, _Compare __comp)
|
williamr@2
|
147 |
{
|
williamr@2
|
148 |
_Distance __topIndex = __holeIndex;
|
williamr@2
|
149 |
_Distance __secondChild = 2 * __holeIndex + 2;
|
williamr@2
|
150 |
while (__secondChild < __len) {
|
williamr@2
|
151 |
if (__comp(*(__first + __secondChild), *(__first + (__secondChild - 1))))
|
williamr@2
|
152 |
__secondChild--;
|
williamr@2
|
153 |
*(__first + __holeIndex) = *(__first + __secondChild);
|
williamr@2
|
154 |
__holeIndex = __secondChild;
|
williamr@2
|
155 |
__secondChild = 2 * (__secondChild + 1);
|
williamr@2
|
156 |
}
|
williamr@2
|
157 |
if (__secondChild == __len) {
|
williamr@2
|
158 |
*(__first + __holeIndex) = *(__first + (__secondChild - 1));
|
williamr@2
|
159 |
__holeIndex = __secondChild - 1;
|
williamr@2
|
160 |
}
|
williamr@2
|
161 |
__push_heap(__first, __holeIndex, __topIndex, __val, __comp);
|
williamr@2
|
162 |
}
|
williamr@2
|
163 |
|
williamr@2
|
164 |
|
williamr@2
|
165 |
template <class _RandomAccessIterator, class _Tp, class _Compare>
|
williamr@2
|
166 |
inline void
|
williamr@2
|
167 |
__pop_heap_aux(_RandomAccessIterator __first,
|
williamr@2
|
168 |
_RandomAccessIterator __last, _Tp*, _Compare __comp)
|
williamr@2
|
169 |
{
|
williamr@2
|
170 |
__pop_heap(__first, __last - 1, __last - 1, _Tp(*(__last - 1)), __comp,
|
williamr@2
|
171 |
_STLP_DISTANCE_TYPE(__first, _RandomAccessIterator));
|
williamr@2
|
172 |
}
|
williamr@2
|
173 |
|
williamr@2
|
174 |
|
williamr@2
|
175 |
template <class _RandomAccessIterator, class _Compare>
|
williamr@2
|
176 |
void
|
williamr@2
|
177 |
pop_heap(_RandomAccessIterator __first,
|
williamr@2
|
178 |
_RandomAccessIterator __last, _Compare __comp)
|
williamr@2
|
179 |
{
|
williamr@2
|
180 |
__pop_heap_aux(__first, __last, _STLP_VALUE_TYPE(__first, _RandomAccessIterator), __comp);
|
williamr@2
|
181 |
}
|
williamr@2
|
182 |
|
williamr@2
|
183 |
template <class _RandomAccessIterator, class _Tp, class _Distance>
|
williamr@2
|
184 |
_STLP_INLINE_LOOP
|
williamr@2
|
185 |
void
|
williamr@2
|
186 |
__make_heap(_RandomAccessIterator __first,
|
williamr@2
|
187 |
_RandomAccessIterator __last, _Tp*, _Distance*)
|
williamr@2
|
188 |
{
|
williamr@2
|
189 |
if (__last - __first < 2) return;
|
williamr@2
|
190 |
_Distance __len = __last - __first;
|
williamr@2
|
191 |
_Distance __parent = (__len - 2)/2;
|
williamr@2
|
192 |
|
williamr@2
|
193 |
while (true) {
|
williamr@2
|
194 |
__adjust_heap(__first, __parent, __len, _Tp(*(__first + __parent)));
|
williamr@2
|
195 |
if (__parent == 0) return;
|
williamr@2
|
196 |
__parent--;
|
williamr@2
|
197 |
}
|
williamr@2
|
198 |
}
|
williamr@2
|
199 |
|
williamr@2
|
200 |
template <class _RandomAccessIterator>
|
williamr@2
|
201 |
void
|
williamr@2
|
202 |
make_heap(_RandomAccessIterator __first, _RandomAccessIterator __last)
|
williamr@2
|
203 |
{
|
williamr@2
|
204 |
__make_heap(__first, __last,
|
williamr@2
|
205 |
_STLP_VALUE_TYPE(__first, _RandomAccessIterator), _STLP_DISTANCE_TYPE(__first, _RandomAccessIterator));
|
williamr@2
|
206 |
}
|
williamr@2
|
207 |
|
williamr@2
|
208 |
template <class _RandomAccessIterator, class _Compare,
|
williamr@2
|
209 |
class _Tp, class _Distance>
|
williamr@2
|
210 |
_STLP_INLINE_LOOP
|
williamr@2
|
211 |
void
|
williamr@2
|
212 |
__make_heap(_RandomAccessIterator __first, _RandomAccessIterator __last,
|
williamr@2
|
213 |
_Compare __comp, _Tp*, _Distance*)
|
williamr@2
|
214 |
{
|
williamr@2
|
215 |
if (__last - __first < 2) return;
|
williamr@2
|
216 |
_Distance __len = __last - __first;
|
williamr@2
|
217 |
_Distance __parent = (__len - 2)/2;
|
williamr@2
|
218 |
|
williamr@2
|
219 |
while (true) {
|
williamr@2
|
220 |
__adjust_heap(__first, __parent, __len, _Tp(*(__first + __parent)),
|
williamr@2
|
221 |
__comp);
|
williamr@2
|
222 |
if (__parent == 0) return;
|
williamr@2
|
223 |
__parent--;
|
williamr@2
|
224 |
}
|
williamr@2
|
225 |
}
|
williamr@2
|
226 |
|
williamr@2
|
227 |
template <class _RandomAccessIterator, class _Compare>
|
williamr@2
|
228 |
void
|
williamr@2
|
229 |
make_heap(_RandomAccessIterator __first,
|
williamr@2
|
230 |
_RandomAccessIterator __last, _Compare __comp)
|
williamr@2
|
231 |
{
|
williamr@2
|
232 |
__make_heap(__first, __last, __comp,
|
williamr@2
|
233 |
_STLP_VALUE_TYPE(__first, _RandomAccessIterator), _STLP_DISTANCE_TYPE(__first, _RandomAccessIterator));
|
williamr@2
|
234 |
}
|
williamr@2
|
235 |
|
williamr@2
|
236 |
_STLP_END_NAMESPACE
|
williamr@2
|
237 |
|
williamr@2
|
238 |
#endif /* _STLP_HEAP_C */
|
williamr@2
|
239 |
|
williamr@2
|
240 |
// Local Variables:
|
williamr@2
|
241 |
// mode:C++
|
williamr@2
|
242 |
// End:
|