... |
... |
@@ -55,16 +55,16 @@
|
55 |
55 |
|
56 |
56 |
/// \brief Exception thrown by RadixHeap.
|
57 |
57 |
///
|
58 |
58 |
/// This exception is thrown when an item is inserted into a
|
59 |
59 |
/// RadixHeap with a priority smaller than the last erased one.
|
60 |
60 |
/// \see RadixHeap
|
61 |
|
class UnderFlowPriorityError : public Exception {
|
|
61 |
class PriorityUnderflowError : public Exception {
|
62 |
62 |
public:
|
63 |
63 |
virtual const char* what() const throw() {
|
64 |
|
return "lemon::RadixHeap::UnderFlowPriorityError";
|
|
64 |
return "lemon::RadixHeap::PriorityUnderflowError";
|
65 |
65 |
}
|
66 |
66 |
};
|
67 |
67 |
|
68 |
68 |
/// \brief Type to represent the states of the items.
|
69 |
69 |
///
|
70 |
70 |
/// Each item has a state associated to it. It can be "in heap",
|
... |
... |
@@ -91,14 +91,14 @@
|
91 |
91 |
struct RadixBox {
|
92 |
92 |
int first;
|
93 |
93 |
int min, size;
|
94 |
94 |
RadixBox(int _min, int _size) : first(-1), min(_min), size(_size) {}
|
95 |
95 |
};
|
96 |
96 |
|
97 |
|
std::vector<RadixItem> data;
|
98 |
|
std::vector<RadixBox> boxes;
|
|
97 |
std::vector<RadixItem> _data;
|
|
98 |
std::vector<RadixBox> _boxes;
|
99 |
99 |
|
100 |
100 |
ItemIntMap &_iim;
|
101 |
101 |
|
102 |
102 |
public:
|
103 |
103 |
|
104 |
104 |
/// \brief Constructor.
|
... |
... |
@@ -109,171 +109,171 @@
|
109 |
109 |
/// The assigned value must be \c PRE_HEAP (<tt>-1</tt>) for each item.
|
110 |
110 |
/// \param minimum The initial minimum value of the heap.
|
111 |
111 |
/// \param capacity The initial capacity of the heap.
|
112 |
112 |
RadixHeap(ItemIntMap &map, int minimum = 0, int capacity = 0)
|
113 |
113 |
: _iim(map)
|
114 |
114 |
{
|
115 |
|
boxes.push_back(RadixBox(minimum, 1));
|
116 |
|
boxes.push_back(RadixBox(minimum + 1, 1));
|
117 |
|
while (lower(boxes.size() - 1, capacity + minimum - 1)) {
|
|
115 |
_boxes.push_back(RadixBox(minimum, 1));
|
|
116 |
_boxes.push_back(RadixBox(minimum + 1, 1));
|
|
117 |
while (lower(_boxes.size() - 1, capacity + minimum - 1)) {
|
118 |
118 |
extend();
|
119 |
119 |
}
|
120 |
120 |
}
|
121 |
121 |
|
122 |
122 |
/// \brief The number of items stored in the heap.
|
123 |
123 |
///
|
124 |
124 |
/// This function returns the number of items stored in the heap.
|
125 |
|
int size() const { return data.size(); }
|
|
125 |
int size() const { return _data.size(); }
|
126 |
126 |
|
127 |
127 |
/// \brief Check if the heap is empty.
|
128 |
128 |
///
|
129 |
129 |
/// This function returns \c true if the heap is empty.
|
130 |
|
bool empty() const { return data.empty(); }
|
|
130 |
bool empty() const { return _data.empty(); }
|
131 |
131 |
|
132 |
132 |
/// \brief Make the heap empty.
|
133 |
133 |
///
|
134 |
134 |
/// This functon makes the heap empty.
|
135 |
135 |
/// It does not change the cross reference map. If you want to reuse
|
136 |
136 |
/// a heap that is not surely empty, you should first clear it and
|
137 |
137 |
/// then you should set the cross reference map to \c PRE_HEAP
|
138 |
138 |
/// for each item.
|
139 |
139 |
/// \param minimum The minimum value of the heap.
|
140 |
140 |
/// \param capacity The capacity of the heap.
|
141 |
141 |
void clear(int minimum = 0, int capacity = 0) {
|
142 |
|
data.clear(); boxes.clear();
|
143 |
|
boxes.push_back(RadixBox(minimum, 1));
|
144 |
|
boxes.push_back(RadixBox(minimum + 1, 1));
|
145 |
|
while (lower(boxes.size() - 1, capacity + minimum - 1)) {
|
|
142 |
_data.clear(); _boxes.clear();
|
|
143 |
_boxes.push_back(RadixBox(minimum, 1));
|
|
144 |
_boxes.push_back(RadixBox(minimum + 1, 1));
|
|
145 |
while (lower(_boxes.size() - 1, capacity + minimum - 1)) {
|
146 |
146 |
extend();
|
147 |
147 |
}
|
148 |
148 |
}
|
149 |
149 |
|
150 |
150 |
private:
|
151 |
151 |
|
152 |
152 |
bool upper(int box, Prio pr) {
|
153 |
|
return pr < boxes[box].min;
|
|
153 |
return pr < _boxes[box].min;
|
154 |
154 |
}
|
155 |
155 |
|
156 |
156 |
bool lower(int box, Prio pr) {
|
157 |
|
return pr >= boxes[box].min + boxes[box].size;
|
|
157 |
return pr >= _boxes[box].min + _boxes[box].size;
|
158 |
158 |
}
|
159 |
159 |
|
160 |
160 |
// Remove item from the box list
|
161 |
161 |
void remove(int index) {
|
162 |
|
if (data[index].prev >= 0) {
|
163 |
|
data[data[index].prev].next = data[index].next;
|
|
162 |
if (_data[index].prev >= 0) {
|
|
163 |
_data[_data[index].prev].next = _data[index].next;
|
164 |
164 |
} else {
|
165 |
|
boxes[data[index].box].first = data[index].next;
|
|
165 |
_boxes[_data[index].box].first = _data[index].next;
|
166 |
166 |
}
|
167 |
|
if (data[index].next >= 0) {
|
168 |
|
data[data[index].next].prev = data[index].prev;
|
|
167 |
if (_data[index].next >= 0) {
|
|
168 |
_data[_data[index].next].prev = _data[index].prev;
|
169 |
169 |
}
|
170 |
170 |
}
|
171 |
171 |
|
172 |
172 |
// Insert item into the box list
|
173 |
173 |
void insert(int box, int index) {
|
174 |
|
if (boxes[box].first == -1) {
|
175 |
|
boxes[box].first = index;
|
176 |
|
data[index].next = data[index].prev = -1;
|
|
174 |
if (_boxes[box].first == -1) {
|
|
175 |
_boxes[box].first = index;
|
|
176 |
_data[index].next = _data[index].prev = -1;
|
177 |
177 |
} else {
|
178 |
|
data[index].next = boxes[box].first;
|
179 |
|
data[boxes[box].first].prev = index;
|
180 |
|
data[index].prev = -1;
|
181 |
|
boxes[box].first = index;
|
|
178 |
_data[index].next = _boxes[box].first;
|
|
179 |
_data[_boxes[box].first].prev = index;
|
|
180 |
_data[index].prev = -1;
|
|
181 |
_boxes[box].first = index;
|
182 |
182 |
}
|
183 |
|
data[index].box = box;
|
|
183 |
_data[index].box = box;
|
184 |
184 |
}
|
185 |
185 |
|
186 |
186 |
// Add a new box to the box list
|
187 |
187 |
void extend() {
|
188 |
|
int min = boxes.back().min + boxes.back().size;
|
189 |
|
int bs = 2 * boxes.back().size;
|
190 |
|
boxes.push_back(RadixBox(min, bs));
|
|
188 |
int min = _boxes.back().min + _boxes.back().size;
|
|
189 |
int bs = 2 * _boxes.back().size;
|
|
190 |
_boxes.push_back(RadixBox(min, bs));
|
191 |
191 |
}
|
192 |
192 |
|
193 |
193 |
// Move an item up into the proper box.
|
194 |
|
void bubble_up(int index) {
|
195 |
|
if (!lower(data[index].box, data[index].prio)) return;
|
|
194 |
void bubbleUp(int index) {
|
|
195 |
if (!lower(_data[index].box, _data[index].prio)) return;
|
196 |
196 |
remove(index);
|
197 |
|
int box = findUp(data[index].box, data[index].prio);
|
|
197 |
int box = findUp(_data[index].box, _data[index].prio);
|
198 |
198 |
insert(box, index);
|
199 |
199 |
}
|
200 |
200 |
|
201 |
201 |
// Find up the proper box for the item with the given priority
|
202 |
202 |
int findUp(int start, int pr) {
|
203 |
203 |
while (lower(start, pr)) {
|
204 |
|
if (++start == int(boxes.size())) {
|
|
204 |
if (++start == int(_boxes.size())) {
|
205 |
205 |
extend();
|
206 |
206 |
}
|
207 |
207 |
}
|
208 |
208 |
return start;
|
209 |
209 |
}
|
210 |
210 |
|
211 |
211 |
// Move an item down into the proper box
|
212 |
|
void bubble_down(int index) {
|
213 |
|
if (!upper(data[index].box, data[index].prio)) return;
|
|
212 |
void bubbleDown(int index) {
|
|
213 |
if (!upper(_data[index].box, _data[index].prio)) return;
|
214 |
214 |
remove(index);
|
215 |
|
int box = findDown(data[index].box, data[index].prio);
|
|
215 |
int box = findDown(_data[index].box, _data[index].prio);
|
216 |
216 |
insert(box, index);
|
217 |
217 |
}
|
218 |
218 |
|
219 |
219 |
// Find down the proper box for the item with the given priority
|
220 |
220 |
int findDown(int start, int pr) {
|
221 |
221 |
while (upper(start, pr)) {
|
222 |
|
if (--start < 0) throw UnderFlowPriorityError();
|
|
222 |
if (--start < 0) throw PriorityUnderflowError();
|
223 |
223 |
}
|
224 |
224 |
return start;
|
225 |
225 |
}
|
226 |
226 |
|
227 |
227 |
// Find the first non-empty box
|
228 |
228 |
int findFirst() {
|
229 |
229 |
int first = 0;
|
230 |
|
while (boxes[first].first == -1) ++first;
|
|
230 |
while (_boxes[first].first == -1) ++first;
|
231 |
231 |
return first;
|
232 |
232 |
}
|
233 |
233 |
|
234 |
234 |
// Gives back the minimum priority of the given box
|
235 |
235 |
int minValue(int box) {
|
236 |
|
int min = data[boxes[box].first].prio;
|
237 |
|
for (int k = boxes[box].first; k != -1; k = data[k].next) {
|
238 |
|
if (data[k].prio < min) min = data[k].prio;
|
|
236 |
int min = _data[_boxes[box].first].prio;
|
|
237 |
for (int k = _boxes[box].first; k != -1; k = _data[k].next) {
|
|
238 |
if (_data[k].prio < min) min = _data[k].prio;
|
239 |
239 |
}
|
240 |
240 |
return min;
|
241 |
241 |
}
|
242 |
242 |
|
243 |
243 |
// Rearrange the items of the heap and make the first box non-empty
|
244 |
244 |
void moveDown() {
|
245 |
245 |
int box = findFirst();
|
246 |
246 |
if (box == 0) return;
|
247 |
247 |
int min = minValue(box);
|
248 |
248 |
for (int i = 0; i <= box; ++i) {
|
249 |
|
boxes[i].min = min;
|
250 |
|
min += boxes[i].size;
|
|
249 |
_boxes[i].min = min;
|
|
250 |
min += _boxes[i].size;
|
251 |
251 |
}
|
252 |
|
int curr = boxes[box].first, next;
|
|
252 |
int curr = _boxes[box].first, next;
|
253 |
253 |
while (curr != -1) {
|
254 |
|
next = data[curr].next;
|
255 |
|
bubble_down(curr);
|
|
254 |
next = _data[curr].next;
|
|
255 |
bubbleDown(curr);
|
256 |
256 |
curr = next;
|
257 |
257 |
}
|
258 |
258 |
}
|
259 |
259 |
|
260 |
|
void relocate_last(int index) {
|
261 |
|
if (index != int(data.size()) - 1) {
|
262 |
|
data[index] = data.back();
|
263 |
|
if (data[index].prev != -1) {
|
264 |
|
data[data[index].prev].next = index;
|
|
260 |
void relocateLast(int index) {
|
|
261 |
if (index != int(_data.size()) - 1) {
|
|
262 |
_data[index] = _data.back();
|
|
263 |
if (_data[index].prev != -1) {
|
|
264 |
_data[_data[index].prev].next = index;
|
265 |
265 |
} else {
|
266 |
|
boxes[data[index].box].first = index;
|
|
266 |
_boxes[_data[index].box].first = index;
|
267 |
267 |
}
|
268 |
|
if (data[index].next != -1) {
|
269 |
|
data[data[index].next].prev = index;
|
|
268 |
if (_data[index].next != -1) {
|
|
269 |
_data[_data[index].next].prev = index;
|
270 |
270 |
}
|
271 |
|
_iim[data[index].item] = index;
|
|
271 |
_iim[_data[index].item] = index;
|
272 |
272 |
}
|
273 |
|
data.pop_back();
|
|
273 |
_data.pop_back();
|
274 |
274 |
}
|
275 |
275 |
|
276 |
276 |
public:
|
277 |
277 |
|
278 |
278 |
/// \brief Insert an item into the heap with the given priority.
|
279 |
279 |
///
|
... |
... |
@@ -281,73 +281,73 @@
|
281 |
281 |
/// given priority.
|
282 |
282 |
/// \param i The item to insert.
|
283 |
283 |
/// \param p The priority of the item.
|
284 |
284 |
/// \pre \e i must not be stored in the heap.
|
285 |
285 |
/// \warning This method may throw an \c UnderFlowPriorityException.
|
286 |
286 |
void push(const Item &i, const Prio &p) {
|
287 |
|
int n = data.size();
|
|
287 |
int n = _data.size();
|
288 |
288 |
_iim.set(i, n);
|
289 |
|
data.push_back(RadixItem(i, p));
|
290 |
|
while (lower(boxes.size() - 1, p)) {
|
|
289 |
_data.push_back(RadixItem(i, p));
|
|
290 |
while (lower(_boxes.size() - 1, p)) {
|
291 |
291 |
extend();
|
292 |
292 |
}
|
293 |
|
int box = findDown(boxes.size() - 1, p);
|
|
293 |
int box = findDown(_boxes.size() - 1, p);
|
294 |
294 |
insert(box, n);
|
295 |
295 |
}
|
296 |
296 |
|
297 |
297 |
/// \brief Return the item having minimum priority.
|
298 |
298 |
///
|
299 |
299 |
/// This function returns the item having minimum priority.
|
300 |
300 |
/// \pre The heap must be non-empty.
|
301 |
301 |
Item top() const {
|
302 |
302 |
const_cast<RadixHeap<ItemIntMap>&>(*this).moveDown();
|
303 |
|
return data[boxes[0].first].item;
|
|
303 |
return _data[_boxes[0].first].item;
|
304 |
304 |
}
|
305 |
305 |
|
306 |
306 |
/// \brief The minimum priority.
|
307 |
307 |
///
|
308 |
308 |
/// This function returns the minimum priority.
|
309 |
309 |
/// \pre The heap must be non-empty.
|
310 |
310 |
Prio prio() const {
|
311 |
311 |
const_cast<RadixHeap<ItemIntMap>&>(*this).moveDown();
|
312 |
|
return data[boxes[0].first].prio;
|
|
312 |
return _data[_boxes[0].first].prio;
|
313 |
313 |
}
|
314 |
314 |
|
315 |
315 |
/// \brief Remove the item having minimum priority.
|
316 |
316 |
///
|
317 |
317 |
/// This function removes the item having minimum priority.
|
318 |
318 |
/// \pre The heap must be non-empty.
|
319 |
319 |
void pop() {
|
320 |
320 |
moveDown();
|
321 |
|
int index = boxes[0].first;
|
322 |
|
_iim[data[index].item] = POST_HEAP;
|
|
321 |
int index = _boxes[0].first;
|
|
322 |
_iim[_data[index].item] = POST_HEAP;
|
323 |
323 |
remove(index);
|
324 |
|
relocate_last(index);
|
|
324 |
relocateLast(index);
|
325 |
325 |
}
|
326 |
326 |
|
327 |
327 |
/// \brief Remove the given item from the heap.
|
328 |
328 |
///
|
329 |
329 |
/// This function removes the given item from the heap if it is
|
330 |
330 |
/// already stored.
|
331 |
331 |
/// \param i The item to delete.
|
332 |
332 |
/// \pre \e i must be in the heap.
|
333 |
333 |
void erase(const Item &i) {
|
334 |
334 |
int index = _iim[i];
|
335 |
335 |
_iim[i] = POST_HEAP;
|
336 |
336 |
remove(index);
|
337 |
|
relocate_last(index);
|
|
337 |
relocateLast(index);
|
338 |
338 |
}
|
339 |
339 |
|
340 |
340 |
/// \brief The priority of the given item.
|
341 |
341 |
///
|
342 |
342 |
/// This function returns the priority of the given item.
|
343 |
343 |
/// \param i The item.
|
344 |
344 |
/// \pre \e i must be in the heap.
|
345 |
345 |
Prio operator[](const Item &i) const {
|
346 |
346 |
int idx = _iim[i];
|
347 |
|
return data[idx].prio;
|
|
347 |
return _data[idx].prio;
|
348 |
348 |
}
|
349 |
349 |
|
350 |
350 |
/// \brief Set the priority of an item or insert it, if it is
|
351 |
351 |
/// not stored in the heap.
|
352 |
352 |
///
|
353 |
353 |
/// This method sets the priority of the given item if it is
|
... |
... |
@@ -359,44 +359,44 @@
|
359 |
359 |
/// \warning This method may throw an \c UnderFlowPriorityException.
|
360 |
360 |
void set(const Item &i, const Prio &p) {
|
361 |
361 |
int idx = _iim[i];
|
362 |
362 |
if( idx < 0 ) {
|
363 |
363 |
push(i, p);
|
364 |
364 |
}
|
365 |
|
else if( p >= data[idx].prio ) {
|
366 |
|
data[idx].prio = p;
|
367 |
|
bubble_up(idx);
|
|
365 |
else if( p >= _data[idx].prio ) {
|
|
366 |
_data[idx].prio = p;
|
|
367 |
bubbleUp(idx);
|
368 |
368 |
} else {
|
369 |
|
data[idx].prio = p;
|
370 |
|
bubble_down(idx);
|
|
369 |
_data[idx].prio = p;
|
|
370 |
bubbleDown(idx);
|
371 |
371 |
}
|
372 |
372 |
}
|
373 |
373 |
|
374 |
374 |
/// \brief Decrease the priority of an item to the given value.
|
375 |
375 |
///
|
376 |
376 |
/// This function decreases the priority of an item to the given value.
|
377 |
377 |
/// \param i The item.
|
378 |
378 |
/// \param p The priority.
|
379 |
379 |
/// \pre \e i must be stored in the heap with priority at least \e p.
|
380 |
380 |
/// \warning This method may throw an \c UnderFlowPriorityException.
|
381 |
381 |
void decrease(const Item &i, const Prio &p) {
|
382 |
382 |
int idx = _iim[i];
|
383 |
|
data[idx].prio = p;
|
384 |
|
bubble_down(idx);
|
|
383 |
_data[idx].prio = p;
|
|
384 |
bubbleDown(idx);
|
385 |
385 |
}
|
386 |
386 |
|
387 |
387 |
/// \brief Increase the priority of an item to the given value.
|
388 |
388 |
///
|
389 |
389 |
/// This function increases the priority of an item to the given value.
|
390 |
390 |
/// \param i The item.
|
391 |
391 |
/// \param p The priority.
|
392 |
392 |
/// \pre \e i must be stored in the heap with priority at most \e p.
|
393 |
393 |
void increase(const Item &i, const Prio &p) {
|
394 |
394 |
int idx = _iim[i];
|
395 |
|
data[idx].prio = p;
|
396 |
|
bubble_up(idx);
|
|
395 |
_data[idx].prio = p;
|
|
396 |
bubbleUp(idx);
|
397 |
397 |
}
|
398 |
398 |
|
399 |
399 |
/// \brief Return the state of an item.
|
400 |
400 |
///
|
401 |
401 |
/// This method returns \c PRE_HEAP if the given item has never
|
402 |
402 |
/// been in the heap, \c IN_HEAP if it is in the heap at the moment,
|