annotate gcc/fibonacci_heap.h @ 158:494b0b89df80 default tip

...
author Shinji KONO <kono@ie.u-ryukyu.ac.jp>
date Mon, 25 May 2020 18:13:55 +0900
parents 1830386684a0
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
111
kono
parents:
diff changeset
1 /* Fibonacci heap for GNU compiler.
145
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
2 Copyright (C) 1998-2020 Free Software Foundation, Inc.
111
kono
parents:
diff changeset
3 Contributed by Daniel Berlin (dan@cgsoftware.com).
kono
parents:
diff changeset
4 Re-implemented in C++ by Martin Liska <mliska@suse.cz>
kono
parents:
diff changeset
5
kono
parents:
diff changeset
6 This file is part of GCC.
kono
parents:
diff changeset
7
kono
parents:
diff changeset
8 GCC is free software; you can redistribute it and/or modify it under
kono
parents:
diff changeset
9 the terms of the GNU General Public License as published by the Free
kono
parents:
diff changeset
10 Software Foundation; either version 3, or (at your option) any later
kono
parents:
diff changeset
11 version.
kono
parents:
diff changeset
12
kono
parents:
diff changeset
13 GCC is distributed in the hope that it will be useful, but WITHOUT ANY
kono
parents:
diff changeset
14 WARRANTY; without even the implied warranty of MERCHANTABILITY or
kono
parents:
diff changeset
15 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
kono
parents:
diff changeset
16 for more details.
kono
parents:
diff changeset
17
kono
parents:
diff changeset
18 You should have received a copy of the GNU General Public License
kono
parents:
diff changeset
19 along with GCC; see the file COPYING3. If not see
kono
parents:
diff changeset
20 <http://www.gnu.org/licenses/>. */
kono
parents:
diff changeset
21
kono
parents:
diff changeset
22 /* Fibonacci heaps are somewhat complex, but, there's an article in
kono
parents:
diff changeset
23 DDJ that explains them pretty well:
kono
parents:
diff changeset
24
kono
parents:
diff changeset
25 http://www.ddj.com/articles/1997/9701/9701o/9701o.htm?topic=algoritms
kono
parents:
diff changeset
26
kono
parents:
diff changeset
27 Introduction to algorithms by Corman and Rivest also goes over them.
kono
parents:
diff changeset
28
kono
parents:
diff changeset
29 The original paper that introduced them is "Fibonacci heaps and their
kono
parents:
diff changeset
30 uses in improved network optimization algorithms" by Tarjan and
kono
parents:
diff changeset
31 Fredman (JACM 34(3), July 1987).
kono
parents:
diff changeset
32
kono
parents:
diff changeset
33 Amortized and real worst case time for operations:
kono
parents:
diff changeset
34
kono
parents:
diff changeset
35 ExtractMin: O(lg n) amortized. O(n) worst case.
kono
parents:
diff changeset
36 DecreaseKey: O(1) amortized. O(lg n) worst case.
kono
parents:
diff changeset
37 Insert: O(1) amortized.
kono
parents:
diff changeset
38 Union: O(1) amortized. */
kono
parents:
diff changeset
39
kono
parents:
diff changeset
40 #ifndef GCC_FIBONACCI_HEAP_H
kono
parents:
diff changeset
41 #define GCC_FIBONACCI_HEAP_H
kono
parents:
diff changeset
42
kono
parents:
diff changeset
43 /* Forward definition. */
kono
parents:
diff changeset
44
kono
parents:
diff changeset
45 template<class K, class V>
kono
parents:
diff changeset
46 class fibonacci_heap;
kono
parents:
diff changeset
47
kono
parents:
diff changeset
48 /* Fibonacci heap node class. */
kono
parents:
diff changeset
49
kono
parents:
diff changeset
50 template<class K, class V>
kono
parents:
diff changeset
51 class fibonacci_node
kono
parents:
diff changeset
52 {
kono
parents:
diff changeset
53 typedef fibonacci_node<K,V> fibonacci_node_t;
kono
parents:
diff changeset
54 friend class fibonacci_heap<K,V>;
kono
parents:
diff changeset
55
kono
parents:
diff changeset
56 public:
kono
parents:
diff changeset
57 /* Default constructor. */
kono
parents:
diff changeset
58 fibonacci_node (): m_parent (NULL), m_child (NULL), m_left (this),
145
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
59 m_right (this), m_data (NULL), m_degree (0), m_mark (0)
111
kono
parents:
diff changeset
60 {
kono
parents:
diff changeset
61 }
kono
parents:
diff changeset
62
kono
parents:
diff changeset
63 /* Constructor for a node with given KEY. */
kono
parents:
diff changeset
64 fibonacci_node (K key, V *data = NULL): m_parent (NULL), m_child (NULL),
kono
parents:
diff changeset
65 m_left (this), m_right (this), m_key (key), m_data (data),
kono
parents:
diff changeset
66 m_degree (0), m_mark (0)
kono
parents:
diff changeset
67 {
kono
parents:
diff changeset
68 }
kono
parents:
diff changeset
69
kono
parents:
diff changeset
70 /* Compare fibonacci node with OTHER node. */
kono
parents:
diff changeset
71 int compare (fibonacci_node_t *other)
kono
parents:
diff changeset
72 {
kono
parents:
diff changeset
73 if (m_key < other->m_key)
kono
parents:
diff changeset
74 return -1;
kono
parents:
diff changeset
75 if (m_key > other->m_key)
kono
parents:
diff changeset
76 return 1;
kono
parents:
diff changeset
77 return 0;
kono
parents:
diff changeset
78 }
kono
parents:
diff changeset
79
kono
parents:
diff changeset
80 /* Compare the node with a given KEY. */
kono
parents:
diff changeset
81 int compare_data (K key)
kono
parents:
diff changeset
82 {
kono
parents:
diff changeset
83 return fibonacci_node_t (key).compare (this);
kono
parents:
diff changeset
84 }
kono
parents:
diff changeset
85
kono
parents:
diff changeset
86 /* Remove fibonacci heap node. */
kono
parents:
diff changeset
87 fibonacci_node_t *remove ();
kono
parents:
diff changeset
88
kono
parents:
diff changeset
89 /* Link the node with PARENT. */
kono
parents:
diff changeset
90 void link (fibonacci_node_t *parent);
kono
parents:
diff changeset
91
kono
parents:
diff changeset
92 /* Return key associated with the node. */
kono
parents:
diff changeset
93 K get_key ()
kono
parents:
diff changeset
94 {
kono
parents:
diff changeset
95 return m_key;
kono
parents:
diff changeset
96 }
kono
parents:
diff changeset
97
kono
parents:
diff changeset
98 /* Return data associated with the node. */
kono
parents:
diff changeset
99 V *get_data ()
kono
parents:
diff changeset
100 {
kono
parents:
diff changeset
101 return m_data;
kono
parents:
diff changeset
102 }
kono
parents:
diff changeset
103
kono
parents:
diff changeset
104 private:
kono
parents:
diff changeset
105 /* Put node B after this node. */
kono
parents:
diff changeset
106 void insert_after (fibonacci_node_t *b);
kono
parents:
diff changeset
107
kono
parents:
diff changeset
108 /* Insert fibonacci node B after this node. */
kono
parents:
diff changeset
109 void insert_before (fibonacci_node_t *b)
kono
parents:
diff changeset
110 {
kono
parents:
diff changeset
111 m_left->insert_after (b);
kono
parents:
diff changeset
112 }
kono
parents:
diff changeset
113
kono
parents:
diff changeset
114 /* Parent node. */
kono
parents:
diff changeset
115 fibonacci_node *m_parent;
kono
parents:
diff changeset
116 /* Child node. */
kono
parents:
diff changeset
117 fibonacci_node *m_child;
kono
parents:
diff changeset
118 /* Left sibling. */
kono
parents:
diff changeset
119 fibonacci_node *m_left;
kono
parents:
diff changeset
120 /* Right node. */
kono
parents:
diff changeset
121 fibonacci_node *m_right;
kono
parents:
diff changeset
122 /* Key associated with node. */
kono
parents:
diff changeset
123 K m_key;
kono
parents:
diff changeset
124 /* Data associated with node. */
kono
parents:
diff changeset
125 V *m_data;
kono
parents:
diff changeset
126
kono
parents:
diff changeset
127 #if defined (__GNUC__) && (!defined (SIZEOF_INT) || SIZEOF_INT < 4)
kono
parents:
diff changeset
128 /* Degree of the node. */
kono
parents:
diff changeset
129 __extension__ unsigned long int m_degree : 31;
kono
parents:
diff changeset
130 /* Mark of the node. */
kono
parents:
diff changeset
131 __extension__ unsigned long int m_mark : 1;
kono
parents:
diff changeset
132 #else
kono
parents:
diff changeset
133 /* Degree of the node. */
kono
parents:
diff changeset
134 unsigned int m_degree : 31;
kono
parents:
diff changeset
135 /* Mark of the node. */
kono
parents:
diff changeset
136 unsigned int m_mark : 1;
kono
parents:
diff changeset
137 #endif
kono
parents:
diff changeset
138 };
kono
parents:
diff changeset
139
kono
parents:
diff changeset
140 /* Fibonacci heap class. */
kono
parents:
diff changeset
141 template<class K, class V>
kono
parents:
diff changeset
142 class fibonacci_heap
kono
parents:
diff changeset
143 {
kono
parents:
diff changeset
144 typedef fibonacci_node<K,V> fibonacci_node_t;
kono
parents:
diff changeset
145 friend class fibonacci_node<K,V>;
kono
parents:
diff changeset
146
kono
parents:
diff changeset
147 public:
145
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
148 /* Default constructor. ALLOCATOR is optional and is primarily useful
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
149 when heaps are going to be merged (in that case they need to be allocated
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
150 in same alloc pool). */
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
151 fibonacci_heap (K global_min_key, pool_allocator *allocator = NULL):
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
152 m_nodes (0), m_min (NULL), m_root (NULL),
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
153 m_global_min_key (global_min_key),
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
154 m_allocator (allocator), m_own_allocator (false)
111
kono
parents:
diff changeset
155 {
145
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
156 if (!m_allocator)
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
157 {
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
158 m_allocator = new pool_allocator ("Fibonacci heap",
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
159 sizeof (fibonacci_node_t));
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
160 m_own_allocator = true;
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
161 }
111
kono
parents:
diff changeset
162 }
kono
parents:
diff changeset
163
kono
parents:
diff changeset
164 /* Destructor. */
kono
parents:
diff changeset
165 ~fibonacci_heap ()
kono
parents:
diff changeset
166 {
145
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
167 /* Actual memory will be released by the destructor of m_allocator. */
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
168 if (need_finalization_p<fibonacci_node_t> () || !m_own_allocator)
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
169 while (m_min != NULL)
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
170 {
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
171 fibonacci_node_t *n = extract_minimum_node ();
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
172 n->~fibonacci_node_t ();
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
173 if (!m_own_allocator)
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
174 m_allocator->remove (n);
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
175 }
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
176 if (m_own_allocator)
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
177 delete m_allocator;
111
kono
parents:
diff changeset
178 }
kono
parents:
diff changeset
179
kono
parents:
diff changeset
180 /* Insert new node given by KEY and DATA associated with the key. */
kono
parents:
diff changeset
181 fibonacci_node_t *insert (K key, V *data);
kono
parents:
diff changeset
182
kono
parents:
diff changeset
183 /* Return true if no entry is present. */
145
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
184 bool empty () const
111
kono
parents:
diff changeset
185 {
kono
parents:
diff changeset
186 return m_nodes == 0;
kono
parents:
diff changeset
187 }
kono
parents:
diff changeset
188
kono
parents:
diff changeset
189 /* Return the number of nodes. */
145
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
190 size_t nodes () const
111
kono
parents:
diff changeset
191 {
kono
parents:
diff changeset
192 return m_nodes;
kono
parents:
diff changeset
193 }
kono
parents:
diff changeset
194
kono
parents:
diff changeset
195 /* Return minimal key presented in the heap. */
145
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
196 K min_key () const
111
kono
parents:
diff changeset
197 {
kono
parents:
diff changeset
198 if (m_min == NULL)
kono
parents:
diff changeset
199 gcc_unreachable ();
kono
parents:
diff changeset
200
kono
parents:
diff changeset
201 return m_min->m_key;
kono
parents:
diff changeset
202 }
kono
parents:
diff changeset
203
kono
parents:
diff changeset
204 /* For given NODE, set new KEY value. */
kono
parents:
diff changeset
205 K replace_key (fibonacci_node_t *node, K key)
kono
parents:
diff changeset
206 {
kono
parents:
diff changeset
207 K okey = node->m_key;
kono
parents:
diff changeset
208
kono
parents:
diff changeset
209 replace_key_data (node, key, node->m_data);
kono
parents:
diff changeset
210 return okey;
kono
parents:
diff changeset
211 }
kono
parents:
diff changeset
212
kono
parents:
diff changeset
213 /* For given NODE, decrease value to new KEY. */
kono
parents:
diff changeset
214 K decrease_key (fibonacci_node_t *node, K key)
kono
parents:
diff changeset
215 {
kono
parents:
diff changeset
216 gcc_assert (key <= node->m_key);
kono
parents:
diff changeset
217 return replace_key (node, key);
kono
parents:
diff changeset
218 }
kono
parents:
diff changeset
219
kono
parents:
diff changeset
220 /* For given NODE, set new KEY and DATA value. */
kono
parents:
diff changeset
221 V *replace_key_data (fibonacci_node_t *node, K key, V *data);
kono
parents:
diff changeset
222
kono
parents:
diff changeset
223 /* Extract minimum node in the heap. If RELEASE is specified,
kono
parents:
diff changeset
224 memory is released. */
kono
parents:
diff changeset
225 V *extract_min (bool release = true);
kono
parents:
diff changeset
226
kono
parents:
diff changeset
227 /* Return value associated with minimum node in the heap. */
145
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
228 V *min () const
111
kono
parents:
diff changeset
229 {
kono
parents:
diff changeset
230 if (m_min == NULL)
kono
parents:
diff changeset
231 return NULL;
kono
parents:
diff changeset
232
kono
parents:
diff changeset
233 return m_min->m_data;
kono
parents:
diff changeset
234 }
kono
parents:
diff changeset
235
kono
parents:
diff changeset
236 /* Replace data associated with NODE and replace it with DATA. */
kono
parents:
diff changeset
237 V *replace_data (fibonacci_node_t *node, V *data)
kono
parents:
diff changeset
238 {
kono
parents:
diff changeset
239 return replace_key_data (node, node->m_key, data);
kono
parents:
diff changeset
240 }
kono
parents:
diff changeset
241
kono
parents:
diff changeset
242 /* Delete NODE in the heap. */
kono
parents:
diff changeset
243 V *delete_node (fibonacci_node_t *node, bool release = true);
kono
parents:
diff changeset
244
kono
parents:
diff changeset
245 /* Union the heap with HEAPB. */
kono
parents:
diff changeset
246 fibonacci_heap *union_with (fibonacci_heap *heapb);
kono
parents:
diff changeset
247
kono
parents:
diff changeset
248 private:
kono
parents:
diff changeset
249 /* Insert new NODE given by KEY and DATA associated with the key. */
kono
parents:
diff changeset
250 fibonacci_node_t *insert (fibonacci_node_t *node, K key, V *data);
kono
parents:
diff changeset
251
kono
parents:
diff changeset
252 /* Insert new NODE that has already filled key and value. */
kono
parents:
diff changeset
253 fibonacci_node_t *insert_node (fibonacci_node_t *node);
kono
parents:
diff changeset
254
kono
parents:
diff changeset
255 /* Insert it into the root list. */
kono
parents:
diff changeset
256 void insert_root (fibonacci_node_t *node);
kono
parents:
diff changeset
257
kono
parents:
diff changeset
258 /* Remove NODE from PARENT's child list. */
kono
parents:
diff changeset
259 void cut (fibonacci_node_t *node, fibonacci_node_t *parent);
kono
parents:
diff changeset
260
kono
parents:
diff changeset
261 /* Process cut of node Y and do it recursivelly. */
kono
parents:
diff changeset
262 void cascading_cut (fibonacci_node_t *y);
kono
parents:
diff changeset
263
kono
parents:
diff changeset
264 /* Extract minimum node from the heap. */
kono
parents:
diff changeset
265 fibonacci_node_t * extract_minimum_node ();
kono
parents:
diff changeset
266
kono
parents:
diff changeset
267 /* Remove root NODE from the heap. */
kono
parents:
diff changeset
268 void remove_root (fibonacci_node_t *node);
kono
parents:
diff changeset
269
kono
parents:
diff changeset
270 /* Consolidate heap. */
kono
parents:
diff changeset
271 void consolidate ();
kono
parents:
diff changeset
272
kono
parents:
diff changeset
273 /* Number of nodes. */
kono
parents:
diff changeset
274 size_t m_nodes;
kono
parents:
diff changeset
275 /* Minimum node of the heap. */
kono
parents:
diff changeset
276 fibonacci_node_t *m_min;
kono
parents:
diff changeset
277 /* Root node of the heap. */
kono
parents:
diff changeset
278 fibonacci_node_t *m_root;
kono
parents:
diff changeset
279 /* Global minimum given in the heap construction. */
kono
parents:
diff changeset
280 K m_global_min_key;
145
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
281
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
282 /* Allocator used to hold nodes. */
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
283 pool_allocator *m_allocator;
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
284 /* True if alocator is owned by the current heap only. */
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
285 bool m_own_allocator;
111
kono
parents:
diff changeset
286 };
kono
parents:
diff changeset
287
kono
parents:
diff changeset
288 /* Remove fibonacci heap node. */
kono
parents:
diff changeset
289
kono
parents:
diff changeset
290 template<class K, class V>
kono
parents:
diff changeset
291 fibonacci_node<K,V> *
kono
parents:
diff changeset
292 fibonacci_node<K,V>::remove ()
kono
parents:
diff changeset
293 {
kono
parents:
diff changeset
294 fibonacci_node<K,V> *ret;
kono
parents:
diff changeset
295
kono
parents:
diff changeset
296 if (this == m_left)
kono
parents:
diff changeset
297 ret = NULL;
kono
parents:
diff changeset
298 else
kono
parents:
diff changeset
299 ret = m_left;
kono
parents:
diff changeset
300
kono
parents:
diff changeset
301 if (m_parent != NULL && m_parent->m_child == this)
kono
parents:
diff changeset
302 m_parent->m_child = ret;
kono
parents:
diff changeset
303
kono
parents:
diff changeset
304 m_right->m_left = m_left;
kono
parents:
diff changeset
305 m_left->m_right = m_right;
kono
parents:
diff changeset
306
kono
parents:
diff changeset
307 m_parent = NULL;
kono
parents:
diff changeset
308 m_left = this;
kono
parents:
diff changeset
309 m_right = this;
kono
parents:
diff changeset
310
kono
parents:
diff changeset
311 return ret;
kono
parents:
diff changeset
312 }
kono
parents:
diff changeset
313
kono
parents:
diff changeset
314 /* Link the node with PARENT. */
kono
parents:
diff changeset
315
kono
parents:
diff changeset
316 template<class K, class V>
kono
parents:
diff changeset
317 void
kono
parents:
diff changeset
318 fibonacci_node<K,V>::link (fibonacci_node<K,V> *parent)
kono
parents:
diff changeset
319 {
kono
parents:
diff changeset
320 if (parent->m_child == NULL)
kono
parents:
diff changeset
321 parent->m_child = this;
kono
parents:
diff changeset
322 else
kono
parents:
diff changeset
323 parent->m_child->insert_before (this);
kono
parents:
diff changeset
324 m_parent = parent;
kono
parents:
diff changeset
325 parent->m_degree++;
kono
parents:
diff changeset
326 m_mark = 0;
kono
parents:
diff changeset
327 }
kono
parents:
diff changeset
328
kono
parents:
diff changeset
329 /* Put node B after this node. */
kono
parents:
diff changeset
330
kono
parents:
diff changeset
331 template<class K, class V>
kono
parents:
diff changeset
332 void
kono
parents:
diff changeset
333 fibonacci_node<K,V>::insert_after (fibonacci_node<K,V> *b)
kono
parents:
diff changeset
334 {
kono
parents:
diff changeset
335 fibonacci_node<K,V> *a = this;
kono
parents:
diff changeset
336
kono
parents:
diff changeset
337 if (a == a->m_right)
kono
parents:
diff changeset
338 {
kono
parents:
diff changeset
339 a->m_right = b;
kono
parents:
diff changeset
340 a->m_left = b;
kono
parents:
diff changeset
341 b->m_right = a;
kono
parents:
diff changeset
342 b->m_left = a;
kono
parents:
diff changeset
343 }
kono
parents:
diff changeset
344 else
kono
parents:
diff changeset
345 {
kono
parents:
diff changeset
346 b->m_right = a->m_right;
kono
parents:
diff changeset
347 a->m_right->m_left = b;
kono
parents:
diff changeset
348 a->m_right = b;
kono
parents:
diff changeset
349 b->m_left = a;
kono
parents:
diff changeset
350 }
kono
parents:
diff changeset
351 }
kono
parents:
diff changeset
352
kono
parents:
diff changeset
353 /* Insert new node given by KEY and DATA associated with the key. */
kono
parents:
diff changeset
354
kono
parents:
diff changeset
355 template<class K, class V>
kono
parents:
diff changeset
356 fibonacci_node<K,V>*
kono
parents:
diff changeset
357 fibonacci_heap<K,V>::insert (K key, V *data)
kono
parents:
diff changeset
358 {
kono
parents:
diff changeset
359 /* Create the new node. */
145
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
360 fibonacci_node<K,V> *node = new (m_allocator->allocate ())
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
361 fibonacci_node_t (key, data);
111
kono
parents:
diff changeset
362
kono
parents:
diff changeset
363 return insert_node (node);
kono
parents:
diff changeset
364 }
kono
parents:
diff changeset
365
kono
parents:
diff changeset
366 /* Insert new NODE given by DATA associated with the key. */
kono
parents:
diff changeset
367
kono
parents:
diff changeset
368 template<class K, class V>
kono
parents:
diff changeset
369 fibonacci_node<K,V>*
kono
parents:
diff changeset
370 fibonacci_heap<K,V>::insert (fibonacci_node_t *node, K key, V *data)
kono
parents:
diff changeset
371 {
kono
parents:
diff changeset
372 /* Set the node's data. */
kono
parents:
diff changeset
373 node->m_data = data;
kono
parents:
diff changeset
374 node->m_key = key;
kono
parents:
diff changeset
375
kono
parents:
diff changeset
376 return insert_node (node);
kono
parents:
diff changeset
377 }
kono
parents:
diff changeset
378
kono
parents:
diff changeset
379 /* Insert new NODE that has already filled key and value. */
kono
parents:
diff changeset
380
kono
parents:
diff changeset
381 template<class K, class V>
kono
parents:
diff changeset
382 fibonacci_node<K,V>*
kono
parents:
diff changeset
383 fibonacci_heap<K,V>::insert_node (fibonacci_node_t *node)
kono
parents:
diff changeset
384 {
kono
parents:
diff changeset
385 /* Insert it into the root list. */
kono
parents:
diff changeset
386 insert_root (node);
kono
parents:
diff changeset
387
kono
parents:
diff changeset
388 /* If their was no minimum, or this key is less than the min,
kono
parents:
diff changeset
389 it's the new min. */
kono
parents:
diff changeset
390 if (m_min == NULL || node->m_key < m_min->m_key)
kono
parents:
diff changeset
391 m_min = node;
kono
parents:
diff changeset
392
kono
parents:
diff changeset
393 m_nodes++;
kono
parents:
diff changeset
394
kono
parents:
diff changeset
395 return node;
kono
parents:
diff changeset
396 }
kono
parents:
diff changeset
397
kono
parents:
diff changeset
398 /* For given NODE, set new KEY and DATA value. */
kono
parents:
diff changeset
399
kono
parents:
diff changeset
400 template<class K, class V>
kono
parents:
diff changeset
401 V*
kono
parents:
diff changeset
402 fibonacci_heap<K,V>::replace_key_data (fibonacci_node<K,V> *node, K key,
kono
parents:
diff changeset
403 V *data)
kono
parents:
diff changeset
404 {
kono
parents:
diff changeset
405 K okey;
kono
parents:
diff changeset
406 fibonacci_node<K,V> *y;
kono
parents:
diff changeset
407 V *odata = node->m_data;
kono
parents:
diff changeset
408
kono
parents:
diff changeset
409 /* If we wanted to, we do a real increase by redeleting and
kono
parents:
diff changeset
410 inserting. */
kono
parents:
diff changeset
411 if (node->compare_data (key) > 0)
kono
parents:
diff changeset
412 {
kono
parents:
diff changeset
413 delete_node (node, false);
kono
parents:
diff changeset
414
kono
parents:
diff changeset
415 node = new (node) fibonacci_node_t ();
kono
parents:
diff changeset
416 insert (node, key, data);
kono
parents:
diff changeset
417
kono
parents:
diff changeset
418 return odata;
kono
parents:
diff changeset
419 }
kono
parents:
diff changeset
420
kono
parents:
diff changeset
421 okey = node->m_key;
kono
parents:
diff changeset
422 node->m_data = data;
kono
parents:
diff changeset
423 node->m_key = key;
kono
parents:
diff changeset
424 y = node->m_parent;
kono
parents:
diff changeset
425
kono
parents:
diff changeset
426 /* Short-circuit if the key is the same, as we then don't have to
kono
parents:
diff changeset
427 do anything. Except if we're trying to force the new node to
kono
parents:
diff changeset
428 be the new minimum for delete. */
kono
parents:
diff changeset
429 if (okey == key && okey != m_global_min_key)
kono
parents:
diff changeset
430 return odata;
kono
parents:
diff changeset
431
kono
parents:
diff changeset
432 /* These two compares are specifically <= 0 to make sure that in the case
kono
parents:
diff changeset
433 of equality, a node we replaced the data on, becomes the new min. This
kono
parents:
diff changeset
434 is needed so that delete's call to extractmin gets the right node. */
kono
parents:
diff changeset
435 if (y != NULL && node->compare (y) <= 0)
kono
parents:
diff changeset
436 {
kono
parents:
diff changeset
437 cut (node, y);
kono
parents:
diff changeset
438 cascading_cut (y);
kono
parents:
diff changeset
439 }
kono
parents:
diff changeset
440
kono
parents:
diff changeset
441 if (node->compare (m_min) <= 0)
kono
parents:
diff changeset
442 m_min = node;
kono
parents:
diff changeset
443
kono
parents:
diff changeset
444 return odata;
kono
parents:
diff changeset
445 }
kono
parents:
diff changeset
446
kono
parents:
diff changeset
447 /* Extract minimum node in the heap. Delete fibonacci node if RELEASE
kono
parents:
diff changeset
448 is true. */
kono
parents:
diff changeset
449
kono
parents:
diff changeset
450 template<class K, class V>
kono
parents:
diff changeset
451 V*
kono
parents:
diff changeset
452 fibonacci_heap<K,V>::extract_min (bool release)
kono
parents:
diff changeset
453 {
kono
parents:
diff changeset
454 fibonacci_node<K,V> *z;
kono
parents:
diff changeset
455 V *ret = NULL;
kono
parents:
diff changeset
456
kono
parents:
diff changeset
457 /* If we don't have a min set, it means we have no nodes. */
kono
parents:
diff changeset
458 if (m_min != NULL)
kono
parents:
diff changeset
459 {
kono
parents:
diff changeset
460 /* Otherwise, extract the min node, free the node, and return the
kono
parents:
diff changeset
461 node's data. */
kono
parents:
diff changeset
462 z = extract_minimum_node ();
kono
parents:
diff changeset
463 ret = z->m_data;
kono
parents:
diff changeset
464
kono
parents:
diff changeset
465 if (release)
145
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
466 {
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
467 z->~fibonacci_node_t ();
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
468 m_allocator->remove (z);
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
469 }
111
kono
parents:
diff changeset
470 }
kono
parents:
diff changeset
471
kono
parents:
diff changeset
472 return ret;
kono
parents:
diff changeset
473 }
kono
parents:
diff changeset
474
kono
parents:
diff changeset
475 /* Delete NODE in the heap, if RELEASE is specified memory is released. */
kono
parents:
diff changeset
476
kono
parents:
diff changeset
477 template<class K, class V>
kono
parents:
diff changeset
478 V*
kono
parents:
diff changeset
479 fibonacci_heap<K,V>::delete_node (fibonacci_node<K,V> *node, bool release)
kono
parents:
diff changeset
480 {
kono
parents:
diff changeset
481 V *ret = node->m_data;
kono
parents:
diff changeset
482
kono
parents:
diff changeset
483 /* To perform delete, we just make it the min key, and extract. */
kono
parents:
diff changeset
484 replace_key (node, m_global_min_key);
kono
parents:
diff changeset
485 if (node != m_min)
kono
parents:
diff changeset
486 {
kono
parents:
diff changeset
487 fprintf (stderr, "Can't force minimum on fibheap.\n");
kono
parents:
diff changeset
488 abort ();
kono
parents:
diff changeset
489 }
kono
parents:
diff changeset
490 extract_min (release);
kono
parents:
diff changeset
491
kono
parents:
diff changeset
492 return ret;
kono
parents:
diff changeset
493 }
kono
parents:
diff changeset
494
kono
parents:
diff changeset
495 /* Union the heap with HEAPB. One of the heaps is going to be deleted. */
kono
parents:
diff changeset
496
kono
parents:
diff changeset
497 template<class K, class V>
kono
parents:
diff changeset
498 fibonacci_heap<K,V>*
kono
parents:
diff changeset
499 fibonacci_heap<K,V>::union_with (fibonacci_heap<K,V> *heapb)
kono
parents:
diff changeset
500 {
kono
parents:
diff changeset
501 fibonacci_heap<K,V> *heapa = this;
kono
parents:
diff changeset
502
kono
parents:
diff changeset
503 fibonacci_node<K,V> *a_root, *b_root;
kono
parents:
diff changeset
504
145
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
505 /* Both heaps must share allocator. */
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
506 gcc_checking_assert (m_allocator == heapb->m_allocator);
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
507
111
kono
parents:
diff changeset
508 /* If one of the heaps is empty, the union is just the other heap. */
kono
parents:
diff changeset
509 if ((a_root = heapa->m_root) == NULL)
kono
parents:
diff changeset
510 {
kono
parents:
diff changeset
511 delete (heapa);
kono
parents:
diff changeset
512 return heapb;
kono
parents:
diff changeset
513 }
kono
parents:
diff changeset
514 if ((b_root = heapb->m_root) == NULL)
kono
parents:
diff changeset
515 {
kono
parents:
diff changeset
516 delete (heapb);
kono
parents:
diff changeset
517 return heapa;
kono
parents:
diff changeset
518 }
kono
parents:
diff changeset
519
kono
parents:
diff changeset
520 /* Merge them to the next nodes on the opposite chain. */
kono
parents:
diff changeset
521 a_root->m_left->m_right = b_root;
kono
parents:
diff changeset
522 b_root->m_left->m_right = a_root;
kono
parents:
diff changeset
523 std::swap (a_root->m_left, b_root->m_left);
kono
parents:
diff changeset
524 heapa->m_nodes += heapb->m_nodes;
kono
parents:
diff changeset
525
kono
parents:
diff changeset
526 /* And set the new minimum, if it's changed. */
kono
parents:
diff changeset
527 if (heapb->m_min->compare (heapa->m_min) < 0)
kono
parents:
diff changeset
528 heapa->m_min = heapb->m_min;
kono
parents:
diff changeset
529
kono
parents:
diff changeset
530 /* Set m_min to NULL to not to delete live fibonacci nodes. */
kono
parents:
diff changeset
531 heapb->m_min = NULL;
kono
parents:
diff changeset
532 delete (heapb);
kono
parents:
diff changeset
533
kono
parents:
diff changeset
534 return heapa;
kono
parents:
diff changeset
535 }
kono
parents:
diff changeset
536
kono
parents:
diff changeset
537 /* Insert it into the root list. */
kono
parents:
diff changeset
538
kono
parents:
diff changeset
539 template<class K, class V>
kono
parents:
diff changeset
540 void
kono
parents:
diff changeset
541 fibonacci_heap<K,V>::insert_root (fibonacci_node_t *node)
kono
parents:
diff changeset
542 {
kono
parents:
diff changeset
543 /* If the heap is currently empty, the new node becomes the singleton
kono
parents:
diff changeset
544 circular root list. */
kono
parents:
diff changeset
545 if (m_root == NULL)
kono
parents:
diff changeset
546 {
kono
parents:
diff changeset
547 m_root = node;
kono
parents:
diff changeset
548 node->m_left = node;
kono
parents:
diff changeset
549 node->m_right = node;
kono
parents:
diff changeset
550 return;
kono
parents:
diff changeset
551 }
kono
parents:
diff changeset
552
kono
parents:
diff changeset
553 /* Otherwise, insert it in the circular root list between the root
kono
parents:
diff changeset
554 and it's right node. */
kono
parents:
diff changeset
555 m_root->insert_after (node);
kono
parents:
diff changeset
556 }
kono
parents:
diff changeset
557
kono
parents:
diff changeset
558 /* Remove NODE from PARENT's child list. */
kono
parents:
diff changeset
559
kono
parents:
diff changeset
560 template<class K, class V>
kono
parents:
diff changeset
561 void
kono
parents:
diff changeset
562 fibonacci_heap<K,V>::cut (fibonacci_node<K,V> *node,
kono
parents:
diff changeset
563 fibonacci_node<K,V> *parent)
kono
parents:
diff changeset
564 {
kono
parents:
diff changeset
565 node->remove ();
kono
parents:
diff changeset
566 parent->m_degree--;
kono
parents:
diff changeset
567 insert_root (node);
kono
parents:
diff changeset
568 node->m_parent = NULL;
kono
parents:
diff changeset
569 node->m_mark = 0;
kono
parents:
diff changeset
570 }
kono
parents:
diff changeset
571
kono
parents:
diff changeset
572 /* Process cut of node Y and do it recursivelly. */
kono
parents:
diff changeset
573
kono
parents:
diff changeset
574 template<class K, class V>
kono
parents:
diff changeset
575 void
kono
parents:
diff changeset
576 fibonacci_heap<K,V>::cascading_cut (fibonacci_node<K,V> *y)
kono
parents:
diff changeset
577 {
kono
parents:
diff changeset
578 fibonacci_node<K,V> *z;
kono
parents:
diff changeset
579
kono
parents:
diff changeset
580 while ((z = y->m_parent) != NULL)
kono
parents:
diff changeset
581 {
kono
parents:
diff changeset
582 if (y->m_mark == 0)
kono
parents:
diff changeset
583 {
kono
parents:
diff changeset
584 y->m_mark = 1;
kono
parents:
diff changeset
585 return;
kono
parents:
diff changeset
586 }
kono
parents:
diff changeset
587 else
kono
parents:
diff changeset
588 {
kono
parents:
diff changeset
589 cut (y, z);
kono
parents:
diff changeset
590 y = z;
kono
parents:
diff changeset
591 }
kono
parents:
diff changeset
592 }
kono
parents:
diff changeset
593 }
kono
parents:
diff changeset
594
kono
parents:
diff changeset
595 /* Extract minimum node from the heap. */
kono
parents:
diff changeset
596
kono
parents:
diff changeset
597 template<class K, class V>
kono
parents:
diff changeset
598 fibonacci_node<K,V>*
kono
parents:
diff changeset
599 fibonacci_heap<K,V>::extract_minimum_node ()
kono
parents:
diff changeset
600 {
kono
parents:
diff changeset
601 fibonacci_node<K,V> *ret = m_min;
kono
parents:
diff changeset
602 fibonacci_node<K,V> *x, *y, *orig;
kono
parents:
diff changeset
603
kono
parents:
diff changeset
604 /* Attach the child list of the minimum node to the root list of the heap.
kono
parents:
diff changeset
605 If there is no child list, we don't do squat. */
kono
parents:
diff changeset
606 for (x = ret->m_child, orig = NULL; x != orig && x != NULL; x = y)
kono
parents:
diff changeset
607 {
kono
parents:
diff changeset
608 if (orig == NULL)
kono
parents:
diff changeset
609 orig = x;
kono
parents:
diff changeset
610 y = x->m_right;
kono
parents:
diff changeset
611 x->m_parent = NULL;
kono
parents:
diff changeset
612 insert_root (x);
kono
parents:
diff changeset
613 }
kono
parents:
diff changeset
614
kono
parents:
diff changeset
615 /* Remove the old root. */
kono
parents:
diff changeset
616 remove_root (ret);
kono
parents:
diff changeset
617 m_nodes--;
kono
parents:
diff changeset
618
kono
parents:
diff changeset
619 /* If we are left with no nodes, then the min is NULL. */
kono
parents:
diff changeset
620 if (m_nodes == 0)
kono
parents:
diff changeset
621 m_min = NULL;
kono
parents:
diff changeset
622 else
kono
parents:
diff changeset
623 {
kono
parents:
diff changeset
624 /* Otherwise, consolidate to find new minimum, as well as do the reorg
kono
parents:
diff changeset
625 work that needs to be done. */
kono
parents:
diff changeset
626 m_min = ret->m_right;
kono
parents:
diff changeset
627 consolidate ();
kono
parents:
diff changeset
628 }
kono
parents:
diff changeset
629
kono
parents:
diff changeset
630 return ret;
kono
parents:
diff changeset
631 }
kono
parents:
diff changeset
632
kono
parents:
diff changeset
633 /* Remove root NODE from the heap. */
kono
parents:
diff changeset
634
kono
parents:
diff changeset
635 template<class K, class V>
kono
parents:
diff changeset
636 void
kono
parents:
diff changeset
637 fibonacci_heap<K,V>::remove_root (fibonacci_node<K,V> *node)
kono
parents:
diff changeset
638 {
kono
parents:
diff changeset
639 if (node->m_left == node)
kono
parents:
diff changeset
640 m_root = NULL;
kono
parents:
diff changeset
641 else
kono
parents:
diff changeset
642 m_root = node->remove ();
kono
parents:
diff changeset
643 }
kono
parents:
diff changeset
644
kono
parents:
diff changeset
645 /* Consolidate heap. */
kono
parents:
diff changeset
646
kono
parents:
diff changeset
647 template<class K, class V>
kono
parents:
diff changeset
648 void fibonacci_heap<K,V>::consolidate ()
kono
parents:
diff changeset
649 {
145
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
650 const int D = 1 + 8 * sizeof (long);
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
651 fibonacci_node<K,V> *a[D];
111
kono
parents:
diff changeset
652 fibonacci_node<K,V> *w, *x, *y;
kono
parents:
diff changeset
653 int i, d;
kono
parents:
diff changeset
654
145
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
655 memset (a, 0, sizeof (a));
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
656
111
kono
parents:
diff changeset
657 while ((w = m_root) != NULL)
kono
parents:
diff changeset
658 {
kono
parents:
diff changeset
659 x = w;
kono
parents:
diff changeset
660 remove_root (w);
kono
parents:
diff changeset
661 d = x->m_degree;
145
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
662 gcc_checking_assert (d < D);
111
kono
parents:
diff changeset
663 while (a[d] != NULL)
kono
parents:
diff changeset
664 {
kono
parents:
diff changeset
665 y = a[d];
kono
parents:
diff changeset
666 if (x->compare (y) > 0)
kono
parents:
diff changeset
667 std::swap (x, y);
kono
parents:
diff changeset
668 y->link (x);
kono
parents:
diff changeset
669 a[d] = NULL;
kono
parents:
diff changeset
670 d++;
kono
parents:
diff changeset
671 }
kono
parents:
diff changeset
672 a[d] = x;
kono
parents:
diff changeset
673 }
kono
parents:
diff changeset
674 m_min = NULL;
kono
parents:
diff changeset
675 for (i = 0; i < D; i++)
kono
parents:
diff changeset
676 if (a[i] != NULL)
kono
parents:
diff changeset
677 {
kono
parents:
diff changeset
678 insert_root (a[i]);
kono
parents:
diff changeset
679 if (m_min == NULL || a[i]->compare (m_min) < 0)
kono
parents:
diff changeset
680 m_min = a[i];
kono
parents:
diff changeset
681 }
kono
parents:
diff changeset
682 }
kono
parents:
diff changeset
683
kono
parents:
diff changeset
684 #endif // GCC_FIBONACCI_HEAP_H