| ... |
... |
@@ -34,129 +34,129 @@
|
| 34 |
34 |
|
| 35 |
35 |
/// \ingroup min_cut
|
| 36 |
36 |
///
|
| 37 |
37 |
/// \brief Gomory-Hu cut tree algorithm
|
| 38 |
38 |
///
|
| 39 |
39 |
/// The Gomory-Hu tree is a tree on the node set of the graph, but it
|
| 40 |
40 |
/// may contain edges which are not in the original digraph. It has the
|
| 41 |
41 |
/// property that the minimum capacity edge of the path between two nodes
|
| 42 |
42 |
/// in this tree has the same weight as the minimum cut in the digraph
|
| 43 |
43 |
/// between these nodes. Moreover the components obtained by removing
|
| 44 |
44 |
/// this edge from the tree determine the corresponding minimum cut.
|
| 45 |
45 |
///
|
| 46 |
46 |
/// Therefore once this tree is computed, the minimum cut between any pair
|
| 47 |
47 |
/// of nodes can easily be obtained.
|
| 48 |
48 |
///
|
| 49 |
49 |
/// The algorithm calculates \e n-1 distinct minimum cuts (currently with
|
| 50 |
50 |
/// the \ref Preflow algorithm), therefore the algorithm has
|
| 51 |
51 |
/// \f$(O(n^3\sqrt{e})\f$ overall time complexity. It calculates a
|
| 52 |
52 |
/// rooted Gomory-Hu tree, its structure and the weights can be obtained
|
| 53 |
53 |
/// by \c predNode(), \c predValue() and \c rootDist().
|
| 54 |
54 |
///
|
| 55 |
55 |
/// The members \c minCutMap() and \c minCutValue() calculate
|
| 56 |
56 |
/// the minimum cut and the minimum cut value between any two node
|
| 57 |
57 |
/// in the digraph. You can also list (iterate on) the nodes and the
|
| 58 |
58 |
/// edges of the cuts using MinCutNodeIt and MinCutEdgeIt.
|
| 59 |
59 |
///
|
| 60 |
60 |
/// \tparam GR The undirected graph data structure the algorithm will run on
|
| 61 |
61 |
/// \tparam CAP type of the EdgeMap describing the Edge capacities.
|
| 62 |
62 |
/// it is typename GR::template EdgeMap<int> by default.
|
| 63 |
63 |
template <typename GR,
|
| 64 |
64 |
typename CAP = typename GR::template EdgeMap<int>
|
| 65 |
65 |
>
|
| 66 |
|
class GomoryHuTree {
|
|
66 |
class GomoryHu {
|
| 67 |
67 |
public:
|
| 68 |
68 |
|
| 69 |
69 |
/// The graph type
|
| 70 |
70 |
typedef GR Graph;
|
| 71 |
71 |
/// The type if the edge capacity map
|
| 72 |
72 |
typedef CAP Capacity;
|
| 73 |
73 |
/// The value type of capacities
|
| 74 |
74 |
typedef typename Capacity::Value Value;
|
| 75 |
75 |
|
| 76 |
76 |
private:
|
| 77 |
77 |
|
| 78 |
78 |
TEMPLATE_GRAPH_TYPEDEFS(Graph);
|
| 79 |
79 |
|
| 80 |
80 |
const Graph& _graph;
|
| 81 |
81 |
const Capacity& _capacity;
|
| 82 |
82 |
|
| 83 |
83 |
Node _root;
|
| 84 |
84 |
typename Graph::template NodeMap<Node>* _pred;
|
| 85 |
85 |
typename Graph::template NodeMap<Value>* _weight;
|
| 86 |
86 |
typename Graph::template NodeMap<int>* _order;
|
| 87 |
87 |
|
| 88 |
88 |
void createStructures() {
|
| 89 |
89 |
if (!_pred) {
|
| 90 |
90 |
_pred = new typename Graph::template NodeMap<Node>(_graph);
|
| 91 |
91 |
}
|
| 92 |
92 |
if (!_weight) {
|
| 93 |
93 |
_weight = new typename Graph::template NodeMap<Value>(_graph);
|
| 94 |
94 |
}
|
| 95 |
95 |
if (!_order) {
|
| 96 |
96 |
_order = new typename Graph::template NodeMap<int>(_graph);
|
| 97 |
97 |
}
|
| 98 |
98 |
}
|
| 99 |
99 |
|
| 100 |
100 |
void destroyStructures() {
|
| 101 |
101 |
if (_pred) {
|
| 102 |
102 |
delete _pred;
|
| 103 |
103 |
}
|
| 104 |
104 |
if (_weight) {
|
| 105 |
105 |
delete _weight;
|
| 106 |
106 |
}
|
| 107 |
107 |
if (_order) {
|
| 108 |
108 |
delete _order;
|
| 109 |
109 |
}
|
| 110 |
110 |
}
|
| 111 |
111 |
|
| 112 |
112 |
public:
|
| 113 |
113 |
|
| 114 |
114 |
/// \brief Constructor
|
| 115 |
115 |
///
|
| 116 |
116 |
/// Constructor
|
| 117 |
117 |
/// \param graph The graph the algorithm will run on.
|
| 118 |
118 |
/// \param capacity The capacity map.
|
| 119 |
|
GomoryHuTree(const Graph& graph, const Capacity& capacity)
|
|
119 |
GomoryHu(const Graph& graph, const Capacity& capacity)
|
| 120 |
120 |
: _graph(graph), _capacity(capacity),
|
| 121 |
121 |
_pred(0), _weight(0), _order(0)
|
| 122 |
122 |
{
|
| 123 |
123 |
checkConcept<concepts::ReadMap<Edge, Value>, Capacity>();
|
| 124 |
124 |
}
|
| 125 |
125 |
|
| 126 |
126 |
|
| 127 |
127 |
/// \brief Destructor
|
| 128 |
128 |
///
|
| 129 |
129 |
/// Destructor
|
| 130 |
|
~GomoryHuTree() {
|
|
130 |
~GomoryHu() {
|
| 131 |
131 |
destroyStructures();
|
| 132 |
132 |
}
|
| 133 |
133 |
|
| 134 |
134 |
// \brief Initialize the internal data structures.
|
| 135 |
135 |
//
|
| 136 |
136 |
// This function initializes the internal data structures.
|
| 137 |
137 |
//
|
| 138 |
138 |
void init() {
|
| 139 |
139 |
createStructures();
|
| 140 |
140 |
|
| 141 |
141 |
_root = NodeIt(_graph);
|
| 142 |
142 |
for (NodeIt n(_graph); n != INVALID; ++n) {
|
| 143 |
143 |
_pred->set(n, _root);
|
| 144 |
144 |
_order->set(n, -1);
|
| 145 |
145 |
}
|
| 146 |
146 |
_pred->set(_root, INVALID);
|
| 147 |
147 |
_weight->set(_root, std::numeric_limits<Value>::max());
|
| 148 |
148 |
}
|
| 149 |
149 |
|
| 150 |
150 |
|
| 151 |
151 |
// \brief Start the algorithm
|
| 152 |
152 |
//
|
| 153 |
153 |
// This function starts the algorithm.
|
| 154 |
154 |
//
|
| 155 |
155 |
// \pre \ref init() must be called before using this function.
|
| 156 |
156 |
//
|
| 157 |
157 |
void start() {
|
| 158 |
158 |
Preflow<Graph, Capacity> fa(_graph, _capacity, _root, INVALID);
|
| 159 |
159 |
|
| 160 |
160 |
for (NodeIt n(_graph); n != INVALID; ++n) {
|
| 161 |
161 |
if (n == _root) continue;
|
| 162 |
162 |
|
| ... |
... |
@@ -311,87 +311,87 @@
|
| 311 |
311 |
}
|
| 312 |
312 |
|
| 313 |
313 |
typename Graph::template NodeMap<bool> reached(_graph, false);
|
| 314 |
314 |
reached.set(_root, true);
|
| 315 |
315 |
cutMap.set(_root, !s_root);
|
| 316 |
316 |
reached.set(rn, true);
|
| 317 |
317 |
cutMap.set(rn, s_root);
|
| 318 |
318 |
|
| 319 |
319 |
std::vector<Node> st;
|
| 320 |
320 |
for (NodeIt n(_graph); n != INVALID; ++n) {
|
| 321 |
321 |
st.clear();
|
| 322 |
322 |
Node nn = n;
|
| 323 |
323 |
while (!reached[nn]) {
|
| 324 |
324 |
st.push_back(nn);
|
| 325 |
325 |
nn = (*_pred)[nn];
|
| 326 |
326 |
}
|
| 327 |
327 |
while (!st.empty()) {
|
| 328 |
328 |
cutMap.set(st.back(), cutMap[nn]);
|
| 329 |
329 |
st.pop_back();
|
| 330 |
330 |
}
|
| 331 |
331 |
}
|
| 332 |
332 |
|
| 333 |
333 |
return value;
|
| 334 |
334 |
}
|
| 335 |
335 |
|
| 336 |
336 |
///@}
|
| 337 |
337 |
|
| 338 |
338 |
friend class MinCutNodeIt;
|
| 339 |
339 |
|
| 340 |
340 |
/// Iterate on the nodes of a minimum cut
|
| 341 |
341 |
|
| 342 |
342 |
/// This iterator class lists the nodes of a minimum cut found by
|
| 343 |
|
/// GomoryHuTree. Before using it, you must allocate a GomoryHuTree class,
|
| 344 |
|
/// and call its \ref GomoryHuTree::run() "run()" method.
|
|
343 |
/// GomoryHu. Before using it, you must allocate a GomoryHu class,
|
|
344 |
/// and call its \ref GomoryHu::run() "run()" method.
|
| 345 |
345 |
///
|
| 346 |
346 |
/// This example counts the nodes in the minimum cut separating \c s from
|
| 347 |
347 |
/// \c t.
|
| 348 |
348 |
/// \code
|
| 349 |
|
/// GomoruHuTree<Graph> gom(g, capacities);
|
|
349 |
/// GomoruHu<Graph> gom(g, capacities);
|
| 350 |
350 |
/// gom.run();
|
| 351 |
351 |
/// int sum=0;
|
| 352 |
|
/// for(GomoruHuTree<Graph>::MinCutNodeIt n(gom,s,t);n!=INVALID;++n) ++sum;
|
|
352 |
/// for(GomoruHu<Graph>::MinCutNodeIt n(gom,s,t);n!=INVALID;++n) ++sum;
|
| 353 |
353 |
/// \endcode
|
| 354 |
354 |
class MinCutNodeIt
|
| 355 |
355 |
{
|
| 356 |
356 |
bool _side;
|
| 357 |
357 |
typename Graph::NodeIt _node_it;
|
| 358 |
358 |
typename Graph::template NodeMap<bool> _cut;
|
| 359 |
359 |
public:
|
| 360 |
360 |
/// Constructor
|
| 361 |
361 |
|
| 362 |
362 |
/// Constructor
|
| 363 |
363 |
///
|
| 364 |
|
MinCutNodeIt(GomoryHuTree const &gomory,
|
| 365 |
|
///< The GomoryHuTree class. You must call its
|
|
364 |
MinCutNodeIt(GomoryHu const &gomory,
|
|
365 |
///< The GomoryHu class. You must call its
|
| 366 |
366 |
/// run() method
|
| 367 |
367 |
/// before initializing this iterator
|
| 368 |
368 |
const Node& s, ///< Base node
|
| 369 |
369 |
const Node& t,
|
| 370 |
370 |
///< The node you want to separate from Node s.
|
| 371 |
371 |
bool side=true
|
| 372 |
372 |
///< If it is \c true (default) then the iterator lists
|
| 373 |
373 |
/// the nodes of the component containing \c s,
|
| 374 |
374 |
/// otherwise it lists the other component.
|
| 375 |
375 |
/// \note As the minimum cut is not always unique,
|
| 376 |
376 |
/// \code
|
| 377 |
377 |
/// MinCutNodeIt(gomory, s, t, true);
|
| 378 |
378 |
/// \endcode
|
| 379 |
379 |
/// and
|
| 380 |
380 |
/// \code
|
| 381 |
381 |
/// MinCutNodeIt(gomory, t, s, false);
|
| 382 |
382 |
/// \endcode
|
| 383 |
383 |
/// does not necessarily give the same set of nodes.
|
| 384 |
384 |
/// However it is ensured that
|
| 385 |
385 |
/// \code
|
| 386 |
386 |
/// MinCutNodeIt(gomory, s, t, true);
|
| 387 |
387 |
/// \endcode
|
| 388 |
388 |
/// and
|
| 389 |
389 |
/// \code
|
| 390 |
390 |
/// MinCutNodeIt(gomory, s, t, false);
|
| 391 |
391 |
/// \endcode
|
| 392 |
392 |
/// together list each node exactly once.
|
| 393 |
393 |
)
|
| 394 |
394 |
: _side(side), _cut(gomory._graph)
|
| 395 |
395 |
{
|
| 396 |
396 |
gomory.minCutMap(s,t,_cut);
|
| 397 |
397 |
for(_node_it=typename Graph::NodeIt(gomory._graph);
|
| ... |
... |
@@ -408,99 +408,99 @@
|
| 408 |
408 |
}
|
| 409 |
409 |
bool operator==(Invalid) { return _node_it==INVALID; }
|
| 410 |
410 |
bool operator!=(Invalid) { return _node_it!=INVALID; }
|
| 411 |
411 |
/// Next node
|
| 412 |
412 |
|
| 413 |
413 |
/// Next node
|
| 414 |
414 |
///
|
| 415 |
415 |
MinCutNodeIt &operator++()
|
| 416 |
416 |
{
|
| 417 |
417 |
for(++_node_it;_node_it!=INVALID&&_cut[_node_it]!=_side;++_node_it) {}
|
| 418 |
418 |
return *this;
|
| 419 |
419 |
}
|
| 420 |
420 |
/// Postfix incrementation
|
| 421 |
421 |
|
| 422 |
422 |
/// Postfix incrementation
|
| 423 |
423 |
///
|
| 424 |
424 |
/// \warning This incrementation
|
| 425 |
425 |
/// returns a \c Node, not a \ref MinCutNodeIt, as one may
|
| 426 |
426 |
/// expect.
|
| 427 |
427 |
typename Graph::Node operator++(int)
|
| 428 |
428 |
{
|
| 429 |
429 |
typename Graph::Node n=*this;
|
| 430 |
430 |
++(*this);
|
| 431 |
431 |
return n;
|
| 432 |
432 |
}
|
| 433 |
433 |
};
|
| 434 |
434 |
|
| 435 |
435 |
friend class MinCutEdgeIt;
|
| 436 |
436 |
|
| 437 |
437 |
/// Iterate on the edges of a minimum cut
|
| 438 |
438 |
|
| 439 |
439 |
/// This iterator class lists the edges of a minimum cut found by
|
| 440 |
|
/// GomoryHuTree. Before using it, you must allocate a GomoryHuTree class,
|
| 441 |
|
/// and call its \ref GomoryHuTree::run() "run()" method.
|
|
440 |
/// GomoryHu. Before using it, you must allocate a GomoryHu class,
|
|
441 |
/// and call its \ref GomoryHu::run() "run()" method.
|
| 442 |
442 |
///
|
| 443 |
443 |
/// This example computes the value of the minimum cut separating \c s from
|
| 444 |
444 |
/// \c t.
|
| 445 |
445 |
/// \code
|
| 446 |
|
/// GomoruHuTree<Graph> gom(g, capacities);
|
|
446 |
/// GomoruHu<Graph> gom(g, capacities);
|
| 447 |
447 |
/// gom.run();
|
| 448 |
448 |
/// int value=0;
|
| 449 |
|
/// for(GomoruHuTree<Graph>::MinCutEdgeIt e(gom,s,t);e!=INVALID;++e)
|
|
449 |
/// for(GomoruHu<Graph>::MinCutEdgeIt e(gom,s,t);e!=INVALID;++e)
|
| 450 |
450 |
/// value+=capacities[e];
|
| 451 |
451 |
/// \endcode
|
| 452 |
452 |
/// the result will be the same as it is returned by
|
| 453 |
|
/// \ref GomoryHuTree::minCostValue() "gom.minCostValue(s,t)"
|
|
453 |
/// \ref GomoryHu::minCostValue() "gom.minCostValue(s,t)"
|
| 454 |
454 |
class MinCutEdgeIt
|
| 455 |
455 |
{
|
| 456 |
456 |
bool _side;
|
| 457 |
457 |
const Graph &_graph;
|
| 458 |
458 |
typename Graph::NodeIt _node_it;
|
| 459 |
459 |
typename Graph::OutArcIt _arc_it;
|
| 460 |
460 |
typename Graph::template NodeMap<bool> _cut;
|
| 461 |
461 |
void step()
|
| 462 |
462 |
{
|
| 463 |
463 |
++_arc_it;
|
| 464 |
464 |
while(_node_it!=INVALID && _arc_it==INVALID)
|
| 465 |
465 |
{
|
| 466 |
466 |
for(++_node_it;_node_it!=INVALID&&!_cut[_node_it];++_node_it) {}
|
| 467 |
467 |
if(_node_it!=INVALID)
|
| 468 |
468 |
_arc_it=typename Graph::OutArcIt(_graph,_node_it);
|
| 469 |
469 |
}
|
| 470 |
470 |
}
|
| 471 |
471 |
|
| 472 |
472 |
public:
|
| 473 |
|
MinCutEdgeIt(GomoryHuTree const &gomory,
|
| 474 |
|
///< The GomoryHuTree class. You must call its
|
|
473 |
MinCutEdgeIt(GomoryHu const &gomory,
|
|
474 |
///< The GomoryHu class. You must call its
|
| 475 |
475 |
/// run() method
|
| 476 |
476 |
/// before initializing this iterator
|
| 477 |
477 |
const Node& s, ///< Base node
|
| 478 |
478 |
const Node& t,
|
| 479 |
479 |
///< The node you want to separate from Node s.
|
| 480 |
480 |
bool side=true
|
| 481 |
481 |
///< If it is \c true (default) then the listed arcs
|
| 482 |
482 |
/// will be oriented from the
|
| 483 |
483 |
/// the nodes of the component containing \c s,
|
| 484 |
484 |
/// otherwise they will be oriented in the opposite
|
| 485 |
485 |
/// direction.
|
| 486 |
486 |
)
|
| 487 |
487 |
: _graph(gomory._graph), _cut(_graph)
|
| 488 |
488 |
{
|
| 489 |
489 |
gomory.minCutMap(s,t,_cut);
|
| 490 |
490 |
if(!side)
|
| 491 |
491 |
for(typename Graph::NodeIt n(_graph);n!=INVALID;++n)
|
| 492 |
492 |
_cut[n]=!_cut[n];
|
| 493 |
493 |
|
| 494 |
494 |
for(_node_it=typename Graph::NodeIt(_graph);
|
| 495 |
495 |
_node_it!=INVALID && !_cut[_node_it];
|
| 496 |
496 |
++_node_it) {}
|
| 497 |
497 |
_arc_it = _node_it!=INVALID ?
|
| 498 |
498 |
typename Graph::OutArcIt(_graph,_node_it) : INVALID;
|
| 499 |
499 |
while(_node_it!=INVALID && _arc_it == INVALID)
|
| 500 |
500 |
{
|
| 501 |
501 |
for(++_node_it; _node_it!=INVALID&&!_cut[_node_it]; ++_node_it) {}
|
| 502 |
502 |
if(_node_it!=INVALID)
|
| 503 |
503 |
_arc_it= typename Graph::OutArcIt(_graph,_node_it);
|
| 504 |
504 |
}
|
| 505 |
505 |
while(_arc_it!=INVALID && _cut[_graph.target(_arc_it)]) step();
|
| 506 |
506 |
}
|