1 | 1 |
/* -*- mode: C++; indent-tabs-mode: nil; -*- |
2 | 2 |
* |
3 | 3 |
* This file is a part of LEMON, a generic C++ optimization library. |
4 | 4 |
* |
5 | 5 |
* Copyright (C) 2003-2009 |
6 | 6 |
* Egervary Jeno Kombinatorikus Optimalizalasi Kutatocsoport |
7 | 7 |
* (Egervary Research Group on Combinatorial Optimization, EGRES). |
8 | 8 |
* |
9 | 9 |
* Permission to use, modify and distribute this software is granted |
10 | 10 |
* provided that this copyright notice appears in all copies. For |
11 | 11 |
* precise terms see the accompanying LICENSE file. |
12 | 12 |
* |
13 | 13 |
* This software is provided "AS IS" with no warranty of any kind, |
14 | 14 |
* express or implied, and with no claim as to its suitability for any |
15 | 15 |
* purpose. |
16 | 16 |
* |
17 | 17 |
*/ |
18 | 18 |
|
19 | 19 |
#ifndef LEMON_LIST_GRAPH_H |
20 | 20 |
#define LEMON_LIST_GRAPH_H |
21 | 21 |
|
22 | 22 |
///\ingroup graphs |
23 | 23 |
///\file |
24 | 24 |
///\brief ListDigraph and ListGraph classes. |
25 | 25 |
|
26 | 26 |
#include <lemon/core.h> |
27 | 27 |
#include <lemon/error.h> |
28 | 28 |
#include <lemon/bits/graph_extender.h> |
29 | 29 |
|
30 | 30 |
#include <vector> |
31 | 31 |
#include <list> |
32 | 32 |
|
33 | 33 |
namespace lemon { |
34 | 34 |
|
35 | 35 |
class ListDigraph; |
36 | 36 |
|
37 | 37 |
class ListDigraphBase { |
38 | 38 |
|
39 | 39 |
protected: |
40 | 40 |
struct NodeT { |
41 | 41 |
int first_in, first_out; |
42 | 42 |
int prev, next; |
43 | 43 |
}; |
44 | 44 |
|
45 | 45 |
struct ArcT { |
46 | 46 |
int target, source; |
47 | 47 |
int prev_in, prev_out; |
48 | 48 |
int next_in, next_out; |
49 | 49 |
}; |
50 | 50 |
|
51 | 51 |
std::vector<NodeT> nodes; |
52 | 52 |
|
53 | 53 |
int first_node; |
54 | 54 |
|
55 | 55 |
int first_free_node; |
56 | 56 |
|
57 | 57 |
std::vector<ArcT> arcs; |
58 | 58 |
|
59 | 59 |
int first_free_arc; |
60 | 60 |
|
61 | 61 |
public: |
62 | 62 |
|
63 | 63 |
typedef ListDigraphBase Digraph; |
64 | 64 |
|
65 | 65 |
class Node { |
66 | 66 |
friend class ListDigraphBase; |
67 | 67 |
friend class ListDigraph; |
68 | 68 |
protected: |
69 | 69 |
|
70 | 70 |
int id; |
71 | 71 |
explicit Node(int pid) { id = pid;} |
72 | 72 |
|
73 | 73 |
public: |
74 | 74 |
Node() {} |
75 | 75 |
Node (Invalid) { id = -1; } |
76 | 76 |
bool operator==(const Node& node) const {return id == node.id;} |
77 | 77 |
bool operator!=(const Node& node) const {return id != node.id;} |
78 | 78 |
bool operator<(const Node& node) const {return id < node.id;} |
79 | 79 |
}; |
80 | 80 |
|
81 | 81 |
class Arc { |
82 | 82 |
friend class ListDigraphBase; |
83 | 83 |
friend class ListDigraph; |
84 | 84 |
protected: |
85 | 85 |
|
86 | 86 |
int id; |
87 | 87 |
explicit Arc(int pid) { id = pid;} |
88 | 88 |
|
89 | 89 |
public: |
90 | 90 |
Arc() {} |
91 | 91 |
Arc (Invalid) { id = -1; } |
92 | 92 |
bool operator==(const Arc& arc) const {return id == arc.id;} |
93 | 93 |
bool operator!=(const Arc& arc) const {return id != arc.id;} |
94 | 94 |
bool operator<(const Arc& arc) const {return id < arc.id;} |
95 | 95 |
}; |
96 | 96 |
|
97 | 97 |
|
98 | 98 |
|
99 | 99 |
ListDigraphBase() |
100 | 100 |
: nodes(), first_node(-1), |
101 | 101 |
first_free_node(-1), arcs(), first_free_arc(-1) {} |
102 | 102 |
|
103 | 103 |
|
104 | 104 |
int maxNodeId() const { return nodes.size()-1; } |
105 | 105 |
int maxArcId() const { return arcs.size()-1; } |
106 | 106 |
|
107 | 107 |
Node source(Arc e) const { return Node(arcs[e.id].source); } |
108 | 108 |
Node target(Arc e) const { return Node(arcs[e.id].target); } |
109 | 109 |
|
110 | 110 |
|
111 | 111 |
void first(Node& node) const { |
112 | 112 |
node.id = first_node; |
113 | 113 |
} |
114 | 114 |
|
115 | 115 |
void next(Node& node) const { |
116 | 116 |
node.id = nodes[node.id].next; |
117 | 117 |
} |
118 | 118 |
|
119 | 119 |
|
120 | 120 |
void first(Arc& arc) const { |
121 | 121 |
int n; |
122 | 122 |
for(n = first_node; |
123 |
n!=-1 && nodes[n]. |
|
123 |
n != -1 && nodes[n].first_out == -1; |
|
124 | 124 |
n = nodes[n].next) {} |
125 |
arc.id = (n == -1) ? -1 : nodes[n]. |
|
125 |
arc.id = (n == -1) ? -1 : nodes[n].first_out; |
|
126 | 126 |
} |
127 | 127 |
|
128 | 128 |
void next(Arc& arc) const { |
129 |
if (arcs[arc.id].next_in != -1) { |
|
130 |
arc.id = arcs[arc.id].next_in; |
|
129 |
if (arcs[arc.id].next_out != -1) { |
|
130 |
arc.id = arcs[arc.id].next_out; |
|
131 | 131 |
} else { |
132 | 132 |
int n; |
133 |
for(n = nodes[arcs[arc.id].target].next; |
|
134 |
n!=-1 && nodes[n].first_in == -1; |
|
133 |
for(n = nodes[arcs[arc.id].source].next; |
|
134 |
n != -1 && nodes[n].first_out == -1; |
|
135 | 135 |
n = nodes[n].next) {} |
136 |
arc.id = (n == -1) ? -1 : nodes[n]. |
|
136 |
arc.id = (n == -1) ? -1 : nodes[n].first_out; |
|
137 | 137 |
} |
138 | 138 |
} |
139 | 139 |
|
140 | 140 |
void firstOut(Arc &e, const Node& v) const { |
141 | 141 |
e.id = nodes[v.id].first_out; |
142 | 142 |
} |
143 | 143 |
void nextOut(Arc &e) const { |
144 | 144 |
e.id=arcs[e.id].next_out; |
145 | 145 |
} |
146 | 146 |
|
147 | 147 |
void firstIn(Arc &e, const Node& v) const { |
148 | 148 |
e.id = nodes[v.id].first_in; |
149 | 149 |
} |
150 | 150 |
void nextIn(Arc &e) const { |
151 | 151 |
e.id=arcs[e.id].next_in; |
152 | 152 |
} |
153 | 153 |
|
154 | 154 |
|
155 | 155 |
static int id(Node v) { return v.id; } |
156 | 156 |
static int id(Arc e) { return e.id; } |
157 | 157 |
|
158 | 158 |
static Node nodeFromId(int id) { return Node(id);} |
159 | 159 |
static Arc arcFromId(int id) { return Arc(id);} |
160 | 160 |
|
161 | 161 |
bool valid(Node n) const { |
162 | 162 |
return n.id >= 0 && n.id < static_cast<int>(nodes.size()) && |
163 | 163 |
nodes[n.id].prev != -2; |
164 | 164 |
} |
165 | 165 |
|
166 | 166 |
bool valid(Arc a) const { |
167 | 167 |
return a.id >= 0 && a.id < static_cast<int>(arcs.size()) && |
168 | 168 |
arcs[a.id].prev_in != -2; |
169 | 169 |
} |
170 | 170 |
|
171 | 171 |
Node addNode() { |
172 | 172 |
int n; |
173 | 173 |
|
174 | 174 |
if(first_free_node==-1) { |
175 | 175 |
n = nodes.size(); |
176 | 176 |
nodes.push_back(NodeT()); |
177 | 177 |
} else { |
178 | 178 |
n = first_free_node; |
179 | 179 |
first_free_node = nodes[n].next; |
180 | 180 |
} |
181 | 181 |
|
182 | 182 |
nodes[n].next = first_node; |
183 | 183 |
if(first_node != -1) nodes[first_node].prev = n; |
184 | 184 |
first_node = n; |
185 | 185 |
nodes[n].prev = -1; |
186 | 186 |
|
187 | 187 |
nodes[n].first_in = nodes[n].first_out = -1; |
188 | 188 |
|
189 | 189 |
return Node(n); |
190 | 190 |
} |
191 | 191 |
|
192 | 192 |
Arc addArc(Node u, Node v) { |
193 | 193 |
int n; |
194 | 194 |
|
195 | 195 |
if (first_free_arc == -1) { |
196 | 196 |
n = arcs.size(); |
197 | 197 |
arcs.push_back(ArcT()); |
198 | 198 |
} else { |
199 | 199 |
n = first_free_arc; |
200 | 200 |
first_free_arc = arcs[n].next_in; |
201 | 201 |
} |
202 | 202 |
|
203 | 203 |
arcs[n].source = u.id; |
204 | 204 |
arcs[n].target = v.id; |
205 | 205 |
|
206 | 206 |
arcs[n].next_out = nodes[u.id].first_out; |
207 | 207 |
if(nodes[u.id].first_out != -1) { |
208 | 208 |
arcs[nodes[u.id].first_out].prev_out = n; |
209 | 209 |
} |
210 | 210 |
|
211 | 211 |
arcs[n].next_in = nodes[v.id].first_in; |
212 | 212 |
if(nodes[v.id].first_in != -1) { |
213 | 213 |
arcs[nodes[v.id].first_in].prev_in = n; |
214 | 214 |
} |
215 | 215 |
|
216 | 216 |
arcs[n].prev_in = arcs[n].prev_out = -1; |
217 | 217 |
|
218 | 218 |
nodes[u.id].first_out = nodes[v.id].first_in = n; |
219 | 219 |
|
220 | 220 |
return Arc(n); |
221 | 221 |
} |
222 | 222 |
|
223 | 223 |
void erase(const Node& node) { |
224 | 224 |
int n = node.id; |
225 | 225 |
|
226 | 226 |
if(nodes[n].next != -1) { |
227 | 227 |
nodes[nodes[n].next].prev = nodes[n].prev; |
228 | 228 |
} |
229 | 229 |
|
230 | 230 |
if(nodes[n].prev != -1) { |
231 | 231 |
nodes[nodes[n].prev].next = nodes[n].next; |
232 | 232 |
} else { |
233 | 233 |
first_node = nodes[n].next; |
234 | 234 |
} |
235 | 235 |
|
236 | 236 |
nodes[n].next = first_free_node; |
237 | 237 |
first_free_node = n; |
238 | 238 |
nodes[n].prev = -2; |
239 | 239 |
|
240 | 240 |
} |
241 | 241 |
|
242 | 242 |
void erase(const Arc& arc) { |
243 | 243 |
int n = arc.id; |
244 | 244 |
|
245 | 245 |
if(arcs[n].next_in!=-1) { |
246 | 246 |
arcs[arcs[n].next_in].prev_in = arcs[n].prev_in; |
247 | 247 |
} |
248 | 248 |
|
249 | 249 |
if(arcs[n].prev_in!=-1) { |
250 | 250 |
arcs[arcs[n].prev_in].next_in = arcs[n].next_in; |
251 | 251 |
} else { |
252 | 252 |
nodes[arcs[n].target].first_in = arcs[n].next_in; |
253 | 253 |
} |
254 | 254 |
|
255 | 255 |
|
256 | 256 |
if(arcs[n].next_out!=-1) { |
257 | 257 |
arcs[arcs[n].next_out].prev_out = arcs[n].prev_out; |
258 | 258 |
} |
259 | 259 |
|
260 | 260 |
if(arcs[n].prev_out!=-1) { |
261 | 261 |
arcs[arcs[n].prev_out].next_out = arcs[n].next_out; |
262 | 262 |
} else { |
263 | 263 |
nodes[arcs[n].source].first_out = arcs[n].next_out; |
264 | 264 |
} |
265 | 265 |
|
266 | 266 |
arcs[n].next_in = first_free_arc; |
267 | 267 |
first_free_arc = n; |
268 | 268 |
arcs[n].prev_in = -2; |
269 | 269 |
} |
270 | 270 |
|
271 | 271 |
void clear() { |
272 | 272 |
arcs.clear(); |
273 | 273 |
nodes.clear(); |
274 | 274 |
first_node = first_free_node = first_free_arc = -1; |
275 | 275 |
} |
276 | 276 |
|
277 | 277 |
protected: |
278 | 278 |
void changeTarget(Arc e, Node n) |
279 | 279 |
{ |
280 | 280 |
if(arcs[e.id].next_in != -1) |
281 | 281 |
arcs[arcs[e.id].next_in].prev_in = arcs[e.id].prev_in; |
282 | 282 |
if(arcs[e.id].prev_in != -1) |
283 | 283 |
arcs[arcs[e.id].prev_in].next_in = arcs[e.id].next_in; |
284 | 284 |
else nodes[arcs[e.id].target].first_in = arcs[e.id].next_in; |
285 | 285 |
if (nodes[n.id].first_in != -1) { |
286 | 286 |
arcs[nodes[n.id].first_in].prev_in = e.id; |
287 | 287 |
} |
288 | 288 |
arcs[e.id].target = n.id; |
289 | 289 |
arcs[e.id].prev_in = -1; |
290 | 290 |
arcs[e.id].next_in = nodes[n.id].first_in; |
291 | 291 |
nodes[n.id].first_in = e.id; |
292 | 292 |
} |
293 | 293 |
void changeSource(Arc e, Node n) |
294 | 294 |
{ |
295 | 295 |
if(arcs[e.id].next_out != -1) |
296 | 296 |
arcs[arcs[e.id].next_out].prev_out = arcs[e.id].prev_out; |
297 | 297 |
if(arcs[e.id].prev_out != -1) |
298 | 298 |
arcs[arcs[e.id].prev_out].next_out = arcs[e.id].next_out; |
299 | 299 |
else nodes[arcs[e.id].source].first_out = arcs[e.id].next_out; |
300 | 300 |
if (nodes[n.id].first_out != -1) { |
301 | 301 |
arcs[nodes[n.id].first_out].prev_out = e.id; |
302 | 302 |
} |
303 | 303 |
arcs[e.id].source = n.id; |
304 | 304 |
arcs[e.id].prev_out = -1; |
305 | 305 |
arcs[e.id].next_out = nodes[n.id].first_out; |
306 | 306 |
nodes[n.id].first_out = e.id; |
307 | 307 |
} |
308 | 308 |
|
309 | 309 |
}; |
310 | 310 |
|
311 | 311 |
typedef DigraphExtender<ListDigraphBase> ExtendedListDigraphBase; |
312 | 312 |
|
313 | 313 |
/// \addtogroup graphs |
314 | 314 |
/// @{ |
315 | 315 |
|
316 | 316 |
///A general directed graph structure. |
317 | 317 |
|
318 | 318 |
///\ref ListDigraph is a versatile and fast directed graph |
319 | 319 |
///implementation based on linked lists that are stored in |
320 | 320 |
///\c std::vector structures. |
321 | 321 |
/// |
322 | 322 |
///This type fully conforms to the \ref concepts::Digraph "Digraph concept" |
323 | 323 |
///and it also provides several useful additional functionalities. |
324 | 324 |
///Most of its member functions and nested classes are documented |
325 | 325 |
///only in the concept class. |
326 | 326 |
/// |
327 | 327 |
///\sa concepts::Digraph |
328 | 328 |
///\sa ListGraph |
329 | 329 |
class ListDigraph : public ExtendedListDigraphBase { |
330 | 330 |
typedef ExtendedListDigraphBase Parent; |
331 | 331 |
|
332 | 332 |
private: |
333 | 333 |
/// Digraphs are \e not copy constructible. Use DigraphCopy instead. |
334 | 334 |
ListDigraph(const ListDigraph &) :ExtendedListDigraphBase() {}; |
335 | 335 |
/// \brief Assignment of a digraph to another one is \e not allowed. |
336 | 336 |
/// Use DigraphCopy instead. |
337 | 337 |
void operator=(const ListDigraph &) {} |
338 | 338 |
public: |
339 | 339 |
|
340 | 340 |
/// Constructor |
341 | 341 |
|
342 | 342 |
/// Constructor. |
343 | 343 |
/// |
344 | 344 |
ListDigraph() {} |
345 | 345 |
|
346 | 346 |
///Add a new node to the digraph. |
347 | 347 |
|
348 | 348 |
///This function adds a new node to the digraph. |
349 | 349 |
///\return The new node. |
350 | 350 |
Node addNode() { return Parent::addNode(); } |
351 | 351 |
|
352 | 352 |
///Add a new arc to the digraph. |
353 | 353 |
|
354 | 354 |
///This function adds a new arc to the digraph with source node \c s |
355 | 355 |
///and target node \c t. |
356 | 356 |
///\return The new arc. |
357 | 357 |
Arc addArc(Node s, Node t) { |
358 | 358 |
return Parent::addArc(s, t); |
359 | 359 |
} |
360 | 360 |
|
361 | 361 |
///\brief Erase a node from the digraph. |
362 | 362 |
/// |
363 | 363 |
///This function erases the given node from the digraph. |
364 | 364 |
void erase(Node n) { Parent::erase(n); } |
365 | 365 |
|
366 | 366 |
///\brief Erase an arc from the digraph. |
367 | 367 |
/// |
368 | 368 |
///This function erases the given arc from the digraph. |
369 | 369 |
void erase(Arc a) { Parent::erase(a); } |
370 | 370 |
|
371 | 371 |
/// Node validity check |
372 | 372 |
|
373 | 373 |
/// This function gives back \c true if the given node is valid, |
374 | 374 |
/// i.e. it is a real node of the digraph. |
375 | 375 |
/// |
376 | 376 |
/// \warning A removed node could become valid again if new nodes are |
377 | 377 |
/// added to the digraph. |
378 | 378 |
bool valid(Node n) const { return Parent::valid(n); } |
379 | 379 |
|
380 | 380 |
/// Arc validity check |
381 | 381 |
|
382 | 382 |
/// This function gives back \c true if the given arc is valid, |
383 | 383 |
/// i.e. it is a real arc of the digraph. |
384 | 384 |
/// |
385 | 385 |
/// \warning A removed arc could become valid again if new arcs are |
386 | 386 |
/// added to the digraph. |
387 | 387 |
bool valid(Arc a) const { return Parent::valid(a); } |
388 | 388 |
|
389 | 389 |
/// Change the target node of an arc |
390 | 390 |
|
391 | 391 |
/// This function changes the target node of the given arc \c a to \c n. |
392 | 392 |
/// |
393 | 393 |
///\note \c ArcIt and \c OutArcIt iterators referencing the changed |
394 | 394 |
///arc remain valid, however \c InArcIt iterators are invalidated. |
395 | 395 |
/// |
396 | 396 |
///\warning This functionality cannot be used together with the Snapshot |
397 | 397 |
///feature. |
398 | 398 |
void changeTarget(Arc a, Node n) { |
399 | 399 |
Parent::changeTarget(a,n); |
400 | 400 |
} |
401 | 401 |
/// Change the source node of an arc |
402 | 402 |
|
403 | 403 |
/// This function changes the source node of the given arc \c a to \c n. |
404 | 404 |
/// |
405 | 405 |
///\note \c InArcIt iterators referencing the changed arc remain |
406 | 406 |
///valid, however \c ArcIt and \c OutArcIt iterators are invalidated. |
407 | 407 |
/// |
408 | 408 |
///\warning This functionality cannot be used together with the Snapshot |
409 | 409 |
///feature. |
410 | 410 |
void changeSource(Arc a, Node n) { |
411 | 411 |
Parent::changeSource(a,n); |
412 | 412 |
} |
413 | 413 |
|
414 | 414 |
/// Reverse the direction of an arc. |
415 | 415 |
|
416 | 416 |
/// This function reverses the direction of the given arc. |
417 | 417 |
///\note \c ArcIt, \c OutArcIt and \c InArcIt iterators referencing |
418 | 418 |
///the changed arc are invalidated. |
419 | 419 |
/// |
420 | 420 |
///\warning This functionality cannot be used together with the Snapshot |
421 | 421 |
///feature. |
422 | 422 |
void reverseArc(Arc a) { |
423 | 423 |
Node t=target(a); |
424 | 424 |
changeTarget(a,source(a)); |
425 | 425 |
changeSource(a,t); |
426 | 426 |
} |
427 | 427 |
|
428 | 428 |
///Contract two nodes. |
429 | 429 |
|
430 | 430 |
///This function contracts the given two nodes. |
431 | 431 |
///Node \c v is removed, but instead of deleting its |
432 | 432 |
///incident arcs, they are joined to node \c u. |
433 | 433 |
///If the last parameter \c r is \c true (this is the default value), |
434 | 434 |
///then the newly created loops are removed. |
435 | 435 |
/// |
436 | 436 |
///\note The moved arcs are joined to node \c u using changeSource() |
437 | 437 |
///or changeTarget(), thus \c ArcIt and \c OutArcIt iterators are |
438 | 438 |
///invalidated for the outgoing arcs of node \c v and \c InArcIt |
439 | 439 |
///iterators are invalidated for the incomming arcs of \c v. |
440 | 440 |
///Moreover all iterators referencing node \c v or the removed |
441 | 441 |
///loops are also invalidated. Other iterators remain valid. |
442 | 442 |
/// |
443 | 443 |
///\warning This functionality cannot be used together with the Snapshot |
444 | 444 |
///feature. |
445 | 445 |
void contract(Node u, Node v, bool r = true) |
446 | 446 |
{ |
447 | 447 |
for(OutArcIt e(*this,v);e!=INVALID;) { |
448 | 448 |
OutArcIt f=e; |
449 | 449 |
++f; |
450 | 450 |
if(r && target(e)==u) erase(e); |
451 | 451 |
else changeSource(e,u); |
452 | 452 |
e=f; |
453 | 453 |
} |
454 | 454 |
for(InArcIt e(*this,v);e!=INVALID;) { |
455 | 455 |
InArcIt f=e; |
456 | 456 |
++f; |
457 | 457 |
if(r && source(e)==u) erase(e); |
458 | 458 |
else changeTarget(e,u); |
459 | 459 |
e=f; |
460 | 460 |
} |
461 | 461 |
erase(v); |
462 | 462 |
} |
463 | 463 |
|
464 | 464 |
///Split a node. |
465 | 465 |
|
466 | 466 |
///This function splits the given node. First, a new node is added |
467 | 467 |
///to the digraph, then the source of each outgoing arc of node \c n |
468 | 468 |
///is moved to this new node. |
469 | 469 |
///If the second parameter \c connect is \c true (this is the default |
470 | 470 |
///value), then a new arc from node \c n to the newly created node |
471 | 471 |
///is also added. |
472 | 472 |
///\return The newly created node. |
473 | 473 |
/// |
474 | 474 |
///\note All iterators remain valid. |
475 | 475 |
/// |
476 | 476 |
///\warning This functionality cannot be used together with the |
477 | 477 |
///Snapshot feature. |
478 | 478 |
Node split(Node n, bool connect = true) { |
479 | 479 |
Node b = addNode(); |
480 | 480 |
nodes[b.id].first_out=nodes[n.id].first_out; |
481 | 481 |
nodes[n.id].first_out=-1; |
482 | 482 |
for(int i=nodes[b.id].first_out; i!=-1; i=arcs[i].next_out) { |
483 | 483 |
arcs[i].source=b.id; |
484 | 484 |
} |
485 | 485 |
if (connect) addArc(n,b); |
486 | 486 |
return b; |
487 | 487 |
} |
488 | 488 |
|
489 | 489 |
///Split an arc. |
490 | 490 |
|
491 | 491 |
///This function splits the given arc. First, a new node \c v is |
492 | 492 |
///added to the digraph, then the target node of the original arc |
493 | 493 |
///is set to \c v. Finally, an arc from \c v to the original target |
494 | 494 |
///is added. |
495 | 495 |
///\return The newly created node. |
496 | 496 |
/// |
497 | 497 |
///\note \c InArcIt iterators referencing the original arc are |
498 | 498 |
///invalidated. Other iterators remain valid. |
499 | 499 |
/// |
500 | 500 |
///\warning This functionality cannot be used together with the |
501 | 501 |
///Snapshot feature. |
502 | 502 |
Node split(Arc a) { |
503 | 503 |
Node v = addNode(); |
504 | 504 |
addArc(v,target(a)); |
505 | 505 |
changeTarget(a,v); |
506 | 506 |
return v; |
507 | 507 |
} |
508 | 508 |
|
509 | 509 |
///Clear the digraph. |
510 | 510 |
|
511 | 511 |
///This function erases all nodes and arcs from the digraph. |
512 | 512 |
/// |
513 | 513 |
void clear() { |
514 | 514 |
Parent::clear(); |
515 | 515 |
} |
516 | 516 |
|
517 | 517 |
/// Reserve memory for nodes. |
518 | 518 |
|
519 | 519 |
/// Using this function, it is possible to avoid superfluous memory |
520 | 520 |
/// allocation: if you know that the digraph you want to build will |
521 | 521 |
/// be large (e.g. it will contain millions of nodes and/or arcs), |
522 | 522 |
/// then it is worth reserving space for this amount before starting |
523 | 523 |
/// to build the digraph. |
524 | 524 |
/// \sa reserveArc() |
525 | 525 |
void reserveNode(int n) { nodes.reserve(n); }; |
526 | 526 |
|
527 | 527 |
/// Reserve memory for arcs. |
528 | 528 |
|
529 | 529 |
/// Using this function, it is possible to avoid superfluous memory |
530 | 530 |
/// allocation: if you know that the digraph you want to build will |
531 | 531 |
/// be large (e.g. it will contain millions of nodes and/or arcs), |
532 | 532 |
/// then it is worth reserving space for this amount before starting |
533 | 533 |
/// to build the digraph. |
534 | 534 |
/// \sa reserveNode() |
535 | 535 |
void reserveArc(int m) { arcs.reserve(m); }; |
536 | 536 |
|
537 | 537 |
/// \brief Class to make a snapshot of the digraph and restore |
538 | 538 |
/// it later. |
539 | 539 |
/// |
540 | 540 |
/// Class to make a snapshot of the digraph and restore it later. |
541 | 541 |
/// |
542 | 542 |
/// The newly added nodes and arcs can be removed using the |
543 | 543 |
/// restore() function. |
544 | 544 |
/// |
545 | 545 |
/// \note After a state is restored, you cannot restore a later state, |
546 | 546 |
/// i.e. you cannot add the removed nodes and arcs again using |
547 | 547 |
/// another Snapshot instance. |
548 | 548 |
/// |
549 | 549 |
/// \warning Node and arc deletions and other modifications (e.g. |
550 | 550 |
/// reversing, contracting, splitting arcs or nodes) cannot be |
551 | 551 |
/// restored. These events invalidate the snapshot. |
552 | 552 |
/// However the arcs and nodes that were added to the digraph after |
553 | 553 |
/// making the current snapshot can be removed without invalidating it. |
554 | 554 |
class Snapshot { |
555 | 555 |
protected: |
556 | 556 |
|
557 | 557 |
typedef Parent::NodeNotifier NodeNotifier; |
558 | 558 |
|
559 | 559 |
class NodeObserverProxy : public NodeNotifier::ObserverBase { |
560 | 560 |
public: |
561 | 561 |
|
562 | 562 |
NodeObserverProxy(Snapshot& _snapshot) |
563 | 563 |
: snapshot(_snapshot) {} |
564 | 564 |
|
565 | 565 |
using NodeNotifier::ObserverBase::attach; |
566 | 566 |
using NodeNotifier::ObserverBase::detach; |
567 | 567 |
using NodeNotifier::ObserverBase::attached; |
568 | 568 |
|
569 | 569 |
protected: |
570 | 570 |
|
571 | 571 |
virtual void add(const Node& node) { |
572 | 572 |
snapshot.addNode(node); |
573 | 573 |
} |
574 | 574 |
virtual void add(const std::vector<Node>& nodes) { |
575 | 575 |
for (int i = nodes.size() - 1; i >= 0; ++i) { |
576 | 576 |
snapshot.addNode(nodes[i]); |
577 | 577 |
} |
578 | 578 |
} |
579 | 579 |
virtual void erase(const Node& node) { |
580 | 580 |
snapshot.eraseNode(node); |
581 | 581 |
} |
582 | 582 |
virtual void erase(const std::vector<Node>& nodes) { |
583 | 583 |
for (int i = 0; i < int(nodes.size()); ++i) { |
584 | 584 |
snapshot.eraseNode(nodes[i]); |
585 | 585 |
} |
586 | 586 |
} |
587 | 587 |
virtual void build() { |
588 | 588 |
Node node; |
589 | 589 |
std::vector<Node> nodes; |
590 | 590 |
for (notifier()->first(node); node != INVALID; |
591 | 591 |
notifier()->next(node)) { |
592 | 592 |
nodes.push_back(node); |
593 | 593 |
} |
594 | 594 |
for (int i = nodes.size() - 1; i >= 0; --i) { |
595 | 595 |
snapshot.addNode(nodes[i]); |
596 | 596 |
} |
597 | 597 |
} |
598 | 598 |
virtual void clear() { |
599 | 599 |
Node node; |
600 | 600 |
for (notifier()->first(node); node != INVALID; |
601 | 601 |
notifier()->next(node)) { |
602 | 602 |
snapshot.eraseNode(node); |
603 | 603 |
} |
604 | 604 |
} |
605 | 605 |
|
606 | 606 |
Snapshot& snapshot; |
607 | 607 |
}; |
608 | 608 |
|
609 | 609 |
class ArcObserverProxy : public ArcNotifier::ObserverBase { |
610 | 610 |
public: |
611 | 611 |
|
612 | 612 |
ArcObserverProxy(Snapshot& _snapshot) |
613 | 613 |
: snapshot(_snapshot) {} |
614 | 614 |
|
615 | 615 |
using ArcNotifier::ObserverBase::attach; |
616 | 616 |
using ArcNotifier::ObserverBase::detach; |
617 | 617 |
using ArcNotifier::ObserverBase::attached; |
618 | 618 |
|
619 | 619 |
protected: |
620 | 620 |
|
621 | 621 |
virtual void add(const Arc& arc) { |
622 | 622 |
snapshot.addArc(arc); |
623 | 623 |
} |
624 | 624 |
virtual void add(const std::vector<Arc>& arcs) { |
625 | 625 |
for (int i = arcs.size() - 1; i >= 0; ++i) { |
626 | 626 |
snapshot.addArc(arcs[i]); |
627 | 627 |
} |
628 | 628 |
} |
629 | 629 |
virtual void erase(const Arc& arc) { |
630 | 630 |
snapshot.eraseArc(arc); |
631 | 631 |
} |
632 | 632 |
virtual void erase(const std::vector<Arc>& arcs) { |
633 | 633 |
for (int i = 0; i < int(arcs.size()); ++i) { |
634 | 634 |
snapshot.eraseArc(arcs[i]); |
635 | 635 |
} |
636 | 636 |
} |
637 | 637 |
virtual void build() { |
638 | 638 |
Arc arc; |
639 | 639 |
std::vector<Arc> arcs; |
640 | 640 |
for (notifier()->first(arc); arc != INVALID; |
641 | 641 |
notifier()->next(arc)) { |
642 | 642 |
arcs.push_back(arc); |
643 | 643 |
} |
644 | 644 |
for (int i = arcs.size() - 1; i >= 0; --i) { |
645 | 645 |
snapshot.addArc(arcs[i]); |
646 | 646 |
} |
647 | 647 |
} |
648 | 648 |
virtual void clear() { |
649 | 649 |
Arc arc; |
650 | 650 |
for (notifier()->first(arc); arc != INVALID; |
651 | 651 |
notifier()->next(arc)) { |
652 | 652 |
snapshot.eraseArc(arc); |
653 | 653 |
} |
654 | 654 |
} |
655 | 655 |
|
656 | 656 |
Snapshot& snapshot; |
657 | 657 |
}; |
658 | 658 |
|
659 | 659 |
ListDigraph *digraph; |
660 | 660 |
|
661 | 661 |
NodeObserverProxy node_observer_proxy; |
662 | 662 |
ArcObserverProxy arc_observer_proxy; |
663 | 663 |
|
664 | 664 |
std::list<Node> added_nodes; |
665 | 665 |
std::list<Arc> added_arcs; |
666 | 666 |
|
667 | 667 |
|
668 | 668 |
void addNode(const Node& node) { |
669 | 669 |
added_nodes.push_front(node); |
670 | 670 |
} |
671 | 671 |
void eraseNode(const Node& node) { |
672 | 672 |
std::list<Node>::iterator it = |
673 | 673 |
std::find(added_nodes.begin(), added_nodes.end(), node); |
674 | 674 |
if (it == added_nodes.end()) { |
675 | 675 |
clear(); |
676 | 676 |
arc_observer_proxy.detach(); |
677 | 677 |
throw NodeNotifier::ImmediateDetach(); |
678 | 678 |
} else { |
679 | 679 |
added_nodes.erase(it); |
680 | 680 |
} |
681 | 681 |
} |
682 | 682 |
|
683 | 683 |
void addArc(const Arc& arc) { |
684 | 684 |
added_arcs.push_front(arc); |
685 | 685 |
} |
686 | 686 |
void eraseArc(const Arc& arc) { |
687 | 687 |
std::list<Arc>::iterator it = |
688 | 688 |
std::find(added_arcs.begin(), added_arcs.end(), arc); |
689 | 689 |
if (it == added_arcs.end()) { |
690 | 690 |
clear(); |
691 | 691 |
node_observer_proxy.detach(); |
692 | 692 |
throw ArcNotifier::ImmediateDetach(); |
693 | 693 |
} else { |
694 | 694 |
added_arcs.erase(it); |
695 | 695 |
} |
696 | 696 |
} |
697 | 697 |
|
698 | 698 |
void attach(ListDigraph &_digraph) { |
699 | 699 |
digraph = &_digraph; |
700 | 700 |
node_observer_proxy.attach(digraph->notifier(Node())); |
701 | 701 |
arc_observer_proxy.attach(digraph->notifier(Arc())); |
702 | 702 |
} |
703 | 703 |
|
704 | 704 |
void detach() { |
705 | 705 |
node_observer_proxy.detach(); |
706 | 706 |
arc_observer_proxy.detach(); |
707 | 707 |
} |
708 | 708 |
|
709 | 709 |
bool attached() const { |
710 | 710 |
return node_observer_proxy.attached(); |
711 | 711 |
} |
712 | 712 |
|
713 | 713 |
void clear() { |
714 | 714 |
added_nodes.clear(); |
715 | 715 |
added_arcs.clear(); |
716 | 716 |
} |
717 | 717 |
|
718 | 718 |
public: |
719 | 719 |
|
720 | 720 |
/// \brief Default constructor. |
721 | 721 |
/// |
722 | 722 |
/// Default constructor. |
723 | 723 |
/// You have to call save() to actually make a snapshot. |
724 | 724 |
Snapshot() |
725 | 725 |
: digraph(0), node_observer_proxy(*this), |
726 | 726 |
arc_observer_proxy(*this) {} |
727 | 727 |
|
728 | 728 |
/// \brief Constructor that immediately makes a snapshot. |
729 | 729 |
/// |
730 | 730 |
/// This constructor immediately makes a snapshot of the given digraph. |
731 | 731 |
Snapshot(ListDigraph &gr) |
732 | 732 |
: node_observer_proxy(*this), |
733 | 733 |
arc_observer_proxy(*this) { |
734 | 734 |
attach(gr); |
735 | 735 |
} |
736 | 736 |
|
737 | 737 |
/// \brief Make a snapshot. |
738 | 738 |
/// |
739 | 739 |
/// This function makes a snapshot of the given digraph. |
740 | 740 |
/// It can be called more than once. In case of a repeated |
741 | 741 |
/// call, the previous snapshot gets lost. |
742 | 742 |
void save(ListDigraph &gr) { |
743 | 743 |
if (attached()) { |
744 | 744 |
detach(); |
745 | 745 |
clear(); |
746 | 746 |
} |
747 | 747 |
attach(gr); |
748 | 748 |
} |
749 | 749 |
|
750 | 750 |
/// \brief Undo the changes until the last snapshot. |
751 | 751 |
/// |
752 | 752 |
/// This function undos the changes until the last snapshot |
753 | 753 |
/// created by save() or Snapshot(ListDigraph&). |
754 | 754 |
/// |
755 | 755 |
/// \warning This method invalidates the snapshot, i.e. repeated |
756 | 756 |
/// restoring is not supported unless you call save() again. |
757 | 757 |
void restore() { |
758 | 758 |
detach(); |
759 | 759 |
for(std::list<Arc>::iterator it = added_arcs.begin(); |
760 | 760 |
it != added_arcs.end(); ++it) { |
761 | 761 |
digraph->erase(*it); |
762 | 762 |
} |
763 | 763 |
for(std::list<Node>::iterator it = added_nodes.begin(); |
764 | 764 |
it != added_nodes.end(); ++it) { |
765 | 765 |
digraph->erase(*it); |
766 | 766 |
} |
767 | 767 |
clear(); |
768 | 768 |
} |
769 | 769 |
|
770 | 770 |
/// \brief Returns \c true if the snapshot is valid. |
771 | 771 |
/// |
772 | 772 |
/// This function returns \c true if the snapshot is valid. |
773 | 773 |
bool valid() const { |
774 | 774 |
return attached(); |
775 | 775 |
} |
776 | 776 |
}; |
777 | 777 |
|
778 | 778 |
}; |
779 | 779 |
|
780 | 780 |
///@} |
781 | 781 |
|
782 | 782 |
class ListGraphBase { |
783 | 783 |
|
784 | 784 |
protected: |
785 | 785 |
|
786 | 786 |
struct NodeT { |
787 | 787 |
int first_out; |
788 | 788 |
int prev, next; |
789 | 789 |
}; |
790 | 790 |
|
791 | 791 |
struct ArcT { |
792 | 792 |
int target; |
793 | 793 |
int prev_out, next_out; |
794 | 794 |
}; |
795 | 795 |
|
796 | 796 |
std::vector<NodeT> nodes; |
797 | 797 |
|
798 | 798 |
int first_node; |
799 | 799 |
|
800 | 800 |
int first_free_node; |
801 | 801 |
|
802 | 802 |
std::vector<ArcT> arcs; |
803 | 803 |
|
804 | 804 |
int first_free_arc; |
805 | 805 |
|
806 | 806 |
public: |
807 | 807 |
|
808 | 808 |
typedef ListGraphBase Graph; |
809 | 809 |
|
810 | 810 |
class Node { |
811 | 811 |
friend class ListGraphBase; |
812 | 812 |
protected: |
813 | 813 |
|
814 | 814 |
int id; |
815 | 815 |
explicit Node(int pid) { id = pid;} |
816 | 816 |
|
817 | 817 |
public: |
818 | 818 |
Node() {} |
819 | 819 |
Node (Invalid) { id = -1; } |
820 | 820 |
bool operator==(const Node& node) const {return id == node.id;} |
821 | 821 |
bool operator!=(const Node& node) const {return id != node.id;} |
822 | 822 |
bool operator<(const Node& node) const {return id < node.id;} |
823 | 823 |
}; |
824 | 824 |
|
825 | 825 |
class Edge { |
826 | 826 |
friend class ListGraphBase; |
827 | 827 |
protected: |
828 | 828 |
|
829 | 829 |
int id; |
830 | 830 |
explicit Edge(int pid) { id = pid;} |
831 | 831 |
|
832 | 832 |
public: |
833 | 833 |
Edge() {} |
834 | 834 |
Edge (Invalid) { id = -1; } |
835 | 835 |
bool operator==(const Edge& edge) const {return id == edge.id;} |
836 | 836 |
bool operator!=(const Edge& edge) const {return id != edge.id;} |
837 | 837 |
bool operator<(const Edge& edge) const {return id < edge.id;} |
838 | 838 |
}; |
839 | 839 |
|
840 | 840 |
class Arc { |
841 | 841 |
friend class ListGraphBase; |
842 | 842 |
protected: |
843 | 843 |
|
844 | 844 |
int id; |
845 | 845 |
explicit Arc(int pid) { id = pid;} |
846 | 846 |
|
847 | 847 |
public: |
848 | 848 |
operator Edge() const { |
849 | 849 |
return id != -1 ? edgeFromId(id / 2) : INVALID; |
850 | 850 |
} |
851 | 851 |
|
852 | 852 |
Arc() {} |
853 | 853 |
Arc (Invalid) { id = -1; } |
854 | 854 |
bool operator==(const Arc& arc) const {return id == arc.id;} |
855 | 855 |
bool operator!=(const Arc& arc) const {return id != arc.id;} |
856 | 856 |
bool operator<(const Arc& arc) const {return id < arc.id;} |
857 | 857 |
}; |
858 | 858 |
|
859 | 859 |
ListGraphBase() |
860 | 860 |
: nodes(), first_node(-1), |
861 | 861 |
first_free_node(-1), arcs(), first_free_arc(-1) {} |
862 | 862 |
|
863 | 863 |
|
864 | 864 |
int maxNodeId() const { return nodes.size()-1; } |
865 | 865 |
int maxEdgeId() const { return arcs.size() / 2 - 1; } |
866 | 866 |
int maxArcId() const { return arcs.size()-1; } |
867 | 867 |
|
868 | 868 |
Node source(Arc e) const { return Node(arcs[e.id ^ 1].target); } |
869 | 869 |
Node target(Arc e) const { return Node(arcs[e.id].target); } |
870 | 870 |
|
871 | 871 |
Node u(Edge e) const { return Node(arcs[2 * e.id].target); } |
872 | 872 |
Node v(Edge e) const { return Node(arcs[2 * e.id + 1].target); } |
873 | 873 |
|
874 | 874 |
static bool direction(Arc e) { |
875 | 875 |
return (e.id & 1) == 1; |
876 | 876 |
} |
877 | 877 |
|
878 | 878 |
static Arc direct(Edge e, bool d) { |
879 | 879 |
return Arc(e.id * 2 + (d ? 1 : 0)); |
880 | 880 |
} |
881 | 881 |
|
882 | 882 |
void first(Node& node) const { |
883 | 883 |
node.id = first_node; |
884 | 884 |
} |
885 | 885 |
|
886 | 886 |
void next(Node& node) const { |
887 | 887 |
node.id = nodes[node.id].next; |
888 | 888 |
} |
889 | 889 |
|
890 | 890 |
void first(Arc& e) const { |
891 | 891 |
int n = first_node; |
892 | 892 |
while (n != -1 && nodes[n].first_out == -1) { |
893 | 893 |
n = nodes[n].next; |
894 | 894 |
} |
895 | 895 |
e.id = (n == -1) ? -1 : nodes[n].first_out; |
896 | 896 |
} |
897 | 897 |
|
898 | 898 |
void next(Arc& e) const { |
899 | 899 |
if (arcs[e.id].next_out != -1) { |
900 | 900 |
e.id = arcs[e.id].next_out; |
901 | 901 |
} else { |
902 | 902 |
int n = nodes[arcs[e.id ^ 1].target].next; |
903 | 903 |
while(n != -1 && nodes[n].first_out == -1) { |
904 | 904 |
n = nodes[n].next; |
0 comments (0 inline)