deba@2038
|
1 |
/* -*- C++ -*-
|
deba@2038
|
2 |
*
|
deba@2038
|
3 |
* This file is a part of LEMON, a generic C++ optimization library
|
deba@2038
|
4 |
*
|
deba@2038
|
5 |
* Copyright (C) 2003-2006
|
deba@2038
|
6 |
* Egervary Jeno Kombinatorikus Optimalizalasi Kutatocsoport
|
deba@2038
|
7 |
* (Egervary Research Group on Combinatorial Optimization, EGRES).
|
deba@2038
|
8 |
*
|
deba@2038
|
9 |
* Permission to use, modify and distribute this software is granted
|
deba@2038
|
10 |
* provided that this copyright notice appears in all copies. For
|
deba@2038
|
11 |
* precise terms see the accompanying LICENSE file.
|
deba@2038
|
12 |
*
|
deba@2038
|
13 |
* This software is provided "AS IS" with no warranty of any kind,
|
deba@2038
|
14 |
* express or implied, and with no claim as to its suitability for any
|
deba@2038
|
15 |
* purpose.
|
deba@2038
|
16 |
*
|
deba@2038
|
17 |
*/
|
deba@2038
|
18 |
|
deba@2038
|
19 |
#ifndef LEMON_BUCKET_HEAP_H
|
deba@2038
|
20 |
#define LEMON_BUCKET_HEAP_H
|
deba@2038
|
21 |
|
deba@2038
|
22 |
///\ingroup auxdat
|
deba@2038
|
23 |
///\file
|
deba@2038
|
24 |
///\brief Bucket Heap implementation.
|
deba@2038
|
25 |
|
deba@2038
|
26 |
#include <vector>
|
deba@2038
|
27 |
#include <utility>
|
deba@2038
|
28 |
#include <functional>
|
deba@2038
|
29 |
|
deba@2038
|
30 |
namespace lemon {
|
deba@2038
|
31 |
|
deba@2038
|
32 |
/// \ingroup auxdat
|
deba@2038
|
33 |
|
deba@2038
|
34 |
/// \brief A Bucket Heap implementation.
|
deba@2038
|
35 |
///
|
deba@2038
|
36 |
/// This class implements the \e bucket \e heap data structure. A \e heap
|
deba@2038
|
37 |
/// is a data structure for storing items with specified values called \e
|
deba@2038
|
38 |
/// priorities in such a way that finding the item with minimum priority is
|
deba@2038
|
39 |
/// efficient. The bucket heap is very simple implementation, it can store
|
deba@2038
|
40 |
/// only integer priorities and it stores for each priority in the [0..C]
|
deba@2038
|
41 |
/// range a list of items. So it should be used only when the priorities
|
deba@2038
|
42 |
/// are small. It is not intended to use as dijkstra heap.
|
deba@2038
|
43 |
///
|
deba@2038
|
44 |
/// \param _Item Type of the items to be stored.
|
deba@2038
|
45 |
/// \param _ItemIntMap A read and writable Item int map, used internally
|
deba@2038
|
46 |
/// to handle the cross references.
|
deba@2038
|
47 |
/// \param minimize If the given parameter is true then the heap gives back
|
deba@2038
|
48 |
/// the lowest priority.
|
deba@2038
|
49 |
template <typename _Item, typename _ItemIntMap, bool minimize = true >
|
deba@2038
|
50 |
class BucketHeap {
|
deba@2038
|
51 |
|
deba@2038
|
52 |
public:
|
deba@2038
|
53 |
typedef _Item Item;
|
deba@2038
|
54 |
typedef int Prio;
|
deba@2038
|
55 |
typedef std::pair<Item, Prio> Pair;
|
deba@2038
|
56 |
typedef _ItemIntMap ItemIntMap;
|
deba@2038
|
57 |
|
deba@2038
|
58 |
/// \brief Type to represent the items states.
|
deba@2038
|
59 |
///
|
deba@2038
|
60 |
/// Each Item element have a state associated to it. It may be "in heap",
|
deba@2038
|
61 |
/// "pre heap" or "post heap". The latter two are indifferent from the
|
deba@2038
|
62 |
/// heap's point of view, but may be useful to the user.
|
deba@2038
|
63 |
///
|
deba@2038
|
64 |
/// The ItemIntMap \e should be initialized in such way that it maps
|
deba@2038
|
65 |
/// PRE_HEAP (-1) to any element to be put in the heap...
|
deba@2038
|
66 |
enum state_enum {
|
deba@2038
|
67 |
IN_HEAP = 0,
|
deba@2038
|
68 |
PRE_HEAP = -1,
|
deba@2038
|
69 |
POST_HEAP = -2
|
deba@2038
|
70 |
};
|
deba@2038
|
71 |
|
deba@2038
|
72 |
public:
|
deba@2038
|
73 |
/// \brief The constructor.
|
deba@2038
|
74 |
///
|
deba@2038
|
75 |
/// The constructor.
|
deba@2038
|
76 |
/// \param _index should be given to the constructor, since it is used
|
deba@2038
|
77 |
/// internally to handle the cross references. The value of the map
|
deba@2038
|
78 |
/// should be PRE_HEAP (-1) for each element.
|
deba@2038
|
79 |
explicit BucketHeap(ItemIntMap &_index) : index(_index), minimal(0) {}
|
deba@2038
|
80 |
|
deba@2038
|
81 |
/// The number of items stored in the heap.
|
deba@2038
|
82 |
///
|
deba@2038
|
83 |
/// \brief Returns the number of items stored in the heap.
|
deba@2038
|
84 |
int size() const { return data.size(); }
|
deba@2038
|
85 |
|
deba@2038
|
86 |
/// \brief Checks if the heap stores no items.
|
deba@2038
|
87 |
///
|
deba@2038
|
88 |
/// Returns \c true if and only if the heap stores no items.
|
deba@2038
|
89 |
bool empty() const { return data.empty(); }
|
deba@2038
|
90 |
|
deba@2038
|
91 |
/// \brief Make empty this heap.
|
deba@2038
|
92 |
///
|
deba@2038
|
93 |
/// Make empty this heap.
|
deba@2038
|
94 |
void clear() {
|
deba@2038
|
95 |
for (int i = 0; i < (int)data.size(); ++i) {
|
deba@2038
|
96 |
index[data[i].item] = -2;
|
deba@2038
|
97 |
}
|
deba@2038
|
98 |
data.clear(); first.clear(); minimal = 0;
|
deba@2038
|
99 |
}
|
deba@2038
|
100 |
|
deba@2038
|
101 |
private:
|
deba@2038
|
102 |
|
deba@2038
|
103 |
void relocate_last(int idx) {
|
deba@2038
|
104 |
if (idx + 1 < (int)data.size()) {
|
deba@2038
|
105 |
data[idx] = data.back();
|
deba@2038
|
106 |
if (data[idx].prev != -1) {
|
deba@2038
|
107 |
data[data[idx].prev].next = idx;
|
deba@2038
|
108 |
} else {
|
deba@2038
|
109 |
first[data[idx].value] = idx;
|
deba@2038
|
110 |
}
|
deba@2038
|
111 |
if (data[idx].next != -1) {
|
deba@2038
|
112 |
data[data[idx].next].prev = idx;
|
deba@2038
|
113 |
}
|
deba@2038
|
114 |
index[data[idx].item] = idx;
|
deba@2038
|
115 |
}
|
deba@2038
|
116 |
data.pop_back();
|
deba@2038
|
117 |
}
|
deba@2038
|
118 |
|
deba@2038
|
119 |
void unlace(int idx) {
|
deba@2038
|
120 |
if (data[idx].prev != -1) {
|
deba@2038
|
121 |
data[data[idx].prev].next = data[idx].next;
|
deba@2038
|
122 |
} else {
|
deba@2038
|
123 |
first[data[idx].value] = data[idx].next;
|
deba@2038
|
124 |
}
|
deba@2038
|
125 |
if (data[idx].next != -1) {
|
deba@2038
|
126 |
data[data[idx].next].prev = data[idx].prev;
|
deba@2038
|
127 |
}
|
deba@2038
|
128 |
}
|
deba@2038
|
129 |
|
deba@2038
|
130 |
void lace(int idx) {
|
deba@2038
|
131 |
if ((int)first.size() <= data[idx].value) {
|
deba@2038
|
132 |
first.resize(data[idx].value + 1, -1);
|
deba@2038
|
133 |
}
|
deba@2038
|
134 |
data[idx].next = first[data[idx].value];
|
deba@2038
|
135 |
if (data[idx].next != -1) {
|
deba@2038
|
136 |
data[data[idx].next].prev = idx;
|
deba@2038
|
137 |
}
|
deba@2038
|
138 |
first[data[idx].value] = idx;
|
deba@2038
|
139 |
data[idx].prev = -1;
|
deba@2038
|
140 |
}
|
deba@2038
|
141 |
|
deba@2038
|
142 |
public:
|
deba@2038
|
143 |
/// \brief Insert a pair of item and priority into the heap.
|
deba@2038
|
144 |
///
|
deba@2038
|
145 |
/// Adds \c p.first to the heap with priority \c p.second.
|
deba@2038
|
146 |
/// \param p The pair to insert.
|
deba@2038
|
147 |
void push(const Pair& p) {
|
deba@2038
|
148 |
push(p.first, p.second);
|
deba@2038
|
149 |
}
|
deba@2038
|
150 |
|
deba@2038
|
151 |
/// \brief Insert an item into the heap with the given priority.
|
deba@2038
|
152 |
///
|
deba@2038
|
153 |
/// Adds \c i to the heap with priority \c p.
|
deba@2038
|
154 |
/// \param i The item to insert.
|
deba@2038
|
155 |
/// \param p The priority of the item.
|
deba@2038
|
156 |
void push(const Item &i, const Prio &p) {
|
deba@2038
|
157 |
int idx = data.size();
|
deba@2038
|
158 |
index[i] = idx;
|
deba@2038
|
159 |
data.push_back(BucketItem(i, p));
|
deba@2038
|
160 |
lace(idx);
|
deba@2038
|
161 |
if (p < minimal) {
|
deba@2038
|
162 |
minimal = p;
|
deba@2038
|
163 |
}
|
deba@2038
|
164 |
}
|
deba@2038
|
165 |
|
deba@2038
|
166 |
/// \brief Returns the item with minimum priority.
|
deba@2038
|
167 |
///
|
deba@2038
|
168 |
/// This method returns the item with minimum priority.
|
deba@2038
|
169 |
/// \pre The heap must be nonempty.
|
deba@2038
|
170 |
Item top() const {
|
deba@2038
|
171 |
while (first[minimal] == -1) {
|
deba@2038
|
172 |
++minimal;
|
deba@2038
|
173 |
}
|
deba@2038
|
174 |
return data[first[minimal]].item;
|
deba@2038
|
175 |
}
|
deba@2038
|
176 |
|
deba@2038
|
177 |
/// \brief Returns the minimum priority.
|
deba@2038
|
178 |
///
|
deba@2038
|
179 |
/// It returns the minimum priority.
|
deba@2038
|
180 |
/// \pre The heap must be nonempty.
|
deba@2038
|
181 |
Prio prio() const {
|
deba@2038
|
182 |
while (first[minimal] == -1) {
|
deba@2038
|
183 |
++minimal;
|
deba@2038
|
184 |
}
|
deba@2038
|
185 |
return minimal;
|
deba@2038
|
186 |
}
|
deba@2038
|
187 |
|
deba@2038
|
188 |
/// \brief Deletes the item with minimum priority.
|
deba@2038
|
189 |
///
|
deba@2038
|
190 |
/// This method deletes the item with minimum priority from the heap.
|
deba@2038
|
191 |
/// \pre The heap must be non-empty.
|
deba@2038
|
192 |
void pop() {
|
deba@2038
|
193 |
while (first[minimal] == -1) {
|
deba@2038
|
194 |
++minimal;
|
deba@2038
|
195 |
}
|
deba@2038
|
196 |
int idx = first[minimal];
|
deba@2038
|
197 |
index[data[idx].item] = -2;
|
deba@2038
|
198 |
unlace(idx);
|
deba@2038
|
199 |
relocate_last(idx);
|
deba@2038
|
200 |
}
|
deba@2038
|
201 |
|
deba@2038
|
202 |
/// \brief Deletes \c i from the heap.
|
deba@2038
|
203 |
///
|
deba@2038
|
204 |
/// This method deletes item \c i from the heap, if \c i was
|
deba@2038
|
205 |
/// already stored in the heap.
|
deba@2038
|
206 |
/// \param i The item to erase.
|
deba@2038
|
207 |
void erase(const Item &i) {
|
deba@2038
|
208 |
int idx = index[i];
|
deba@2038
|
209 |
index[data[idx].item] = -2;
|
deba@2038
|
210 |
unlace(idx);
|
deba@2038
|
211 |
relocate_last(idx);
|
deba@2038
|
212 |
}
|
deba@2038
|
213 |
|
deba@2038
|
214 |
|
deba@2038
|
215 |
/// \brief Returns the priority of \c i.
|
deba@2038
|
216 |
///
|
deba@2038
|
217 |
/// This function returns the priority of item \c i.
|
deba@2038
|
218 |
/// \pre \c i must be in the heap.
|
deba@2038
|
219 |
/// \param i The item.
|
deba@2038
|
220 |
Prio operator[](const Item &i) const {
|
deba@2038
|
221 |
int idx = index[i];
|
deba@2038
|
222 |
return data[idx].value;
|
deba@2038
|
223 |
}
|
deba@2038
|
224 |
|
deba@2038
|
225 |
/// \brief \c i gets to the heap with priority \c p independently
|
deba@2038
|
226 |
/// if \c i was already there.
|
deba@2038
|
227 |
///
|
deba@2038
|
228 |
/// This method calls \ref push(\c i, \c p) if \c i is not stored
|
deba@2038
|
229 |
/// in the heap and sets the priority of \c i to \c p otherwise.
|
deba@2038
|
230 |
/// \param i The item.
|
deba@2038
|
231 |
/// \param p The priority.
|
deba@2038
|
232 |
void set(const Item &i, const Prio &p) {
|
deba@2038
|
233 |
int idx = index[i];
|
deba@2038
|
234 |
if (idx < 0) {
|
deba@2038
|
235 |
push(i,p);
|
deba@2038
|
236 |
} else if (p > data[idx].value) {
|
deba@2038
|
237 |
increase(i, p);
|
deba@2038
|
238 |
} else {
|
deba@2038
|
239 |
decrease(i, p);
|
deba@2038
|
240 |
}
|
deba@2038
|
241 |
}
|
deba@2038
|
242 |
|
deba@2038
|
243 |
/// \brief Decreases the priority of \c i to \c p.
|
deba@2038
|
244 |
|
deba@2038
|
245 |
/// This method decreases the priority of item \c i to \c p.
|
deba@2038
|
246 |
/// \pre \c i must be stored in the heap with priority at least \c
|
deba@2038
|
247 |
/// p relative to \c Compare.
|
deba@2038
|
248 |
/// \param i The item.
|
deba@2038
|
249 |
/// \param p The priority.
|
deba@2038
|
250 |
void decrease(const Item &i, const Prio &p) {
|
deba@2038
|
251 |
int idx = index[i];
|
deba@2038
|
252 |
unlace(idx);
|
deba@2038
|
253 |
data[idx].value = p;
|
deba@2038
|
254 |
if (p < minimal) {
|
deba@2038
|
255 |
minimal = p;
|
deba@2038
|
256 |
}
|
deba@2038
|
257 |
lace(idx);
|
deba@2038
|
258 |
}
|
deba@2038
|
259 |
|
deba@2038
|
260 |
/// \brief Increases the priority of \c i to \c p.
|
deba@2038
|
261 |
///
|
deba@2038
|
262 |
/// This method sets the priority of item \c i to \c p.
|
deba@2038
|
263 |
/// \pre \c i must be stored in the heap with priority at most \c
|
deba@2038
|
264 |
/// p relative to \c Compare.
|
deba@2038
|
265 |
/// \param i The item.
|
deba@2038
|
266 |
/// \param p The priority.
|
deba@2038
|
267 |
void increase(const Item &i, const Prio &p) {
|
deba@2038
|
268 |
int idx = index[i];
|
deba@2038
|
269 |
unlace(idx);
|
deba@2038
|
270 |
data[idx].value = p;
|
deba@2038
|
271 |
lace(idx);
|
deba@2038
|
272 |
}
|
deba@2038
|
273 |
|
deba@2038
|
274 |
/// \brief Returns if \c item is in, has already been in, or has
|
deba@2038
|
275 |
/// never been in the heap.
|
deba@2038
|
276 |
///
|
deba@2038
|
277 |
/// This method returns PRE_HEAP if \c item has never been in the
|
deba@2038
|
278 |
/// heap, IN_HEAP if it is in the heap at the moment, and POST_HEAP
|
deba@2038
|
279 |
/// otherwise. In the latter case it is possible that \c item will
|
deba@2038
|
280 |
/// get back to the heap again.
|
deba@2038
|
281 |
/// \param i The item.
|
deba@2038
|
282 |
state_enum state(const Item &i) const {
|
deba@2038
|
283 |
int idx = index[i];
|
deba@2038
|
284 |
if (idx >= 0) idx = 0;
|
deba@2038
|
285 |
return state_enum(idx);
|
deba@2038
|
286 |
}
|
deba@2038
|
287 |
|
deba@2038
|
288 |
/// \brief Sets the state of the \c item in the heap.
|
deba@2038
|
289 |
///
|
deba@2038
|
290 |
/// Sets the state of the \c item in the heap. It can be used to
|
deba@2038
|
291 |
/// manually clear the heap when it is important to achive the
|
deba@2038
|
292 |
/// better time complexity.
|
deba@2038
|
293 |
/// \param i The item.
|
deba@2038
|
294 |
/// \param st The state. It should not be \c IN_HEAP.
|
deba@2038
|
295 |
void state(const Item& i, state_enum st) {
|
deba@2038
|
296 |
switch (st) {
|
deba@2038
|
297 |
case POST_HEAP:
|
deba@2038
|
298 |
case PRE_HEAP:
|
deba@2038
|
299 |
if (state(i) == IN_HEAP) {
|
deba@2038
|
300 |
erase(i);
|
deba@2038
|
301 |
}
|
deba@2038
|
302 |
index[i] = st;
|
deba@2038
|
303 |
break;
|
deba@2038
|
304 |
case IN_HEAP:
|
deba@2038
|
305 |
break;
|
deba@2038
|
306 |
}
|
deba@2038
|
307 |
}
|
deba@2038
|
308 |
|
deba@2038
|
309 |
private:
|
deba@2038
|
310 |
|
deba@2038
|
311 |
struct BucketItem {
|
deba@2038
|
312 |
BucketItem(const Item& _item, int _value)
|
deba@2038
|
313 |
: item(_item), value(_value) {}
|
deba@2038
|
314 |
|
deba@2038
|
315 |
Item item;
|
deba@2038
|
316 |
int value;
|
deba@2038
|
317 |
|
deba@2038
|
318 |
int prev, next;
|
deba@2038
|
319 |
};
|
deba@2038
|
320 |
|
deba@2038
|
321 |
ItemIntMap& index;
|
deba@2038
|
322 |
std::vector<int> first;
|
deba@2038
|
323 |
std::vector<BucketItem> data;
|
deba@2038
|
324 |
mutable int minimal;
|
deba@2038
|
325 |
|
deba@2038
|
326 |
}; // class BucketHeap
|
deba@2038
|
327 |
|
deba@2038
|
328 |
|
deba@2038
|
329 |
template <typename _Item, typename _ItemIntMap>
|
deba@2038
|
330 |
class BucketHeap<_Item, _ItemIntMap, false> {
|
deba@2038
|
331 |
|
deba@2038
|
332 |
public:
|
deba@2038
|
333 |
typedef _Item Item;
|
deba@2038
|
334 |
typedef int Prio;
|
deba@2038
|
335 |
typedef std::pair<Item, Prio> Pair;
|
deba@2038
|
336 |
typedef _ItemIntMap ItemIntMap;
|
deba@2038
|
337 |
|
deba@2038
|
338 |
enum state_enum {
|
deba@2038
|
339 |
IN_HEAP = 0,
|
deba@2038
|
340 |
PRE_HEAP = -1,
|
deba@2038
|
341 |
POST_HEAP = -2
|
deba@2038
|
342 |
};
|
deba@2038
|
343 |
|
deba@2038
|
344 |
public:
|
deba@2038
|
345 |
|
deba@2038
|
346 |
explicit BucketHeap(ItemIntMap &_index) : index(_index), maximal(-1) {}
|
deba@2038
|
347 |
|
deba@2038
|
348 |
int size() const { return data.size(); }
|
deba@2038
|
349 |
bool empty() const { return data.empty(); }
|
deba@2038
|
350 |
|
deba@2038
|
351 |
void clear() {
|
deba@2038
|
352 |
for (int i = 0; i < (int)data.size(); ++i) {
|
deba@2038
|
353 |
index[data[i].item] = -2;
|
deba@2038
|
354 |
}
|
deba@2038
|
355 |
data.clear(); first.clear(); maximal = -1;
|
deba@2038
|
356 |
}
|
deba@2038
|
357 |
|
deba@2038
|
358 |
private:
|
deba@2038
|
359 |
|
deba@2038
|
360 |
void relocate_last(int idx) {
|
deba@2038
|
361 |
if (idx + 1 != (int)data.size()) {
|
deba@2038
|
362 |
data[idx] = data.back();
|
deba@2038
|
363 |
if (data[idx].prev != -1) {
|
deba@2038
|
364 |
data[data[idx].prev].next = idx;
|
deba@2038
|
365 |
} else {
|
deba@2038
|
366 |
first[data[idx].value] = idx;
|
deba@2038
|
367 |
}
|
deba@2038
|
368 |
if (data[idx].next != -1) {
|
deba@2038
|
369 |
data[data[idx].next].prev = idx;
|
deba@2038
|
370 |
}
|
deba@2038
|
371 |
index[data[idx].item] = idx;
|
deba@2038
|
372 |
}
|
deba@2038
|
373 |
data.pop_back();
|
deba@2038
|
374 |
}
|
deba@2038
|
375 |
|
deba@2038
|
376 |
void unlace(int idx) {
|
deba@2038
|
377 |
if (data[idx].prev != -1) {
|
deba@2038
|
378 |
data[data[idx].prev].next = data[idx].next;
|
deba@2038
|
379 |
} else {
|
deba@2038
|
380 |
first[data[idx].value] = data[idx].next;
|
deba@2038
|
381 |
}
|
deba@2038
|
382 |
if (data[idx].next != -1) {
|
deba@2038
|
383 |
data[data[idx].next].prev = data[idx].prev;
|
deba@2038
|
384 |
}
|
deba@2038
|
385 |
}
|
deba@2038
|
386 |
|
deba@2038
|
387 |
void lace(int idx) {
|
deba@2038
|
388 |
if ((int)first.size() <= data[idx].value) {
|
deba@2038
|
389 |
first.resize(data[idx].value + 1, -1);
|
deba@2038
|
390 |
}
|
deba@2038
|
391 |
data[idx].next = first[data[idx].value];
|
deba@2038
|
392 |
if (data[idx].next != -1) {
|
deba@2038
|
393 |
data[data[idx].next].prev = idx;
|
deba@2038
|
394 |
}
|
deba@2038
|
395 |
first[data[idx].value] = idx;
|
deba@2038
|
396 |
data[idx].prev = -1;
|
deba@2038
|
397 |
}
|
deba@2038
|
398 |
|
deba@2038
|
399 |
public:
|
deba@2038
|
400 |
|
deba@2038
|
401 |
void push(const Pair& p) {
|
deba@2038
|
402 |
push(p.first, p.second);
|
deba@2038
|
403 |
}
|
deba@2038
|
404 |
|
deba@2038
|
405 |
void push(const Item &i, const Prio &p) {
|
deba@2038
|
406 |
int idx = data.size();
|
deba@2038
|
407 |
index[i] = idx;
|
deba@2038
|
408 |
data.push_back(BucketItem(i, p));
|
deba@2038
|
409 |
lace(idx);
|
deba@2038
|
410 |
if (data[idx].value > maximal) {
|
deba@2038
|
411 |
maximal = data[idx].value;
|
deba@2038
|
412 |
}
|
deba@2038
|
413 |
}
|
deba@2038
|
414 |
|
deba@2038
|
415 |
Item top() const {
|
deba@2038
|
416 |
while (first[maximal] == -1) {
|
deba@2038
|
417 |
--maximal;
|
deba@2038
|
418 |
}
|
deba@2038
|
419 |
return data[first[maximal]].item;
|
deba@2038
|
420 |
}
|
deba@2038
|
421 |
|
deba@2038
|
422 |
Prio prio() const {
|
deba@2038
|
423 |
while (first[maximal] == -1) {
|
deba@2038
|
424 |
--maximal;
|
deba@2038
|
425 |
}
|
deba@2038
|
426 |
return maximal;
|
deba@2038
|
427 |
}
|
deba@2038
|
428 |
|
deba@2038
|
429 |
void pop() {
|
deba@2038
|
430 |
while (first[maximal] == -1) {
|
deba@2038
|
431 |
--maximal;
|
deba@2038
|
432 |
}
|
deba@2038
|
433 |
int idx = first[maximal];
|
deba@2038
|
434 |
index[data[idx].item] = -2;
|
deba@2038
|
435 |
unlace(idx);
|
deba@2038
|
436 |
relocate_last(idx);
|
deba@2038
|
437 |
}
|
deba@2038
|
438 |
|
deba@2038
|
439 |
void erase(const Item &i) {
|
deba@2038
|
440 |
int idx = index[i];
|
deba@2038
|
441 |
index[data[idx].item] = -2;
|
deba@2038
|
442 |
unlace(idx);
|
deba@2038
|
443 |
relocate_last(idx);
|
deba@2038
|
444 |
}
|
deba@2038
|
445 |
|
deba@2038
|
446 |
Prio operator[](const Item &i) const {
|
deba@2038
|
447 |
int idx = index[i];
|
deba@2038
|
448 |
return data[idx].value;
|
deba@2038
|
449 |
}
|
deba@2038
|
450 |
|
deba@2038
|
451 |
void set(const Item &i, const Prio &p) {
|
deba@2038
|
452 |
int idx = index[i];
|
deba@2038
|
453 |
if (idx < 0) {
|
deba@2038
|
454 |
push(i,p);
|
deba@2038
|
455 |
} else if (p > data[idx].value) {
|
deba@2038
|
456 |
decrease(i, p);
|
deba@2038
|
457 |
} else {
|
deba@2038
|
458 |
increase(i, p);
|
deba@2038
|
459 |
}
|
deba@2038
|
460 |
}
|
deba@2038
|
461 |
|
deba@2038
|
462 |
void decrease(const Item &i, const Prio &p) {
|
deba@2038
|
463 |
int idx = index[i];
|
deba@2038
|
464 |
unlace(idx);
|
deba@2038
|
465 |
data[idx].value = p;
|
deba@2038
|
466 |
if (p > maximal) {
|
deba@2038
|
467 |
maximal = p;
|
deba@2038
|
468 |
}
|
deba@2038
|
469 |
lace(idx);
|
deba@2038
|
470 |
}
|
deba@2038
|
471 |
|
deba@2038
|
472 |
void increase(const Item &i, const Prio &p) {
|
deba@2038
|
473 |
int idx = index[i];
|
deba@2038
|
474 |
unlace(idx);
|
deba@2038
|
475 |
data[idx].value = p;
|
deba@2038
|
476 |
lace(idx);
|
deba@2038
|
477 |
}
|
deba@2038
|
478 |
|
deba@2038
|
479 |
state_enum state(const Item &i) const {
|
deba@2038
|
480 |
int idx = index[i];
|
deba@2038
|
481 |
if (idx >= 0) idx = 0;
|
deba@2038
|
482 |
return state_enum(idx);
|
deba@2038
|
483 |
}
|
deba@2038
|
484 |
|
deba@2038
|
485 |
void state(const Item& i, state_enum st) {
|
deba@2038
|
486 |
switch (st) {
|
deba@2038
|
487 |
case POST_HEAP:
|
deba@2038
|
488 |
case PRE_HEAP:
|
deba@2038
|
489 |
if (state(i) == IN_HEAP) {
|
deba@2038
|
490 |
erase(i);
|
deba@2038
|
491 |
}
|
deba@2038
|
492 |
index[i] = st;
|
deba@2038
|
493 |
break;
|
deba@2038
|
494 |
case IN_HEAP:
|
deba@2038
|
495 |
break;
|
deba@2038
|
496 |
}
|
deba@2038
|
497 |
}
|
deba@2038
|
498 |
|
deba@2038
|
499 |
private:
|
deba@2038
|
500 |
|
deba@2038
|
501 |
struct BucketItem {
|
deba@2038
|
502 |
BucketItem(const Item& _item, int _value)
|
deba@2038
|
503 |
: item(_item), value(_value) {}
|
deba@2038
|
504 |
|
deba@2038
|
505 |
Item item;
|
deba@2038
|
506 |
int value;
|
deba@2038
|
507 |
|
deba@2038
|
508 |
int prev, next;
|
deba@2038
|
509 |
};
|
deba@2038
|
510 |
|
deba@2038
|
511 |
ItemIntMap& index;
|
deba@2038
|
512 |
std::vector<int> first;
|
deba@2038
|
513 |
std::vector<BucketItem> data;
|
deba@2038
|
514 |
mutable int maximal;
|
deba@2038
|
515 |
|
deba@2038
|
516 |
}; // class BucketHeap
|
deba@2038
|
517 |
|
deba@2038
|
518 |
}
|
deba@2038
|
519 |
|
deba@2038
|
520 |
#endif
|