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