1 /* -*- mode: C++; indent-tabs-mode: nil; -*-
3 * This file is a part of LEMON, a generic C++ optimization library.
5 * Copyright (C) 2003-2010
6 * Egervary Jeno Kombinatorikus Optimalizalasi Kutatocsoport
7 * (Egervary Research Group on Combinatorial Optimization, EGRES).
9 * Permission to use, modify and distribute this software is granted
10 * provided that this copyright notice appears in all copies. For
11 * precise terms see the accompanying LICENSE file.
13 * This software is provided "AS IS" with no warranty of any kind,
14 * express or implied, and with no claim as to its suitability for any
19 #ifndef LEMON_CIRCULATION_H
20 #define LEMON_CIRCULATION_H
22 #include <lemon/tolerance.h>
23 #include <lemon/elevator.h>
28 ///\brief Push-relabel algorithm for finding a feasible circulation.
32 /// \brief Default traits class of Circulation class.
34 /// Default traits class of Circulation class.
36 /// \tparam GR Type of the digraph the algorithm runs on.
37 /// \tparam LM The type of the lower bound map.
38 /// \tparam UM The type of the upper bound (capacity) map.
39 /// \tparam SM The type of the supply map.
40 template <typename GR, typename LM,
41 typename UM, typename SM>
42 struct CirculationDefaultTraits {
44 /// \brief The type of the digraph the algorithm runs on.
47 /// \brief The type of the lower bound map.
49 /// The type of the map that stores the lower bounds on the arcs.
50 /// It must conform to the \ref concepts::ReadMap "ReadMap" concept.
53 /// \brief The type of the upper bound (capacity) map.
55 /// The type of the map that stores the upper bounds (capacities)
57 /// It must conform to the \ref concepts::ReadMap "ReadMap" concept.
60 /// \brief The type of supply map.
62 /// The type of the map that stores the signed supply values of the
64 /// It must conform to the \ref concepts::ReadMap "ReadMap" concept.
67 /// \brief The type of the flow and supply values.
68 typedef typename SupplyMap::Value Value;
70 /// \brief The type of the map that stores the flow values.
72 /// The type of the map that stores the flow values.
73 /// It must conform to the \ref concepts::ReadWriteMap "ReadWriteMap"
76 typedef GR::ArcMap<Value> FlowMap;
78 typedef typename Digraph::template ArcMap<Value> FlowMap;
81 /// \brief Instantiates a FlowMap.
83 /// This function instantiates a \ref FlowMap.
84 /// \param digraph The digraph for which we would like to define
86 static FlowMap* createFlowMap(const Digraph& digraph) {
87 return new FlowMap(digraph);
90 /// \brief The elevator type used by the algorithm.
92 /// The elevator type used by the algorithm.
94 /// \sa Elevator, LinkedElevator
96 typedef lemon::Elevator<GR, GR::Node> Elevator;
98 typedef lemon::Elevator<Digraph, typename Digraph::Node> Elevator;
101 /// \brief Instantiates an Elevator.
103 /// This function instantiates an \ref Elevator.
104 /// \param digraph The digraph for which we would like to define
106 /// \param max_level The maximum level of the elevator.
107 static Elevator* createElevator(const Digraph& digraph, int max_level) {
108 return new Elevator(digraph, max_level);
111 /// \brief The tolerance used by the algorithm
113 /// The tolerance used by the algorithm to handle inexact computation.
114 typedef lemon::Tolerance<Value> Tolerance;
119 \brief Push-relabel algorithm for the network circulation problem.
122 This class implements a push-relabel algorithm for the \e network
123 \e circulation problem.
124 It is to find a feasible circulation when lower and upper bounds
125 are given for the flow values on the arcs and lower bounds are
126 given for the difference between the outgoing and incoming flow
129 The exact formulation of this problem is the following.
130 Let \f$G=(V,A)\f$ be a digraph, \f$lower: A\rightarrow\mathbf{R}\f$
131 \f$upper: A\rightarrow\mathbf{R}\cup\{\infty\}\f$ denote the lower and
132 upper bounds on the arcs, for which \f$lower(uv) \leq upper(uv)\f$
133 holds for all \f$uv\in A\f$, and \f$sup: V\rightarrow\mathbf{R}\f$
134 denotes the signed supply values of the nodes.
135 If \f$sup(u)>0\f$, then \f$u\f$ is a supply node with \f$sup(u)\f$
136 supply, if \f$sup(u)<0\f$, then \f$u\f$ is a demand node with
137 \f$-sup(u)\f$ demand.
138 A feasible circulation is an \f$f: A\rightarrow\mathbf{R}\f$
139 solution of the following problem.
141 \f[ \sum_{uv\in A} f(uv) - \sum_{vu\in A} f(vu)
142 \geq sup(u) \quad \forall u\in V, \f]
143 \f[ lower(uv) \leq f(uv) \leq upper(uv) \quad \forall uv\in A. \f]
145 The sum of the supply values, i.e. \f$\sum_{u\in V} sup(u)\f$ must be
146 zero or negative in order to have a feasible solution (since the sum
147 of the expressions on the left-hand side of the inequalities is zero).
148 It means that the total demand must be greater or equal to the total
149 supply and all the supplies have to be carried out from the supply nodes,
150 but there could be demands that are not satisfied.
151 If \f$\sum_{u\in V} sup(u)\f$ is zero, then all the supply/demand
152 constraints have to be satisfied with equality, i.e. all demands
153 have to be satisfied and all supplies have to be used.
155 If you need the opposite inequalities in the supply/demand constraints
156 (i.e. the total demand is less than the total supply and all the demands
157 have to be satisfied while there could be supplies that are not used),
158 then you could easily transform the problem to the above form by reversing
159 the direction of the arcs and taking the negative of the supply values
160 (e.g. using \ref ReverseDigraph and \ref NegMap adaptors).
162 This algorithm either calculates a feasible circulation, or provides
163 a \ref barrier() "barrier", which prooves that a feasible soultion
166 Note that this algorithm also provides a feasible solution for the
167 \ref min_cost_flow "minimum cost flow problem".
169 \tparam GR The type of the digraph the algorithm runs on.
170 \tparam LM The type of the lower bound map. The default
171 map type is \ref concepts::Digraph::ArcMap "GR::ArcMap<int>".
172 \tparam UM The type of the upper bound (capacity) map.
173 The default map type is \c LM.
174 \tparam SM The type of the supply map. The default map type is
175 \ref concepts::Digraph::NodeMap "GR::NodeMap<UM::Value>".
176 \tparam TR The traits class that defines various types used by the
177 algorithm. By default, it is \ref CirculationDefaultTraits
178 "CirculationDefaultTraits<GR, LM, UM, SM>".
179 In most cases, this parameter should not be set directly,
180 consider to use the named template parameters instead.
183 template< typename GR,
189 template< typename GR,
190 typename LM = typename GR::template ArcMap<int>,
192 typename SM = typename GR::template NodeMap<typename UM::Value>,
193 typename TR = CirculationDefaultTraits<GR, LM, UM, SM> >
198 ///The \ref CirculationDefaultTraits "traits class" of the algorithm.
200 ///The type of the digraph the algorithm runs on.
201 typedef typename Traits::Digraph Digraph;
202 ///The type of the flow and supply values.
203 typedef typename Traits::Value Value;
205 ///The type of the lower bound map.
206 typedef typename Traits::LowerMap LowerMap;
207 ///The type of the upper bound (capacity) map.
208 typedef typename Traits::UpperMap UpperMap;
209 ///The type of the supply map.
210 typedef typename Traits::SupplyMap SupplyMap;
211 ///The type of the flow map.
212 typedef typename Traits::FlowMap FlowMap;
214 ///The type of the elevator.
215 typedef typename Traits::Elevator Elevator;
216 ///The type of the tolerance.
217 typedef typename Traits::Tolerance Tolerance;
221 TEMPLATE_DIGRAPH_TYPEDEFS(Digraph);
228 const SupplyMap *_supply;
236 typedef typename Digraph::template NodeMap<Value> ExcessMap;
244 typedef Circulation Create;
246 ///\name Named Template Parameters
250 template <typename T>
251 struct SetFlowMapTraits : public Traits {
253 static FlowMap *createFlowMap(const Digraph&) {
254 LEMON_ASSERT(false, "FlowMap is not initialized");
255 return 0; // ignore warnings
259 /// \brief \ref named-templ-param "Named parameter" for setting
262 /// \ref named-templ-param "Named parameter" for setting FlowMap
264 template <typename T>
266 : public Circulation<Digraph, LowerMap, UpperMap, SupplyMap,
267 SetFlowMapTraits<T> > {
268 typedef Circulation<Digraph, LowerMap, UpperMap, SupplyMap,
269 SetFlowMapTraits<T> > Create;
272 template <typename T>
273 struct SetElevatorTraits : public Traits {
275 static Elevator *createElevator(const Digraph&, int) {
276 LEMON_ASSERT(false, "Elevator is not initialized");
277 return 0; // ignore warnings
281 /// \brief \ref named-templ-param "Named parameter" for setting
284 /// \ref named-templ-param "Named parameter" for setting Elevator
285 /// type. If this named parameter is used, then an external
286 /// elevator object must be passed to the algorithm using the
287 /// \ref elevator(Elevator&) "elevator()" function before calling
288 /// \ref run() or \ref init().
289 /// \sa SetStandardElevator
290 template <typename T>
292 : public Circulation<Digraph, LowerMap, UpperMap, SupplyMap,
293 SetElevatorTraits<T> > {
294 typedef Circulation<Digraph, LowerMap, UpperMap, SupplyMap,
295 SetElevatorTraits<T> > Create;
298 template <typename T>
299 struct SetStandardElevatorTraits : public Traits {
301 static Elevator *createElevator(const Digraph& digraph, int max_level) {
302 return new Elevator(digraph, max_level);
306 /// \brief \ref named-templ-param "Named parameter" for setting
307 /// Elevator type with automatic allocation
309 /// \ref named-templ-param "Named parameter" for setting Elevator
310 /// type with automatic allocation.
311 /// The Elevator should have standard constructor interface to be
312 /// able to automatically created by the algorithm (i.e. the
313 /// digraph and the maximum level should be passed to it).
314 /// However, an external elevator object could also be passed to the
315 /// algorithm with the \ref elevator(Elevator&) "elevator()" function
316 /// before calling \ref run() or \ref init().
318 template <typename T>
319 struct SetStandardElevator
320 : public Circulation<Digraph, LowerMap, UpperMap, SupplyMap,
321 SetStandardElevatorTraits<T> > {
322 typedef Circulation<Digraph, LowerMap, UpperMap, SupplyMap,
323 SetStandardElevatorTraits<T> > Create;
336 /// The constructor of the class.
338 /// \param graph The digraph the algorithm runs on.
339 /// \param lower The lower bounds for the flow values on the arcs.
340 /// \param upper The upper bounds (capacities) for the flow values
342 /// \param supply The signed supply values of the nodes.
343 Circulation(const Digraph &graph, const LowerMap &lower,
344 const UpperMap &upper, const SupplyMap &supply)
345 : _g(graph), _lo(&lower), _up(&upper), _supply(&supply),
346 _flow(NULL), _local_flow(false), _level(NULL), _local_level(false),
357 bool checkBoundMaps() {
358 for (ArcIt e(_g);e!=INVALID;++e) {
359 if (_tol.less((*_up)[e], (*_lo)[e])) return false;
364 void createStructures() {
365 _node_num = _el = countNodes(_g);
368 _flow = Traits::createFlowMap(_g);
372 _level = Traits::createElevator(_g, _node_num);
376 _excess = new ExcessMap(_g);
380 void destroyStructures() {
394 /// Sets the lower bound map.
396 /// Sets the lower bound map.
397 /// \return <tt>(*this)</tt>
398 Circulation& lowerMap(const LowerMap& map) {
403 /// Sets the upper bound (capacity) map.
405 /// Sets the upper bound (capacity) map.
406 /// \return <tt>(*this)</tt>
407 Circulation& upperMap(const UpperMap& map) {
412 /// Sets the supply map.
414 /// Sets the supply map.
415 /// \return <tt>(*this)</tt>
416 Circulation& supplyMap(const SupplyMap& map) {
421 /// \brief Sets the flow map.
423 /// Sets the flow map.
424 /// If you don't use this function before calling \ref run() or
425 /// \ref init(), an instance will be allocated automatically.
426 /// The destructor deallocates this automatically allocated map,
428 /// \return <tt>(*this)</tt>
429 Circulation& flowMap(FlowMap& map) {
438 /// \brief Sets the elevator used by algorithm.
440 /// Sets the elevator used by algorithm.
441 /// If you don't use this function before calling \ref run() or
442 /// \ref init(), an instance will be allocated automatically.
443 /// The destructor deallocates this automatically allocated elevator,
445 /// \return <tt>(*this)</tt>
446 Circulation& elevator(Elevator& elevator) {
449 _local_level = false;
455 /// \brief Returns a const reference to the elevator.
457 /// Returns a const reference to the elevator.
459 /// \pre Either \ref run() or \ref init() must be called before
460 /// using this function.
461 const Elevator& elevator() const {
465 /// \brief Sets the tolerance used by the algorithm.
467 /// Sets the tolerance object used by the algorithm.
468 /// \return <tt>(*this)</tt>
469 Circulation& tolerance(const Tolerance& tolerance) {
474 /// \brief Returns a const reference to the tolerance.
476 /// Returns a const reference to the tolerance object used by
478 const Tolerance& tolerance() const {
482 /// \name Execution Control
483 /// The simplest way to execute the algorithm is to call \ref run().\n
484 /// If you need better control on the initial solution or the execution,
485 /// you have to call one of the \ref init() functions first, then
486 /// the \ref start() function.
490 /// Initializes the internal data structures.
492 /// Initializes the internal data structures and sets all flow values
493 /// to the lower bound.
496 LEMON_DEBUG(checkBoundMaps(),
497 "Upper bounds must be greater or equal to the lower bounds");
501 for(NodeIt n(_g);n!=INVALID;++n) {
502 (*_excess)[n] = (*_supply)[n];
505 for (ArcIt e(_g);e!=INVALID;++e) {
506 _flow->set(e, (*_lo)[e]);
507 (*_excess)[_g.target(e)] += (*_flow)[e];
508 (*_excess)[_g.source(e)] -= (*_flow)[e];
511 // global relabeling tested, but in general case it provides
512 // worse performance for random digraphs
514 for(NodeIt n(_g);n!=INVALID;++n)
515 _level->initAddItem(n);
516 _level->initFinish();
517 for(NodeIt n(_g);n!=INVALID;++n)
518 if(_tol.positive((*_excess)[n]))
522 /// Initializes the internal data structures using a greedy approach.
524 /// Initializes the internal data structures using a greedy approach
525 /// to construct the initial solution.
528 LEMON_DEBUG(checkBoundMaps(),
529 "Upper bounds must be greater or equal to the lower bounds");
533 for(NodeIt n(_g);n!=INVALID;++n) {
534 (*_excess)[n] = (*_supply)[n];
537 for (ArcIt e(_g);e!=INVALID;++e) {
538 if (!_tol.less(-(*_excess)[_g.target(e)], (*_up)[e])) {
539 _flow->set(e, (*_up)[e]);
540 (*_excess)[_g.target(e)] += (*_up)[e];
541 (*_excess)[_g.source(e)] -= (*_up)[e];
542 } else if (_tol.less(-(*_excess)[_g.target(e)], (*_lo)[e])) {
543 _flow->set(e, (*_lo)[e]);
544 (*_excess)[_g.target(e)] += (*_lo)[e];
545 (*_excess)[_g.source(e)] -= (*_lo)[e];
547 Value fc = -(*_excess)[_g.target(e)];
549 (*_excess)[_g.target(e)] = 0;
550 (*_excess)[_g.source(e)] -= fc;
555 for(NodeIt n(_g);n!=INVALID;++n)
556 _level->initAddItem(n);
557 _level->initFinish();
558 for(NodeIt n(_g);n!=INVALID;++n)
559 if(_tol.positive((*_excess)[n]))
563 ///Executes the algorithm
565 ///This function executes the algorithm.
567 ///\return \c true if a feasible circulation is found.
575 while((act=_level->highestActive())!=INVALID) {
576 int actlevel=(*_level)[act];
577 int mlevel=_node_num;
578 Value exc=(*_excess)[act];
580 for(OutArcIt e(_g,act);e!=INVALID; ++e) {
581 Node v = _g.target(e);
582 Value fc=(*_up)[e]-(*_flow)[e];
583 if(!_tol.positive(fc)) continue;
584 if((*_level)[v]<actlevel) {
585 if(!_tol.less(fc, exc)) {
586 _flow->set(e, (*_flow)[e] + exc);
587 (*_excess)[v] += exc;
588 if(!_level->active(v) && _tol.positive((*_excess)[v]))
591 _level->deactivate(act);
595 _flow->set(e, (*_up)[e]);
597 if(!_level->active(v) && _tol.positive((*_excess)[v]))
602 else if((*_level)[v]<mlevel) mlevel=(*_level)[v];
604 for(InArcIt e(_g,act);e!=INVALID; ++e) {
605 Node v = _g.source(e);
606 Value fc=(*_flow)[e]-(*_lo)[e];
607 if(!_tol.positive(fc)) continue;
608 if((*_level)[v]<actlevel) {
609 if(!_tol.less(fc, exc)) {
610 _flow->set(e, (*_flow)[e] - exc);
611 (*_excess)[v] += exc;
612 if(!_level->active(v) && _tol.positive((*_excess)[v]))
615 _level->deactivate(act);
619 _flow->set(e, (*_lo)[e]);
621 if(!_level->active(v) && _tol.positive((*_excess)[v]))
626 else if((*_level)[v]<mlevel) mlevel=(*_level)[v];
629 (*_excess)[act] = exc;
630 if(!_tol.positive(exc)) _level->deactivate(act);
631 else if(mlevel==_node_num) {
632 _level->liftHighestActiveToTop();
637 _level->liftHighestActive(mlevel+1);
638 if(_level->onLevel(actlevel)==0) {
649 /// Runs the algorithm.
651 /// This function runs the algorithm.
653 /// \return \c true if a feasible circulation is found.
655 /// \note Apart from the return value, c.run() is just a shortcut of
656 /// the following code.
668 /// \name Query Functions
669 /// The results of the circulation algorithm can be obtained using
670 /// these functions.\n
671 /// Either \ref run() or \ref start() should be called before
676 /// \brief Returns the flow value on the given arc.
678 /// Returns the flow value on the given arc.
680 /// \pre Either \ref run() or \ref init() must be called before
681 /// using this function.
682 Value flow(const Arc& arc) const {
683 return (*_flow)[arc];
686 /// \brief Returns a const reference to the flow map.
688 /// Returns a const reference to the arc map storing the found flow.
690 /// \pre Either \ref run() or \ref init() must be called before
691 /// using this function.
692 const FlowMap& flowMap() const {
697 \brief Returns \c true if the given node is in a barrier.
699 Barrier is a set \e B of nodes for which
701 \f[ \sum_{uv\in A: u\in B} upper(uv) -
702 \sum_{uv\in A: v\in B} lower(uv) < \sum_{v\in B} sup(v) \f]
704 holds. The existence of a set with this property prooves that a
705 feasible circualtion cannot exist.
707 This function returns \c true if the given node is in the found
708 barrier. If a feasible circulation is found, the function
709 gives back \c false for every node.
711 \pre Either \ref run() or \ref init() must be called before
717 bool barrier(const Node& node) const
719 return (*_level)[node] >= _el;
722 /// \brief Gives back a barrier.
724 /// This function sets \c bar to the characteristic vector of the
725 /// found barrier. \c bar should be a \ref concepts::WriteMap "writable"
726 /// node map with \c bool (or convertible) value type.
728 /// If a feasible circulation is found, the function gives back an
729 /// empty set, so \c bar[v] will be \c false for all nodes \c v.
731 /// \note This function calls \ref barrier() for each node,
732 /// so it runs in O(n) time.
734 /// \pre Either \ref run() or \ref init() must be called before
735 /// using this function.
738 /// \sa checkBarrier()
739 template<class BarrierMap>
740 void barrierMap(BarrierMap &bar) const
742 for(NodeIt n(_g);n!=INVALID;++n)
743 bar.set(n, (*_level)[n] >= _el);
748 /// \name Checker Functions
749 /// The feasibility of the results can be checked using
750 /// these functions.\n
751 /// Either \ref run() or \ref start() should be called before
756 ///Check if the found flow is a feasible circulation
758 ///Check if the found flow is a feasible circulation,
760 bool checkFlow() const {
761 for(ArcIt e(_g);e!=INVALID;++e)
762 if((*_flow)[e]<(*_lo)[e]||(*_flow)[e]>(*_up)[e]) return false;
763 for(NodeIt n(_g);n!=INVALID;++n)
765 Value dif=-(*_supply)[n];
766 for(InArcIt e(_g,n);e!=INVALID;++e) dif-=(*_flow)[e];
767 for(OutArcIt e(_g,n);e!=INVALID;++e) dif+=(*_flow)[e];
768 if(_tol.negative(dif)) return false;
773 ///Check whether or not the last execution provides a barrier
775 ///Check whether or not the last execution provides a barrier.
778 bool checkBarrier() const
781 Value inf_cap = std::numeric_limits<Value>::has_infinity ?
782 std::numeric_limits<Value>::infinity() :
783 std::numeric_limits<Value>::max();
784 for(NodeIt n(_g);n!=INVALID;++n)
786 delta-=(*_supply)[n];
787 for(ArcIt e(_g);e!=INVALID;++e)
791 if(barrier(s)&&!barrier(t)) {
792 if (_tol.less(inf_cap - (*_up)[e], delta)) return false;
795 else if(barrier(t)&&!barrier(s)) delta-=(*_lo)[e];
797 return _tol.negative(delta);