src/work/marci/augmenting_flow.h
author marci
Tue, 17 Aug 2004 11:20:16 +0000
changeset 762 511200bdb71f
child 775 e46a1f0623a0
permissions -rw-r--r--
technical corrections
marci@762
     1
// -*- C++ -*-
marci@762
     2
#ifndef HUGO_AUGMENTING_FLOW_H
marci@762
     3
#define HUGO_AUGMENTING_FLOW_H
marci@762
     4
marci@762
     5
#include <vector>
marci@762
     6
#include <queue>
marci@762
     7
#include <stack>
marci@762
     8
#include <iostream>
marci@762
     9
marci@762
    10
#include <hugo/graph_wrapper.h>
marci@762
    11
#include <bfs_dfs.h>
marci@762
    12
#include <hugo/invalid.h>
marci@762
    13
#include <hugo/maps.h>
marci@762
    14
#include <for_each_macros.h>
marci@762
    15
marci@762
    16
/// \file
marci@762
    17
/// \brief Maximum flow algorithms.
marci@762
    18
/// \ingroup galgs
marci@762
    19
marci@762
    20
namespace hugo {
marci@762
    21
marci@762
    22
  /// \addtogroup galgs
marci@762
    23
  /// @{                                                                                                                                        
marci@762
    24
  ///Maximum flow algorithms class.
marci@762
    25
marci@762
    26
  ///This class provides various algorithms for finding a flow of
marci@762
    27
  ///maximum value in a directed graph. The \e source node, the \e
marci@762
    28
  ///target node, the \e capacity of the edges and the \e starting \e
marci@762
    29
  ///flow value of the edges should be passed to the algorithm through the
marci@762
    30
  ///constructor. It is possible to change these quantities using the
marci@762
    31
  ///functions \ref resetSource, \ref resetTarget, \ref resetCap and
marci@762
    32
  ///\ref resetFlow. Before any subsequent runs of any algorithm of
marci@762
    33
  ///the class \ref resetFlow should be called. 
marci@762
    34
marci@762
    35
  ///After running an algorithm of the class, the actual flow value 
marci@762
    36
  ///can be obtained by calling \ref flowValue(). The minimum
marci@762
    37
  ///value cut can be written into a \c node map of \c bools by
marci@762
    38
  ///calling \ref minCut. (\ref minMinCut and \ref maxMinCut writes
marci@762
    39
  ///the inclusionwise minimum and maximum of the minimum value
marci@762
    40
  ///cuts, resp.)                                                                                                                               
marci@762
    41
  ///\param Graph The directed graph type the algorithm runs on.
marci@762
    42
  ///\param Num The number type of the capacities and the flow values.
marci@762
    43
  ///\param CapMap The capacity map type.
marci@762
    44
  ///\param FlowMap The flow map type.                                                                                                           
marci@762
    45
  ///\author Marton Makai, Jacint Szabo 
marci@762
    46
//   template <typename Graph, typename Num,
marci@762
    47
// 	    typename CapMap=typename Graph::template EdgeMap<Num>,
marci@762
    48
//             typename FlowMap=typename Graph::template EdgeMap<Num> >
marci@762
    49
//   class MaxFlow {
marci@762
    50
//   protected:
marci@762
    51
//     typedef typename Graph::Node Node;
marci@762
    52
//     typedef typename Graph::NodeIt NodeIt;
marci@762
    53
//     typedef typename Graph::EdgeIt EdgeIt;
marci@762
    54
//     typedef typename Graph::OutEdgeIt OutEdgeIt;
marci@762
    55
//     typedef typename Graph::InEdgeIt InEdgeIt;
marci@762
    56
marci@762
    57
//     typedef typename std::vector<std::stack<Node> > VecStack;
marci@762
    58
//     typedef typename Graph::template NodeMap<Node> NNMap;
marci@762
    59
//     typedef typename std::vector<Node> VecNode;
marci@762
    60
marci@762
    61
//     const Graph* g;
marci@762
    62
//     Node s;
marci@762
    63
//     Node t;
marci@762
    64
//     const CapMap* capacity;
marci@762
    65
//     FlowMap* flow;
marci@762
    66
//     int n;      //the number of nodes of G
marci@762
    67
//     typedef ResGraphWrapper<const Graph, Num, CapMap, FlowMap> ResGW;   
marci@762
    68
//     //typedef ExpResGraphWrapper<const Graph, Num, CapMap, FlowMap> ResGW;
marci@762
    69
//     typedef typename ResGW::OutEdgeIt ResGWOutEdgeIt;
marci@762
    70
//     typedef typename ResGW::Edge ResGWEdge;
marci@762
    71
//     //typedef typename ResGW::template NodeMap<bool> ReachedMap;
marci@762
    72
//     typedef typename Graph::template NodeMap<int> ReachedMap;
marci@762
    73
marci@762
    74
marci@762
    75
//     //level works as a bool map in augmenting path algorithms and is
marci@762
    76
//     //used by bfs for storing reached information.  In preflow, it
marci@762
    77
//     //shows the levels of nodes.     
marci@762
    78
//     ReachedMap level;
marci@762
    79
marci@762
    80
//     //excess is needed only in preflow
marci@762
    81
//     typename Graph::template NodeMap<Num> excess;
marci@762
    82
marci@762
    83
//     //fixme    
marci@762
    84
// //   protected:
marci@762
    85
//     //     MaxFlow() { }
marci@762
    86
//     //     void set(const Graph& _G, Node _s, Node _t, const CapMap& _capacity,
marci@762
    87
//     // 	     FlowMap& _flow)
marci@762
    88
//     //       {
marci@762
    89
//     // 	g=&_G;
marci@762
    90
//     // 	s=_s;
marci@762
    91
//     // 	t=_t;
marci@762
    92
//     // 	capacity=&_capacity;
marci@762
    93
//     // 	flow=&_flow;
marci@762
    94
//     // 	n=_G.nodeNum;
marci@762
    95
//     // 	level.set (_G); //kellene vmi ilyesmi fv
marci@762
    96
//     // 	excess(_G,0); //itt is
marci@762
    97
//     //       }
marci@762
    98
marci@762
    99
//     // constants used for heuristics
marci@762
   100
//     static const int H0=20;
marci@762
   101
//     static const int H1=1;
marci@762
   102
marci@762
   103
//   public:
marci@762
   104
marci@762
   105
//     ///Indicates the property of the starting flow.
marci@762
   106
marci@762
   107
//     ///Indicates the property of the starting flow. The meanings are as follows:
marci@762
   108
//     ///- \c ZERO_FLOW: constant zero flow
marci@762
   109
//     ///- \c GEN_FLOW: any flow, i.e. the sum of the in-flows equals to
marci@762
   110
//     ///the sum of the out-flows in every node except the \e source and
marci@762
   111
//     ///the \e target.
marci@762
   112
//     ///- \c PRE_FLOW: any preflow, i.e. the sum of the in-flows is at 
marci@762
   113
//     ///least the sum of the out-flows in every node except the \e source.
marci@762
   114
//     ///- \c NO_FLOW: indicates an unspecified edge map. \ref flow will be 
marci@762
   115
//     ///set to the constant zero flow in the beginning of the algorithm in this case.
marci@762
   116
//     enum FlowEnum{
marci@762
   117
//       ZERO_FLOW,
marci@762
   118
//       GEN_FLOW,
marci@762
   119
//       PRE_FLOW,
marci@762
   120
//       NO_FLOW
marci@762
   121
//     };
marci@762
   122
marci@762
   123
//     enum StatusEnum {
marci@762
   124
//       AFTER_NOTHING,
marci@762
   125
//       AFTER_AUGMENTING,
marci@762
   126
//       AFTER_FAST_AUGMENTING, 
marci@762
   127
//       AFTER_PRE_FLOW_PHASE_1,      
marci@762
   128
//       AFTER_PRE_FLOW_PHASE_2
marci@762
   129
//     };
marci@762
   130
marci@762
   131
//     /// Don not needle this flag only if necessary.
marci@762
   132
//     StatusEnum status;
marci@762
   133
// //     int number_of_augmentations;
marci@762
   134
marci@762
   135
marci@762
   136
// //     template<typename IntMap>
marci@762
   137
// //     class TrickyReachedMap {
marci@762
   138
// //     protected:
marci@762
   139
// //       IntMap* map;
marci@762
   140
// //       int* number_of_augmentations;
marci@762
   141
// //     public:
marci@762
   142
// //       TrickyReachedMap(IntMap& _map, int& _number_of_augmentations) : 
marci@762
   143
// // 	map(&_map), number_of_augmentations(&_number_of_augmentations) { }
marci@762
   144
// //       void set(const Node& n, bool b) {
marci@762
   145
// // 	if (b)
marci@762
   146
// // 	  map->set(n, *number_of_augmentations);
marci@762
   147
// // 	else 
marci@762
   148
// // 	  map->set(n, *number_of_augmentations-1);
marci@762
   149
// //       }
marci@762
   150
// //       bool operator[](const Node& n) const { 
marci@762
   151
// // 	return (*map)[n]==*number_of_augmentations; 
marci@762
   152
// //       }
marci@762
   153
// //     };
marci@762
   154
    
marci@762
   155
//     MaxFlow(const Graph& _G, Node _s, Node _t, const CapMap& _capacity,
marci@762
   156
// 	    FlowMap& _flow) :
marci@762
   157
//       g(&_G), s(_s), t(_t), capacity(&_capacity),
marci@762
   158
//       flow(&_flow), n(_G.nodeNum()), level(_G), excess(_G,0), 
marci@762
   159
//       status(AFTER_NOTHING) { }
marci@762
   160
marci@762
   161
//     ///Runs a maximum flow algorithm.
marci@762
   162
marci@762
   163
//     ///Runs a preflow algorithm, which is the fastest maximum flow
marci@762
   164
//     ///algorithm up-to-date. The default for \c fe is ZERO_FLOW.
marci@762
   165
//     ///\pre The starting flow must be
marci@762
   166
//     /// - a constant zero flow if \c fe is \c ZERO_FLOW,
marci@762
   167
//     /// - an arbitary flow if \c fe is \c GEN_FLOW,
marci@762
   168
//     /// - an arbitary preflow if \c fe is \c PRE_FLOW,
marci@762
   169
//     /// - any map if \c fe is NO_FLOW.
marci@762
   170
//     void run(FlowEnum fe=ZERO_FLOW) {
marci@762
   171
//       preflow(fe);
marci@762
   172
//     }
marci@762
   173
marci@762
   174
                                                                              
marci@762
   175
//     ///Runs a preflow algorithm.  
marci@762
   176
marci@762
   177
//     ///Runs a preflow algorithm. The preflow algorithms provide the
marci@762
   178
//     ///fastest way to compute a maximum flow in a directed graph.
marci@762
   179
//     ///\pre The starting flow must be
marci@762
   180
//     /// - a constant zero flow if \c fe is \c ZERO_FLOW,
marci@762
   181
//     /// - an arbitary flow if \c fe is \c GEN_FLOW,
marci@762
   182
//     /// - an arbitary preflow if \c fe is \c PRE_FLOW,
marci@762
   183
//     /// - any map if \c fe is NO_FLOW.
marci@762
   184
//     void preflow(FlowEnum fe) {
marci@762
   185
//       preflowPhase1(fe);
marci@762
   186
//       preflowPhase2();
marci@762
   187
//     }
marci@762
   188
//     // Heuristics:
marci@762
   189
//     //   2 phase
marci@762
   190
//     //   gap
marci@762
   191
//     //   list 'level_list' on the nodes on level i implemented by hand
marci@762
   192
//     //   stack 'active' on the active nodes on level i                                                                                    
marci@762
   193
//     //   runs heuristic 'highest label' for H1*n relabels
marci@762
   194
//     //   runs heuristic 'bound decrease' for H0*n relabels, starts with 'highest label'
marci@762
   195
//     //   Parameters H0 and H1 are initialized to 20 and 1.
marci@762
   196
marci@762
   197
//     ///Runs the first phase of the preflow algorithm.
marci@762
   198
marci@762
   199
//     ///The preflow algorithm consists of two phases, this method runs the
marci@762
   200
//     ///first phase. After the first phase the maximum flow value and a
marci@762
   201
//     ///minimum value cut can already be computed, though a maximum flow
marci@762
   202
//     ///is net yet obtained. So after calling this method \ref flowValue
marci@762
   203
//     ///and \ref actMinCut gives proper results.
marci@762
   204
//     ///\warning: \ref minCut, \ref minMinCut and \ref maxMinCut do not
marci@762
   205
//     ///give minimum value cuts unless calling \ref preflowPhase2.
marci@762
   206
//     ///\pre The starting flow must be
marci@762
   207
//     /// - a constant zero flow if \c fe is \c ZERO_FLOW,
marci@762
   208
//     /// - an arbitary flow if \c fe is \c GEN_FLOW,
marci@762
   209
//     /// - an arbitary preflow if \c fe is \c PRE_FLOW,
marci@762
   210
//     /// - any map if \c fe is NO_FLOW.
marci@762
   211
//     void preflowPhase1(FlowEnum fe);
marci@762
   212
marci@762
   213
//     ///Runs the second phase of the preflow algorithm.
marci@762
   214
marci@762
   215
//     ///The preflow algorithm consists of two phases, this method runs
marci@762
   216
//     ///the second phase. After calling \ref preflowPhase1 and then
marci@762
   217
//     ///\ref preflowPhase2 the methods \ref flowValue, \ref minCut,
marci@762
   218
//     ///\ref minMinCut and \ref maxMinCut give proper results.
marci@762
   219
//     ///\pre \ref preflowPhase1 must be called before.
marci@762
   220
//     void preflowPhase2();
marci@762
   221
marci@762
   222
//     /// Returns the maximum value of a flow.
marci@762
   223
marci@762
   224
//     /// Returns the maximum value of a flow, by counting the 
marci@762
   225
//     /// over-flow of the target node \ref t.
marci@762
   226
//     /// It can be called already after running \ref preflowPhase1.
marci@762
   227
//     Num flowValue() const {
marci@762
   228
//       Num a=0;
marci@762
   229
//       FOR_EACH_INC_LOC(InEdgeIt, e, *g, t) a+=(*flow)[e];
marci@762
   230
//       FOR_EACH_INC_LOC(OutEdgeIt, e, *g, t) a-=(*flow)[e];
marci@762
   231
//       return a;
marci@762
   232
//       //marci figyu: excess[t] epp ezt adja preflow 1. fazisa utan   
marci@762
   233
//     }
marci@762
   234
marci@762
   235
//     ///Returns a minimum value cut after calling \ref preflowPhase1.
marci@762
   236
marci@762
   237
//     ///After the first phase of the preflow algorithm the maximum flow
marci@762
   238
//     ///value and a minimum value cut can already be computed. This
marci@762
   239
//     ///method can be called after running \ref preflowPhase1 for
marci@762
   240
//     ///obtaining a minimum value cut.
marci@762
   241
//     /// \warning Gives proper result only right after calling \ref
marci@762
   242
//     /// preflowPhase1.
marci@762
   243
//     /// \todo We have to make some status variable which shows the
marci@762
   244
//     /// actual state
marci@762
   245
//     /// of the class. This enables us to determine which methods are valid
marci@762
   246
//     /// for MinCut computation
marci@762
   247
//     template<typename _CutMap>
marci@762
   248
//     void actMinCut(_CutMap& M) const {
marci@762
   249
//       NodeIt v;
marci@762
   250
//       switch (status) {
marci@762
   251
//       case AFTER_PRE_FLOW_PHASE_1:
marci@762
   252
// 	for(g->first(v); g->valid(v); g->next(v)) {
marci@762
   253
// 	  if (level[v] < n) {
marci@762
   254
// 	    M.set(v, false);
marci@762
   255
// 	  } else {
marci@762
   256
// 	    M.set(v, true);
marci@762
   257
// 	  }
marci@762
   258
// 	}
marci@762
   259
// 	break;
marci@762
   260
//       case AFTER_PRE_FLOW_PHASE_2:
marci@762
   261
//       case AFTER_NOTHING:
marci@762
   262
//       case AFTER_AUGMENTING:
marci@762
   263
//       case AFTER_FAST_AUGMENTING:
marci@762
   264
// 	minMinCut(M);
marci@762
   265
// 	break;
marci@762
   266
// //       case AFTER_AUGMENTING:
marci@762
   267
// // 	for(g->first(v); g->valid(v); g->next(v)) {
marci@762
   268
// // 	  if (level[v]) {
marci@762
   269
// // 	    M.set(v, true);
marci@762
   270
// // 	  } else {
marci@762
   271
// // 	    M.set(v, false);
marci@762
   272
// // 	  }
marci@762
   273
// // 	}
marci@762
   274
// // 	break;
marci@762
   275
// //       case AFTER_FAST_AUGMENTING:
marci@762
   276
// // 	for(g->first(v); g->valid(v); g->next(v)) {
marci@762
   277
// // 	  if (level[v]==number_of_augmentations) {
marci@762
   278
// // 	    M.set(v, true);
marci@762
   279
// // 	  } else {
marci@762
   280
// // 	    M.set(v, false);
marci@762
   281
// // 	  }
marci@762
   282
// // 	}
marci@762
   283
// // 	break;
marci@762
   284
//       }
marci@762
   285
//     }
marci@762
   286
marci@762
   287
//     ///Returns the inclusionwise minimum of the minimum value cuts.
marci@762
   288
marci@762
   289
//     ///Sets \c M to the characteristic vector of the minimum value cut
marci@762
   290
//     ///which is inclusionwise minimum. It is computed by processing
marci@762
   291
//     ///a bfs from the source node \c s in the residual graph.
marci@762
   292
//     ///\pre M should be a node map of bools initialized to false.
marci@762
   293
//     ///\pre \c flow must be a maximum flow.
marci@762
   294
//     template<typename _CutMap>
marci@762
   295
//     void minMinCut(_CutMap& M) const {
marci@762
   296
//       std::queue<Node> queue;
marci@762
   297
marci@762
   298
//       M.set(s,true);
marci@762
   299
//       queue.push(s);
marci@762
   300
marci@762
   301
//       while (!queue.empty()) {
marci@762
   302
//         Node w=queue.front();
marci@762
   303
// 	queue.pop();
marci@762
   304
marci@762
   305
// 	OutEdgeIt e;
marci@762
   306
// 	for(g->first(e,w) ; g->valid(e); g->next(e)) {
marci@762
   307
// 	  Node v=g->head(e);
marci@762
   308
// 	  if (!M[v] && (*flow)[e] < (*capacity)[e] ) {
marci@762
   309
// 	    queue.push(v);
marci@762
   310
// 	    M.set(v, true);
marci@762
   311
// 	  }
marci@762
   312
// 	}
marci@762
   313
marci@762
   314
// 	InEdgeIt f;
marci@762
   315
// 	for(g->first(f,w) ; g->valid(f); g->next(f)) {
marci@762
   316
// 	  Node v=g->tail(f);
marci@762
   317
// 	  if (!M[v] && (*flow)[f] > 0 ) {
marci@762
   318
// 	    queue.push(v);
marci@762
   319
// 	    M.set(v, true);
marci@762
   320
// 	  }
marci@762
   321
// 	}
marci@762
   322
//       }
marci@762
   323
//     }
marci@762
   324
marci@762
   325
//     ///Returns the inclusionwise maximum of the minimum value cuts.
marci@762
   326
marci@762
   327
//     ///Sets \c M to the characteristic vector of the minimum value cut
marci@762
   328
//     ///which is inclusionwise maximum. It is computed by processing a
marci@762
   329
//     ///backward bfs from the target node \c t in the residual graph.
marci@762
   330
//     ///\pre M should be a node map of bools initialized to false.
marci@762
   331
//     ///\pre \c flow must be a maximum flow. 
marci@762
   332
//     template<typename _CutMap>
marci@762
   333
//     void maxMinCut(_CutMap& M) const {
marci@762
   334
marci@762
   335
//       NodeIt v;
marci@762
   336
//       for(g->first(v) ; g->valid(v); g->next(v)) {
marci@762
   337
// 	M.set(v, true);
marci@762
   338
//       }
marci@762
   339
marci@762
   340
//       std::queue<Node> queue;
marci@762
   341
marci@762
   342
//       M.set(t,false);
marci@762
   343
//       queue.push(t);
marci@762
   344
marci@762
   345
//       while (!queue.empty()) {
marci@762
   346
//         Node w=queue.front();
marci@762
   347
// 	queue.pop();
marci@762
   348
marci@762
   349
// 	InEdgeIt e;
marci@762
   350
// 	for(g->first(e,w) ; g->valid(e); g->next(e)) {
marci@762
   351
// 	  Node v=g->tail(e);
marci@762
   352
// 	  if (M[v] && (*flow)[e] < (*capacity)[e] ) {
marci@762
   353
// 	    queue.push(v);
marci@762
   354
// 	    M.set(v, false);
marci@762
   355
// 	  }
marci@762
   356
// 	}
marci@762
   357
marci@762
   358
// 	OutEdgeIt f;
marci@762
   359
// 	for(g->first(f,w) ; g->valid(f); g->next(f)) {
marci@762
   360
// 	  Node v=g->head(f);
marci@762
   361
// 	  if (M[v] && (*flow)[f] > 0 ) {
marci@762
   362
// 	    queue.push(v);
marci@762
   363
// 	    M.set(v, false);
marci@762
   364
// 	  }
marci@762
   365
// 	}
marci@762
   366
//       }
marci@762
   367
//     }
marci@762
   368
marci@762
   369
//     ///Returns a minimum value cut.
marci@762
   370
marci@762
   371
//     ///Sets \c M to the characteristic vector of a minimum value cut.
marci@762
   372
//     ///\pre M should be a node map of bools initialized to false.
marci@762
   373
//     ///\pre \c flow must be a maximum flow.    
marci@762
   374
//     template<typename CutMap>
marci@762
   375
//     void minCut(CutMap& M) const { minMinCut(M); }
marci@762
   376
marci@762
   377
//     ///Resets the source node to \c _s.
marci@762
   378
marci@762
   379
//     ///Resets the source node to \c _s.
marci@762
   380
//     /// 
marci@762
   381
//     void resetSource(Node _s) { s=_s; status=AFTER_NOTHING; }
marci@762
   382
marci@762
   383
//     ///Resets the target node to \c _t.
marci@762
   384
marci@762
   385
//     ///Resets the target node to \c _t.
marci@762
   386
//     ///
marci@762
   387
//     void resetTarget(Node _t) { t=_t; status=AFTER_NOTHING; }
marci@762
   388
marci@762
   389
//     /// Resets the edge map of the capacities to _cap.
marci@762
   390
marci@762
   391
//     /// Resets the edge map of the capacities to _cap.
marci@762
   392
//     /// 
marci@762
   393
//     void resetCap(const CapMap& _cap) { capacity=&_cap; status=AFTER_NOTHING; }
marci@762
   394
marci@762
   395
//     /// Resets the edge map of the flows to _flow.
marci@762
   396
marci@762
   397
//     /// Resets the edge map of the flows to _flow.
marci@762
   398
//     /// 
marci@762
   399
//     void resetFlow(FlowMap& _flow) { flow=&_flow; status=AFTER_NOTHING; }
marci@762
   400
marci@762
   401
marci@762
   402
//   private:
marci@762
   403
marci@762
   404
//     int push(Node w, VecStack& active) {
marci@762
   405
marci@762
   406
//       int lev=level[w];
marci@762
   407
//       Num exc=excess[w];
marci@762
   408
//       int newlevel=n;       //bound on the next level of w
marci@762
   409
marci@762
   410
//       OutEdgeIt e;
marci@762
   411
//       for(g->first(e,w); g->valid(e); g->next(e)) {
marci@762
   412
marci@762
   413
// 	if ( (*flow)[e] >= (*capacity)[e] ) continue;
marci@762
   414
// 	Node v=g->head(e);
marci@762
   415
marci@762
   416
// 	if( lev > level[v] ) { //Push is allowed now
marci@762
   417
marci@762
   418
// 	  if ( excess[v]<=0 && v!=t && v!=s ) {
marci@762
   419
// 	    int lev_v=level[v];
marci@762
   420
// 	    active[lev_v].push(v);
marci@762
   421
// 	  }
marci@762
   422
marci@762
   423
// 	  Num cap=(*capacity)[e];
marci@762
   424
// 	  Num flo=(*flow)[e];
marci@762
   425
// 	  Num remcap=cap-flo;
marci@762
   426
marci@762
   427
// 	  if ( remcap >= exc ) { //A nonsaturating push.
marci@762
   428
marci@762
   429
// 	    flow->set(e, flo+exc);
marci@762
   430
// 	    excess.set(v, excess[v]+exc);
marci@762
   431
// 	    exc=0;
marci@762
   432
// 	    break;
marci@762
   433
marci@762
   434
// 	  } else { //A saturating push.
marci@762
   435
// 	    flow->set(e, cap);
marci@762
   436
// 	    excess.set(v, excess[v]+remcap);
marci@762
   437
// 	    exc-=remcap;
marci@762
   438
// 	  }
marci@762
   439
// 	} else if ( newlevel > level[v] ) newlevel = level[v];
marci@762
   440
//       } //for out edges wv
marci@762
   441
marci@762
   442
//       if ( exc > 0 ) {
marci@762
   443
// 	InEdgeIt e;
marci@762
   444
// 	for(g->first(e,w); g->valid(e); g->next(e)) {
marci@762
   445
marci@762
   446
// 	  if( (*flow)[e] <= 0 ) continue;
marci@762
   447
// 	  Node v=g->tail(e);
marci@762
   448
marci@762
   449
// 	  if( lev > level[v] ) { //Push is allowed now
marci@762
   450
marci@762
   451
// 	    if ( excess[v]<=0 && v!=t && v!=s ) {
marci@762
   452
// 	      int lev_v=level[v];
marci@762
   453
// 	      active[lev_v].push(v);
marci@762
   454
// 	    }
marci@762
   455
marci@762
   456
// 	    Num flo=(*flow)[e];
marci@762
   457
marci@762
   458
// 	    if ( flo >= exc ) { //A nonsaturating push.
marci@762
   459
marci@762
   460
// 	      flow->set(e, flo-exc);
marci@762
   461
// 	      excess.set(v, excess[v]+exc);
marci@762
   462
// 	      exc=0;
marci@762
   463
// 	      break;
marci@762
   464
// 	    } else {  //A saturating push.
marci@762
   465
marci@762
   466
// 	      excess.set(v, excess[v]+flo);
marci@762
   467
// 	      exc-=flo;
marci@762
   468
// 	      flow->set(e,0);
marci@762
   469
// 	    }
marci@762
   470
// 	  } else if ( newlevel > level[v] ) newlevel = level[v];
marci@762
   471
// 	} //for in edges vw
marci@762
   472
marci@762
   473
//       } // if w still has excess after the out edge for cycle
marci@762
   474
marci@762
   475
//       excess.set(w, exc);
marci@762
   476
marci@762
   477
//       return newlevel;
marci@762
   478
//     }
marci@762
   479
marci@762
   480
marci@762
   481
//     void preflowPreproc(FlowEnum fe, VecStack& active,
marci@762
   482
// 			VecNode& level_list, NNMap& left, NNMap& right)
marci@762
   483
//     {
marci@762
   484
//       std::queue<Node> bfs_queue;
marci@762
   485
marci@762
   486
//       switch (fe) {
marci@762
   487
//       case NO_FLOW:   //flow is already set to const zero in this case
marci@762
   488
//       case ZERO_FLOW:
marci@762
   489
// 	{
marci@762
   490
// 	  //Reverse_bfs from t, to find the starting level.
marci@762
   491
// 	  level.set(t,0);
marci@762
   492
// 	  bfs_queue.push(t);
marci@762
   493
marci@762
   494
// 	  while (!bfs_queue.empty()) {
marci@762
   495
marci@762
   496
// 	    Node v=bfs_queue.front();
marci@762
   497
// 	    bfs_queue.pop();
marci@762
   498
// 	    int l=level[v]+1;
marci@762
   499
marci@762
   500
// 	    InEdgeIt e;
marci@762
   501
// 	    for(g->first(e,v); g->valid(e); g->next(e)) {
marci@762
   502
// 	      Node w=g->tail(e);
marci@762
   503
// 	      if ( level[w] == n && w != s ) {
marci@762
   504
// 		bfs_queue.push(w);
marci@762
   505
// 		Node first=level_list[l];
marci@762
   506
// 		if ( g->valid(first) ) left.set(first,w);
marci@762
   507
// 		right.set(w,first);
marci@762
   508
// 		level_list[l]=w;
marci@762
   509
// 		level.set(w, l);
marci@762
   510
// 	      }
marci@762
   511
// 	    }
marci@762
   512
// 	  }
marci@762
   513
marci@762
   514
// 	  //the starting flow
marci@762
   515
// 	  OutEdgeIt e;
marci@762
   516
// 	  for(g->first(e,s); g->valid(e); g->next(e))
marci@762
   517
// 	    {
marci@762
   518
// 	      Num c=(*capacity)[e];
marci@762
   519
// 	      if ( c <= 0 ) continue;
marci@762
   520
// 	      Node w=g->head(e);
marci@762
   521
// 	      if ( level[w] < n ) {
marci@762
   522
// 		if ( excess[w] <= 0 && w!=t ) active[level[w]].push(w);
marci@762
   523
// 		flow->set(e, c);
marci@762
   524
// 		excess.set(w, excess[w]+c);
marci@762
   525
// 	      }
marci@762
   526
// 	    }
marci@762
   527
// 	  break;
marci@762
   528
// 	}
marci@762
   529
marci@762
   530
//       case GEN_FLOW:
marci@762
   531
//       case PRE_FLOW:
marci@762
   532
// 	{
marci@762
   533
// 	  //Reverse_bfs from t in the residual graph,
marci@762
   534
// 	  //to find the starting level.
marci@762
   535
// 	  level.set(t,0);
marci@762
   536
// 	  bfs_queue.push(t);
marci@762
   537
marci@762
   538
// 	  while (!bfs_queue.empty()) {
marci@762
   539
marci@762
   540
// 	    Node v=bfs_queue.front();
marci@762
   541
// 	    bfs_queue.pop();
marci@762
   542
// 	    int l=level[v]+1;
marci@762
   543
marci@762
   544
// 	    InEdgeIt e;
marci@762
   545
// 	    for(g->first(e,v); g->valid(e); g->next(e)) {
marci@762
   546
// 	      if ( (*capacity)[e] <= (*flow)[e] ) continue;
marci@762
   547
// 	      Node w=g->tail(e);
marci@762
   548
// 	      if ( level[w] == n && w != s ) {
marci@762
   549
// 		bfs_queue.push(w);
marci@762
   550
// 		Node first=level_list[l];
marci@762
   551
// 		if ( g->valid(first) ) left.set(first,w);
marci@762
   552
// 		right.set(w,first);
marci@762
   553
// 		level_list[l]=w;
marci@762
   554
// 		level.set(w, l);
marci@762
   555
// 	      }
marci@762
   556
// 	    }
marci@762
   557
marci@762
   558
// 	    OutEdgeIt f;
marci@762
   559
// 	    for(g->first(f,v); g->valid(f); g->next(f)) {
marci@762
   560
// 	      if ( 0 >= (*flow)[f] ) continue;
marci@762
   561
// 	      Node w=g->head(f);
marci@762
   562
// 	      if ( level[w] == n && w != s ) {
marci@762
   563
// 		bfs_queue.push(w);
marci@762
   564
// 		Node first=level_list[l];
marci@762
   565
// 		if ( g->valid(first) ) left.set(first,w);
marci@762
   566
// 		right.set(w,first);
marci@762
   567
// 		level_list[l]=w;
marci@762
   568
// 		level.set(w, l);
marci@762
   569
// 	      }
marci@762
   570
// 	    }
marci@762
   571
// 	  }
marci@762
   572
marci@762
   573
marci@762
   574
// 	  //the starting flow
marci@762
   575
// 	  OutEdgeIt e;
marci@762
   576
// 	  for(g->first(e,s); g->valid(e); g->next(e))
marci@762
   577
// 	    {
marci@762
   578
// 	      Num rem=(*capacity)[e]-(*flow)[e];
marci@762
   579
// 	      if ( rem <= 0 ) continue;
marci@762
   580
// 	      Node w=g->head(e);
marci@762
   581
// 	      if ( level[w] < n ) {
marci@762
   582
// 		if ( excess[w] <= 0 && w!=t ) active[level[w]].push(w);
marci@762
   583
// 		flow->set(e, (*capacity)[e]);
marci@762
   584
// 		excess.set(w, excess[w]+rem);
marci@762
   585
// 	      }
marci@762
   586
// 	    }
marci@762
   587
marci@762
   588
// 	  InEdgeIt f;
marci@762
   589
// 	  for(g->first(f,s); g->valid(f); g->next(f))
marci@762
   590
// 	    {
marci@762
   591
// 	      if ( (*flow)[f] <= 0 ) continue;
marci@762
   592
// 	      Node w=g->tail(f);
marci@762
   593
// 	      if ( level[w] < n ) {
marci@762
   594
// 		if ( excess[w] <= 0 && w!=t ) active[level[w]].push(w);
marci@762
   595
// 		excess.set(w, excess[w]+(*flow)[f]);
marci@762
   596
// 		flow->set(f, 0);
marci@762
   597
// 	      }
marci@762
   598
// 	    }
marci@762
   599
// 	  break;
marci@762
   600
// 	} //case PRE_FLOW
marci@762
   601
//       }
marci@762
   602
//     } //preflowPreproc
marci@762
   603
marci@762
   604
marci@762
   605
marci@762
   606
//     void relabel(Node w, int newlevel, VecStack& active,
marci@762
   607
// 		 VecNode& level_list, NNMap& left,
marci@762
   608
// 		 NNMap& right, int& b, int& k, bool what_heur )
marci@762
   609
//     {
marci@762
   610
marci@762
   611
//       //FIXME jacint: ez mitol num
marci@762
   612
// //      Num lev=level[w];
marci@762
   613
//       int lev=level[w];
marci@762
   614
marci@762
   615
//       Node right_n=right[w];
marci@762
   616
//       Node left_n=left[w];
marci@762
   617
marci@762
   618
//       //unlacing starts
marci@762
   619
//       if ( g->valid(right_n) ) {
marci@762
   620
// 	if ( g->valid(left_n) ) {
marci@762
   621
// 	  right.set(left_n, right_n);
marci@762
   622
// 	  left.set(right_n, left_n);
marci@762
   623
// 	} else {
marci@762
   624
// 	  level_list[lev]=right_n;
marci@762
   625
// 	  left.set(right_n, INVALID);
marci@762
   626
// 	}
marci@762
   627
//       } else {
marci@762
   628
// 	if ( g->valid(left_n) ) {
marci@762
   629
// 	  right.set(left_n, INVALID);
marci@762
   630
// 	} else {
marci@762
   631
// 	  level_list[lev]=INVALID;
marci@762
   632
// 	}
marci@762
   633
//       }
marci@762
   634
//       //unlacing ends
marci@762
   635
marci@762
   636
//       if ( !g->valid(level_list[lev]) ) {
marci@762
   637
marci@762
   638
// 	//gapping starts
marci@762
   639
// 	for (int i=lev; i!=k ; ) {
marci@762
   640
// 	  Node v=level_list[++i];
marci@762
   641
// 	  while ( g->valid(v) ) {
marci@762
   642
// 	    level.set(v,n);
marci@762
   643
// 	    v=right[v];
marci@762
   644
// 	  }
marci@762
   645
// 	  level_list[i]=INVALID;
marci@762
   646
// 	  if ( !what_heur ) {
marci@762
   647
// 	    while ( !active[i].empty() ) {
marci@762
   648
// 	      active[i].pop();    //FIXME: ezt szebben kene
marci@762
   649
// 	    }
marci@762
   650
// 	  }
marci@762
   651
// 	}
marci@762
   652
marci@762
   653
// 	level.set(w,n);
marci@762
   654
// 	b=lev-1;
marci@762
   655
// 	k=b;
marci@762
   656
// 	//gapping ends
marci@762
   657
marci@762
   658
//       } else {
marci@762
   659
marci@762
   660
// 	if ( newlevel == n ) level.set(w,n);
marci@762
   661
// 	else {
marci@762
   662
// 	  level.set(w,++newlevel);
marci@762
   663
// 	  active[newlevel].push(w);
marci@762
   664
// 	  if ( what_heur ) b=newlevel;
marci@762
   665
// 	  if ( k < newlevel ) ++k;      //now k=newlevel
marci@762
   666
// 	  Node first=level_list[newlevel];
marci@762
   667
// 	  if ( g->valid(first) ) left.set(first,w);
marci@762
   668
// 	  right.set(w,first);
marci@762
   669
// 	  left.set(w,INVALID);
marci@762
   670
// 	  level_list[newlevel]=w;
marci@762
   671
// 	}
marci@762
   672
//       }
marci@762
   673
marci@762
   674
//     } //relabel
marci@762
   675
marci@762
   676
//   };
marci@762
   677
marci@762
   678
marci@762
   679
marci@762
   680
//   template <typename Graph, typename Num, typename CapMap, typename FlowMap>
marci@762
   681
//   void MaxFlow<Graph, Num, CapMap, FlowMap>::preflowPhase1(FlowEnum fe)
marci@762
   682
//   {
marci@762
   683
marci@762
   684
//     int heur0=(int)(H0*n);  //time while running 'bound decrease'
marci@762
   685
//     int heur1=(int)(H1*n);  //time while running 'highest label'
marci@762
   686
//     int heur=heur1;         //starting time interval (#of relabels)
marci@762
   687
//     int numrelabel=0;
marci@762
   688
marci@762
   689
//     bool what_heur=1;
marci@762
   690
//     //It is 0 in case 'bound decrease' and 1 in case 'highest label'
marci@762
   691
marci@762
   692
//     bool end=false;
marci@762
   693
//     //Needed for 'bound decrease', true means no active nodes are above bound
marci@762
   694
//     //b.
marci@762
   695
marci@762
   696
//     int k=n-2;  //bound on the highest level under n containing a node
marci@762
   697
//     int b=k;    //bound on the highest level under n of an active node
marci@762
   698
marci@762
   699
//     VecStack active(n);
marci@762
   700
marci@762
   701
//     NNMap left(*g, INVALID);
marci@762
   702
//     NNMap right(*g, INVALID);
marci@762
   703
//     VecNode level_list(n,INVALID);
marci@762
   704
//     //List of the nodes in level i<n, set to n.
marci@762
   705
marci@762
   706
//     NodeIt v;
marci@762
   707
//     for(g->first(v); g->valid(v); g->next(v)) level.set(v,n);
marci@762
   708
//     //setting each node to level n
marci@762
   709
marci@762
   710
//     if ( fe == NO_FLOW ) {
marci@762
   711
//       EdgeIt e;
marci@762
   712
//       for(g->first(e); g->valid(e); g->next(e)) flow->set(e,0);
marci@762
   713
//     }
marci@762
   714
marci@762
   715
//     switch (fe) { //computing the excess
marci@762
   716
//     case PRE_FLOW:
marci@762
   717
//       {
marci@762
   718
// 	NodeIt v;
marci@762
   719
// 	for(g->first(v); g->valid(v); g->next(v)) {
marci@762
   720
// 	  Num exc=0;
marci@762
   721
marci@762
   722
// 	  InEdgeIt e;
marci@762
   723
// 	  for(g->first(e,v); g->valid(e); g->next(e)) exc+=(*flow)[e];
marci@762
   724
// 	  OutEdgeIt f;
marci@762
   725
// 	  for(g->first(f,v); g->valid(f); g->next(f)) exc-=(*flow)[f];
marci@762
   726
marci@762
   727
// 	  excess.set(v,exc);
marci@762
   728
marci@762
   729
// 	  //putting the active nodes into the stack
marci@762
   730
// 	  int lev=level[v];
marci@762
   731
// 	  if ( exc > 0 && lev < n && v != t ) active[lev].push(v);
marci@762
   732
// 	}
marci@762
   733
// 	break;
marci@762
   734
//       }
marci@762
   735
//     case GEN_FLOW:
marci@762
   736
//       {
marci@762
   737
// 	NodeIt v;
marci@762
   738
// 	for(g->first(v); g->valid(v); g->next(v)) excess.set(v,0);
marci@762
   739
marci@762
   740
// 	Num exc=0;
marci@762
   741
// 	InEdgeIt e;
marci@762
   742
// 	for(g->first(e,t); g->valid(e); g->next(e)) exc+=(*flow)[e];
marci@762
   743
// 	OutEdgeIt f;
marci@762
   744
// 	for(g->first(f,t); g->valid(f); g->next(f)) exc-=(*flow)[f];
marci@762
   745
// 	excess.set(t,exc);
marci@762
   746
// 	break;
marci@762
   747
//       }
marci@762
   748
//     case ZERO_FLOW:
marci@762
   749
//     case NO_FLOW:
marci@762
   750
//       {
marci@762
   751
// 	NodeIt v;
marci@762
   752
//         for(g->first(v); g->valid(v); g->next(v)) excess.set(v,0);
marci@762
   753
// 	break;
marci@762
   754
//       }
marci@762
   755
//     }
marci@762
   756
marci@762
   757
//     preflowPreproc(fe, active, level_list, left, right);
marci@762
   758
//     //End of preprocessing
marci@762
   759
marci@762
   760
marci@762
   761
//     //Push/relabel on the highest level active nodes.
marci@762
   762
//     while ( true ) {
marci@762
   763
//       if ( b == 0 ) {
marci@762
   764
// 	if ( !what_heur && !end && k > 0 ) {
marci@762
   765
// 	  b=k;
marci@762
   766
// 	  end=true;
marci@762
   767
// 	} else break;
marci@762
   768
//       }
marci@762
   769
marci@762
   770
//       if ( active[b].empty() ) --b;
marci@762
   771
//       else {
marci@762
   772
// 	end=false;
marci@762
   773
// 	Node w=active[b].top();
marci@762
   774
// 	active[b].pop();
marci@762
   775
// 	int newlevel=push(w,active);
marci@762
   776
// 	if ( excess[w] > 0 ) relabel(w, newlevel, active, level_list,
marci@762
   777
// 				     left, right, b, k, what_heur);
marci@762
   778
marci@762
   779
// 	++numrelabel;
marci@762
   780
// 	if ( numrelabel >= heur ) {
marci@762
   781
// 	  numrelabel=0;
marci@762
   782
// 	  if ( what_heur ) {
marci@762
   783
// 	    what_heur=0;
marci@762
   784
// 	    heur=heur0;
marci@762
   785
// 	    end=false;
marci@762
   786
// 	  } else {
marci@762
   787
// 	    what_heur=1;
marci@762
   788
// 	    heur=heur1;
marci@762
   789
// 	    b=k;
marci@762
   790
// 	  }
marci@762
   791
// 	}
marci@762
   792
//       }
marci@762
   793
//     }
marci@762
   794
marci@762
   795
//     status=AFTER_PRE_FLOW_PHASE_1;
marci@762
   796
//   }
marci@762
   797
marci@762
   798
marci@762
   799
marci@762
   800
//   template <typename Graph, typename Num, typename CapMap, typename FlowMap>
marci@762
   801
//   void MaxFlow<Graph, Num, CapMap, FlowMap>::preflowPhase2()
marci@762
   802
//   {
marci@762
   803
marci@762
   804
//     int k=n-2;  //bound on the highest level under n containing a node
marci@762
   805
//     int b=k;    //bound on the highest level under n of an active node
marci@762
   806
marci@762
   807
//     VecStack active(n);
marci@762
   808
//     level.set(s,0);
marci@762
   809
//     std::queue<Node> bfs_queue;
marci@762
   810
//     bfs_queue.push(s);
marci@762
   811
marci@762
   812
//     while (!bfs_queue.empty()) {
marci@762
   813
marci@762
   814
//       Node v=bfs_queue.front();
marci@762
   815
//       bfs_queue.pop();
marci@762
   816
//       int l=level[v]+1;
marci@762
   817
marci@762
   818
//       InEdgeIt e;
marci@762
   819
//       for(g->first(e,v); g->valid(e); g->next(e)) {
marci@762
   820
// 	if ( (*capacity)[e] <= (*flow)[e] ) continue;
marci@762
   821
// 	Node u=g->tail(e);
marci@762
   822
// 	if ( level[u] >= n ) {
marci@762
   823
// 	  bfs_queue.push(u);
marci@762
   824
// 	  level.set(u, l);
marci@762
   825
// 	  if ( excess[u] > 0 ) active[l].push(u);
marci@762
   826
// 	}
marci@762
   827
//       }
marci@762
   828
marci@762
   829
//       OutEdgeIt f;
marci@762
   830
//       for(g->first(f,v); g->valid(f); g->next(f)) {
marci@762
   831
// 	if ( 0 >= (*flow)[f] ) continue;
marci@762
   832
// 	Node u=g->head(f);
marci@762
   833
// 	if ( level[u] >= n ) {
marci@762
   834
// 	  bfs_queue.push(u);
marci@762
   835
// 	  level.set(u, l);
marci@762
   836
// 	  if ( excess[u] > 0 ) active[l].push(u);
marci@762
   837
// 	}
marci@762
   838
//       }
marci@762
   839
//     }
marci@762
   840
//     b=n-2;
marci@762
   841
marci@762
   842
//     while ( true ) {
marci@762
   843
marci@762
   844
//       if ( b == 0 ) break;
marci@762
   845
marci@762
   846
//       if ( active[b].empty() ) --b;
marci@762
   847
//       else {
marci@762
   848
// 	Node w=active[b].top();
marci@762
   849
// 	active[b].pop();
marci@762
   850
// 	int newlevel=push(w,active);
marci@762
   851
marci@762
   852
// 	//relabel
marci@762
   853
// 	if ( excess[w] > 0 ) {
marci@762
   854
// 	  level.set(w,++newlevel);
marci@762
   855
// 	  active[newlevel].push(w);
marci@762
   856
// 	  b=newlevel;
marci@762
   857
// 	}
marci@762
   858
//       }  // if stack[b] is nonempty
marci@762
   859
//     } // while(true)
marci@762
   860
marci@762
   861
//     status=AFTER_PRE_FLOW_PHASE_2;
marci@762
   862
//   }
marci@762
   863
marci@762
   864
marci@762
   865
  template <typename Graph, typename Num,
marci@762
   866
	    typename CapMap=typename Graph::template EdgeMap<Num>,
marci@762
   867
            typename FlowMap=typename Graph::template EdgeMap<Num> >
marci@762
   868
  class AugmentingFlow {
marci@762
   869
  protected:
marci@762
   870
    typedef typename Graph::Node Node;
marci@762
   871
    typedef typename Graph::NodeIt NodeIt;
marci@762
   872
    typedef typename Graph::EdgeIt EdgeIt;
marci@762
   873
    typedef typename Graph::OutEdgeIt OutEdgeIt;
marci@762
   874
    typedef typename Graph::InEdgeIt InEdgeIt;
marci@762
   875
marci@762
   876
//    typedef typename std::vector<std::stack<Node> > VecStack;
marci@762
   877
//    typedef typename Graph::template NodeMap<Node> NNMap;
marci@762
   878
//    typedef typename std::vector<Node> VecNode;
marci@762
   879
marci@762
   880
    const Graph* g;
marci@762
   881
    Node s;
marci@762
   882
    Node t;
marci@762
   883
    const CapMap* capacity;
marci@762
   884
    FlowMap* flow;
marci@762
   885
//    int n;      //the number of nodes of G
marci@762
   886
    typedef ResGraphWrapper<const Graph, Num, CapMap, FlowMap> ResGW;   
marci@762
   887
    //typedef ExpResGraphWrapper<const Graph, Num, CapMap, FlowMap> ResGW;
marci@762
   888
    typedef typename ResGW::OutEdgeIt ResGWOutEdgeIt;
marci@762
   889
    typedef typename ResGW::Edge ResGWEdge;
marci@762
   890
    //typedef typename ResGW::template NodeMap<bool> ReachedMap;
marci@762
   891
    typedef typename Graph::template NodeMap<int> ReachedMap;
marci@762
   892
marci@762
   893
marci@762
   894
    //level works as a bool map in augmenting path algorithms and is
marci@762
   895
    //used by bfs for storing reached information.  In preflow, it
marci@762
   896
    //shows the levels of nodes.     
marci@762
   897
    ReachedMap level;
marci@762
   898
marci@762
   899
    //excess is needed only in preflow
marci@762
   900
//    typename Graph::template NodeMap<Num> excess;
marci@762
   901
marci@762
   902
    //fixme    
marci@762
   903
//   protected:
marci@762
   904
    //     MaxFlow() { }
marci@762
   905
    //     void set(const Graph& _G, Node _s, Node _t, const CapMap& _capacity,
marci@762
   906
    // 	     FlowMap& _flow)
marci@762
   907
    //       {
marci@762
   908
    // 	g=&_G;
marci@762
   909
    // 	s=_s;
marci@762
   910
    // 	t=_t;
marci@762
   911
    // 	capacity=&_capacity;
marci@762
   912
    // 	flow=&_flow;
marci@762
   913
    // 	n=_G.nodeNum;
marci@762
   914
    // 	level.set (_G); //kellene vmi ilyesmi fv
marci@762
   915
    // 	excess(_G,0); //itt is
marci@762
   916
    //       }
marci@762
   917
marci@762
   918
    // constants used for heuristics
marci@762
   919
//    static const int H0=20;
marci@762
   920
//    static const int H1=1;
marci@762
   921
marci@762
   922
  public:
marci@762
   923
marci@762
   924
    ///Indicates the property of the starting flow.
marci@762
   925
marci@762
   926
    ///Indicates the property of the starting flow. The meanings are as follows:
marci@762
   927
    ///- \c ZERO_FLOW: constant zero flow
marci@762
   928
    ///- \c GEN_FLOW: any flow, i.e. the sum of the in-flows equals to
marci@762
   929
    ///the sum of the out-flows in every node except the \e source and
marci@762
   930
    ///the \e target.
marci@762
   931
    ///- \c PRE_FLOW: any preflow, i.e. the sum of the in-flows is at 
marci@762
   932
    ///least the sum of the out-flows in every node except the \e source.
marci@762
   933
    ///- \c NO_FLOW: indicates an unspecified edge map. \ref flow will be 
marci@762
   934
    ///set to the constant zero flow in the beginning of the algorithm in this case.
marci@762
   935
    enum FlowEnum{
marci@762
   936
      ZERO_FLOW,
marci@762
   937
      GEN_FLOW,
marci@762
   938
      PRE_FLOW,
marci@762
   939
      NO_FLOW
marci@762
   940
    };
marci@762
   941
marci@762
   942
    enum StatusEnum {
marci@762
   943
      AFTER_NOTHING,
marci@762
   944
      AFTER_AUGMENTING,
marci@762
   945
      AFTER_FAST_AUGMENTING, 
marci@762
   946
      AFTER_PRE_FLOW_PHASE_1,      
marci@762
   947
      AFTER_PRE_FLOW_PHASE_2
marci@762
   948
    };
marci@762
   949
marci@762
   950
    /// Don not needle this flag only if necessary.
marci@762
   951
    StatusEnum status;
marci@762
   952
    int number_of_augmentations;
marci@762
   953
marci@762
   954
marci@762
   955
    template<typename IntMap>
marci@762
   956
    class TrickyReachedMap {
marci@762
   957
    protected:
marci@762
   958
      IntMap* map;
marci@762
   959
      int* number_of_augmentations;
marci@762
   960
    public:
marci@762
   961
      TrickyReachedMap(IntMap& _map, int& _number_of_augmentations) : 
marci@762
   962
	map(&_map), number_of_augmentations(&_number_of_augmentations) { }
marci@762
   963
      void set(const Node& n, bool b) {
marci@762
   964
	if (b)
marci@762
   965
	  map->set(n, *number_of_augmentations);
marci@762
   966
	else 
marci@762
   967
	  map->set(n, *number_of_augmentations-1);
marci@762
   968
      }
marci@762
   969
      bool operator[](const Node& n) const { 
marci@762
   970
	return (*map)[n]==*number_of_augmentations; 
marci@762
   971
      }
marci@762
   972
    };
marci@762
   973
    
marci@762
   974
    AugmentingFlow(const Graph& _G, Node _s, Node _t, const CapMap& _capacity,
marci@762
   975
		   FlowMap& _flow) :
marci@762
   976
      g(&_G), s(_s), t(_t), capacity(&_capacity),
marci@762
   977
      flow(&_flow), //n(_G.nodeNum()), 
marci@762
   978
      level(_G), //excess(_G,0), 
marci@762
   979
      status(AFTER_NOTHING), number_of_augmentations(0) { }
marci@762
   980
marci@762
   981
    /// Starting from a flow, this method searches for an augmenting path
marci@762
   982
    /// according to the Edmonds-Karp algorithm
marci@762
   983
    /// and augments the flow on if any.
marci@762
   984
    /// The return value shows if the augmentation was succesful.
marci@762
   985
    bool augmentOnShortestPath();
marci@762
   986
    bool augmentOnShortestPath2();
marci@762
   987
marci@762
   988
    /// Starting from a flow, this method searches for an augmenting blocking
marci@762
   989
    /// flow according to Dinits' algorithm and augments the flow on if any.
marci@762
   990
    /// The blocking flow is computed in a physically constructed
marci@762
   991
    /// residual graph of type \c Mutablegraph.
marci@762
   992
    /// The return value show sif the augmentation was succesful.
marci@762
   993
    template<typename MutableGraph> bool augmentOnBlockingFlow();
marci@762
   994
marci@762
   995
    /// The same as \c augmentOnBlockingFlow<MutableGraph> but the
marci@762
   996
    /// residual graph is not constructed physically.
marci@762
   997
    /// The return value shows if the augmentation was succesful.
marci@762
   998
    bool augmentOnBlockingFlow2();
marci@762
   999
marci@762
  1000
    template<typename _CutMap>
marci@762
  1001
    void actMinCut(_CutMap& M) const {
marci@762
  1002
      NodeIt v;
marci@762
  1003
      switch (status) {
marci@762
  1004
	case AFTER_PRE_FLOW_PHASE_1:
marci@762
  1005
//	std::cout << "AFTER_PRE_FLOW_PHASE_1" << std::endl;
marci@762
  1006
// 	for(g->first(v); g->valid(v); g->next(v)) {
marci@762
  1007
// 	  if (level[v] < n) {
marci@762
  1008
// 	    M.set(v, false);
marci@762
  1009
// 	  } else {
marci@762
  1010
// 	    M.set(v, true);
marci@762
  1011
// 	  }
marci@762
  1012
// 	}
marci@762
  1013
	break;
marci@762
  1014
      case AFTER_PRE_FLOW_PHASE_2:
marci@762
  1015
//	std::cout << "AFTER_PRE_FLOW_PHASE_2" << std::endl;
marci@762
  1016
	break;
marci@762
  1017
      case AFTER_NOTHING:
marci@762
  1018
//	std::cout << "AFTER_NOTHING" << std::endl;
marci@762
  1019
	minMinCut(M);
marci@762
  1020
	break;
marci@762
  1021
      case AFTER_AUGMENTING:
marci@762
  1022
//	std::cout << "AFTER_AUGMENTING" << std::endl;
marci@762
  1023
	for(g->first(v); g->valid(v); g->next(v)) {
marci@762
  1024
	  if (level[v]) {
marci@762
  1025
	    M.set(v, true);
marci@762
  1026
	  } else {
marci@762
  1027
	    M.set(v, false);
marci@762
  1028
	  }
marci@762
  1029
	}
marci@762
  1030
	break;
marci@762
  1031
      case AFTER_FAST_AUGMENTING:
marci@762
  1032
//	std::cout << "AFTER_FAST_AUGMENTING" << std::endl;
marci@762
  1033
	for(g->first(v); g->valid(v); g->next(v)) {
marci@762
  1034
	  if (level[v]==number_of_augmentations) {
marci@762
  1035
	    M.set(v, true);
marci@762
  1036
	  } else {
marci@762
  1037
	    M.set(v, false);
marci@762
  1038
	  }
marci@762
  1039
	}
marci@762
  1040
	break;
marci@762
  1041
      }
marci@762
  1042
    }
marci@762
  1043
marci@762
  1044
    template<typename _CutMap>
marci@762
  1045
    void minMinCut(_CutMap& M) const {
marci@762
  1046
      std::queue<Node> queue;
marci@762
  1047
marci@762
  1048
      M.set(s,true);
marci@762
  1049
      queue.push(s);
marci@762
  1050
marci@762
  1051
      while (!queue.empty()) {
marci@762
  1052
        Node w=queue.front();
marci@762
  1053
	queue.pop();
marci@762
  1054
marci@762
  1055
	OutEdgeIt e;
marci@762
  1056
	for(g->first(e,w) ; g->valid(e); g->next(e)) {
marci@762
  1057
	  Node v=g->head(e);
marci@762
  1058
	  if (!M[v] && (*flow)[e] < (*capacity)[e] ) {
marci@762
  1059
	    queue.push(v);
marci@762
  1060
	    M.set(v, true);
marci@762
  1061
	  }
marci@762
  1062
	}
marci@762
  1063
marci@762
  1064
	InEdgeIt f;
marci@762
  1065
	for(g->first(f,w) ; g->valid(f); g->next(f)) {
marci@762
  1066
	  Node v=g->tail(f);
marci@762
  1067
	  if (!M[v] && (*flow)[f] > 0 ) {
marci@762
  1068
	    queue.push(v);
marci@762
  1069
	    M.set(v, true);
marci@762
  1070
	  }
marci@762
  1071
	}
marci@762
  1072
      }
marci@762
  1073
    }
marci@762
  1074
marci@762
  1075
    template<typename _CutMap>
marci@762
  1076
    void minMinCut2(_CutMap& M) const {
marci@762
  1077
      ResGW res_graph(*g, *capacity, *flow);
marci@762
  1078
      BfsIterator<ResGW, _CutMap> bfs(res_graph, M);
marci@762
  1079
      bfs.pushAndSetReached(s);
marci@762
  1080
      while (!bfs.finished()) ++bfs;
marci@762
  1081
    }
marci@762
  1082
marci@762
  1083
    Num flowValue() const {
marci@762
  1084
      Num a=0;
marci@762
  1085
      FOR_EACH_INC_LOC(InEdgeIt, e, *g, t) a+=(*flow)[e];
marci@762
  1086
      FOR_EACH_INC_LOC(OutEdgeIt, e, *g, t) a-=(*flow)[e];
marci@762
  1087
      return a;
marci@762
  1088
      //marci figyu: excess[t] epp ezt adja preflow 1. fazisa utan   
marci@762
  1089
    }
marci@762
  1090
marci@762
  1091
    template<typename MapGraphWrapper>
marci@762
  1092
    class DistanceMap {
marci@762
  1093
    protected:
marci@762
  1094
      const MapGraphWrapper* g;
marci@762
  1095
      typename MapGraphWrapper::template NodeMap<int> dist;
marci@762
  1096
    public:
marci@762
  1097
      DistanceMap(MapGraphWrapper& _g) : g(&_g), dist(*g, g->nodeNum()) { }
marci@762
  1098
      void set(const typename MapGraphWrapper::Node& n, int a) {
marci@762
  1099
	dist.set(n, a);
marci@762
  1100
      }
marci@762
  1101
      int operator[](const typename MapGraphWrapper::Node& n) const { 
marci@762
  1102
	return dist[n]; 
marci@762
  1103
      }
marci@762
  1104
      //       int get(const typename MapGraphWrapper::Node& n) const {
marci@762
  1105
      // 	return dist[n]; }
marci@762
  1106
      //       bool get(const typename MapGraphWrapper::Edge& e) const {
marci@762
  1107
      // 	return (dist.get(g->tail(e))<dist.get(g->head(e))); }
marci@762
  1108
      bool operator[](const typename MapGraphWrapper::Edge& e) const {
marci@762
  1109
	return (dist[g->tail(e)]<dist[g->head(e)]);
marci@762
  1110
      }
marci@762
  1111
    };
marci@762
  1112
marci@762
  1113
  };
marci@762
  1114
marci@762
  1115
marci@762
  1116
marci@762
  1117
  template <typename Graph, typename Num, typename CapMap, typename FlowMap>
marci@762
  1118
  bool AugmentingFlow<Graph, Num, CapMap, FlowMap>::augmentOnShortestPath()
marci@762
  1119
  {
marci@762
  1120
    ResGW res_graph(*g, *capacity, *flow);
marci@762
  1121
    bool _augment=false;
marci@762
  1122
marci@762
  1123
    //ReachedMap level(res_graph);
marci@762
  1124
    FOR_EACH_LOC(typename Graph::NodeIt, e, *g) level.set(e, 0);
marci@762
  1125
    BfsIterator<ResGW, ReachedMap> bfs(res_graph, level);
marci@762
  1126
    bfs.pushAndSetReached(s);
marci@762
  1127
marci@762
  1128
    typename ResGW::template NodeMap<ResGWEdge> pred(res_graph);
marci@762
  1129
    pred.set(s, INVALID);
marci@762
  1130
marci@762
  1131
    typename ResGW::template NodeMap<Num> free(res_graph);
marci@762
  1132
marci@762
  1133
    //searching for augmenting path
marci@762
  1134
    while ( !bfs.finished() ) {
marci@762
  1135
      ResGWOutEdgeIt e=bfs;
marci@762
  1136
      if (res_graph.valid(e) && bfs.isBNodeNewlyReached()) {
marci@762
  1137
	Node v=res_graph.tail(e);
marci@762
  1138
	Node w=res_graph.head(e);
marci@762
  1139
	pred.set(w, e);
marci@762
  1140
	if (res_graph.valid(pred[v])) {
marci@762
  1141
	  free.set(w, std::min(free[v], res_graph.resCap(e)));
marci@762
  1142
	} else {
marci@762
  1143
	  free.set(w, res_graph.resCap(e));
marci@762
  1144
	}
marci@762
  1145
	if (res_graph.head(e)==t) { _augment=true; break; }
marci@762
  1146
      }
marci@762
  1147
marci@762
  1148
      ++bfs;
marci@762
  1149
    } //end of searching augmenting path
marci@762
  1150
marci@762
  1151
    if (_augment) {
marci@762
  1152
      Node n=t;
marci@762
  1153
      Num augment_value=free[t];
marci@762
  1154
      while (res_graph.valid(pred[n])) {
marci@762
  1155
	ResGWEdge e=pred[n];
marci@762
  1156
	res_graph.augment(e, augment_value);
marci@762
  1157
	n=res_graph.tail(e);
marci@762
  1158
      }
marci@762
  1159
    }
marci@762
  1160
marci@762
  1161
    status=AFTER_AUGMENTING;
marci@762
  1162
    return _augment;
marci@762
  1163
  }
marci@762
  1164
marci@762
  1165
  template <typename Graph, typename Num, typename CapMap, typename FlowMap>
marci@762
  1166
  bool AugmentingFlow<Graph, Num, CapMap, FlowMap>::augmentOnShortestPath2()
marci@762
  1167
  {
marci@762
  1168
    ResGW res_graph(*g, *capacity, *flow);
marci@762
  1169
    bool _augment=false;
marci@762
  1170
marci@762
  1171
    if (status!=AFTER_FAST_AUGMENTING) {
marci@762
  1172
      FOR_EACH_LOC(typename Graph::NodeIt, e, *g) level.set(e, 0); 
marci@762
  1173
      number_of_augmentations=1;
marci@762
  1174
    } else {
marci@762
  1175
      ++number_of_augmentations;
marci@762
  1176
    }
marci@762
  1177
    TrickyReachedMap<ReachedMap> 
marci@762
  1178
      tricky_reached_map(level, number_of_augmentations);
marci@762
  1179
    //ReachedMap level(res_graph);
marci@762
  1180
//    FOR_EACH_LOC(typename Graph::NodeIt, e, *g) level.set(e, 0);
marci@762
  1181
    BfsIterator<ResGW, TrickyReachedMap<ReachedMap> > 
marci@762
  1182
      bfs(res_graph, tricky_reached_map);
marci@762
  1183
    bfs.pushAndSetReached(s);
marci@762
  1184
marci@762
  1185
    typename ResGW::template NodeMap<ResGWEdge> pred(res_graph);
marci@762
  1186
    pred.set(s, INVALID);
marci@762
  1187
marci@762
  1188
    typename ResGW::template NodeMap<Num> free(res_graph);
marci@762
  1189
marci@762
  1190
    //searching for augmenting path
marci@762
  1191
    while ( !bfs.finished() ) {
marci@762
  1192
      ResGWOutEdgeIt e=bfs;
marci@762
  1193
      if (res_graph.valid(e) && bfs.isBNodeNewlyReached()) {
marci@762
  1194
	Node v=res_graph.tail(e);
marci@762
  1195
	Node w=res_graph.head(e);
marci@762
  1196
	pred.set(w, e);
marci@762
  1197
	if (res_graph.valid(pred[v])) {
marci@762
  1198
	  free.set(w, std::min(free[v], res_graph.resCap(e)));
marci@762
  1199
	} else {
marci@762
  1200
	  free.set(w, res_graph.resCap(e));
marci@762
  1201
	}
marci@762
  1202
	if (res_graph.head(e)==t) { _augment=true; break; }
marci@762
  1203
      }
marci@762
  1204
marci@762
  1205
      ++bfs;
marci@762
  1206
    } //end of searching augmenting path
marci@762
  1207
marci@762
  1208
    if (_augment) {
marci@762
  1209
      Node n=t;
marci@762
  1210
      Num augment_value=free[t];
marci@762
  1211
      while (res_graph.valid(pred[n])) {
marci@762
  1212
	ResGWEdge e=pred[n];
marci@762
  1213
	res_graph.augment(e, augment_value);
marci@762
  1214
	n=res_graph.tail(e);
marci@762
  1215
      }
marci@762
  1216
    }
marci@762
  1217
marci@762
  1218
    status=AFTER_FAST_AUGMENTING;
marci@762
  1219
    return _augment;
marci@762
  1220
  }
marci@762
  1221
marci@762
  1222
marci@762
  1223
  template <typename Graph, typename Num, typename CapMap, typename FlowMap>
marci@762
  1224
  template<typename MutableGraph>
marci@762
  1225
  bool AugmentingFlow<Graph, Num, CapMap, FlowMap>::augmentOnBlockingFlow()
marci@762
  1226
  {
marci@762
  1227
    typedef MutableGraph MG;
marci@762
  1228
    bool _augment=false;
marci@762
  1229
marci@762
  1230
    ResGW res_graph(*g, *capacity, *flow);
marci@762
  1231
marci@762
  1232
    //bfs for distances on the residual graph
marci@762
  1233
    //ReachedMap level(res_graph);
marci@762
  1234
    FOR_EACH_LOC(typename Graph::NodeIt, e, *g) level.set(e, 0);
marci@762
  1235
    BfsIterator<ResGW, ReachedMap> bfs(res_graph, level);
marci@762
  1236
    bfs.pushAndSetReached(s);
marci@762
  1237
    typename ResGW::template NodeMap<int>
marci@762
  1238
      dist(res_graph); //filled up with 0's
marci@762
  1239
marci@762
  1240
    //F will contain the physical copy of the residual graph
marci@762
  1241
    //with the set of edges which are on shortest paths
marci@762
  1242
    MG F;
marci@762
  1243
    typename ResGW::template NodeMap<typename MG::Node>
marci@762
  1244
      res_graph_to_F(res_graph);
marci@762
  1245
    {
marci@762
  1246
      typename ResGW::NodeIt n;
marci@762
  1247
      for(res_graph.first(n); res_graph.valid(n); res_graph.next(n)) {
marci@762
  1248
	res_graph_to_F.set(n, F.addNode());
marci@762
  1249
      }
marci@762
  1250
    }
marci@762
  1251
marci@762
  1252
    typename MG::Node sF=res_graph_to_F[s];
marci@762
  1253
    typename MG::Node tF=res_graph_to_F[t];
marci@762
  1254
    typename MG::template EdgeMap<ResGWEdge> original_edge(F);
marci@762
  1255
    typename MG::template EdgeMap<Num> residual_capacity(F);
marci@762
  1256
marci@762
  1257
    while ( !bfs.finished() ) {
marci@762
  1258
      ResGWOutEdgeIt e=bfs;
marci@762
  1259
      if (res_graph.valid(e)) {
marci@762
  1260
	if (bfs.isBNodeNewlyReached()) {
marci@762
  1261
	  dist.set(res_graph.head(e), dist[res_graph.tail(e)]+1);
marci@762
  1262
	  typename MG::Edge f=F.addEdge(res_graph_to_F[res_graph.tail(e)],
marci@762
  1263
					res_graph_to_F[res_graph.head(e)]);
marci@762
  1264
	  original_edge.update();
marci@762
  1265
	  original_edge.set(f, e);
marci@762
  1266
	  residual_capacity.update();
marci@762
  1267
	  residual_capacity.set(f, res_graph.resCap(e));
marci@762
  1268
	} else {
marci@762
  1269
	  if (dist[res_graph.head(e)]==(dist[res_graph.tail(e)]+1)) {
marci@762
  1270
	    typename MG::Edge f=F.addEdge(res_graph_to_F[res_graph.tail(e)],
marci@762
  1271
					  res_graph_to_F[res_graph.head(e)]);
marci@762
  1272
	    original_edge.update();
marci@762
  1273
	    original_edge.set(f, e);
marci@762
  1274
	    residual_capacity.update();
marci@762
  1275
	    residual_capacity.set(f, res_graph.resCap(e));
marci@762
  1276
	  }
marci@762
  1277
	}
marci@762
  1278
      }
marci@762
  1279
      ++bfs;
marci@762
  1280
    } //computing distances from s in the residual graph
marci@762
  1281
marci@762
  1282
    bool __augment=true;
marci@762
  1283
marci@762
  1284
    while (__augment) {
marci@762
  1285
      __augment=false;
marci@762
  1286
      //computing blocking flow with dfs
marci@762
  1287
      DfsIterator< MG, typename MG::template NodeMap<bool> > dfs(F);
marci@762
  1288
      typename MG::template NodeMap<typename MG::Edge> pred(F);
marci@762
  1289
      pred.set(sF, INVALID);
marci@762
  1290
      //invalid iterators for sources
marci@762
  1291
marci@762
  1292
      typename MG::template NodeMap<Num> free(F);
marci@762
  1293
marci@762
  1294
      dfs.pushAndSetReached(sF);
marci@762
  1295
      while (!dfs.finished()) {
marci@762
  1296
	++dfs;
marci@762
  1297
	if (F.valid(/*typename MG::OutEdgeIt*/(dfs))) {
marci@762
  1298
	  if (dfs.isBNodeNewlyReached()) {
marci@762
  1299
	    typename MG::Node v=F.aNode(dfs);
marci@762
  1300
	    typename MG::Node w=F.bNode(dfs);
marci@762
  1301
	    pred.set(w, dfs);
marci@762
  1302
	    if (F.valid(pred[v])) {
marci@762
  1303
	      free.set(w, std::min(free[v], residual_capacity[dfs]));
marci@762
  1304
	    } else {
marci@762
  1305
	      free.set(w, residual_capacity[dfs]);
marci@762
  1306
	    }
marci@762
  1307
	    if (w==tF) {
marci@762
  1308
	      __augment=true;
marci@762
  1309
	      _augment=true;
marci@762
  1310
	      break;
marci@762
  1311
	    }
marci@762
  1312
marci@762
  1313
	  } else {
marci@762
  1314
	    F.erase(/*typename MG::OutEdgeIt*/(dfs));
marci@762
  1315
	  }
marci@762
  1316
	}
marci@762
  1317
      }
marci@762
  1318
marci@762
  1319
      if (__augment) {
marci@762
  1320
	typename MG::Node n=tF;
marci@762
  1321
	Num augment_value=free[tF];
marci@762
  1322
	while (F.valid(pred[n])) {
marci@762
  1323
	  typename MG::Edge e=pred[n];
marci@762
  1324
	  res_graph.augment(original_edge[e], augment_value);
marci@762
  1325
	  n=F.tail(e);
marci@762
  1326
	  if (residual_capacity[e]==augment_value)
marci@762
  1327
	    F.erase(e);
marci@762
  1328
	  else
marci@762
  1329
	    residual_capacity.set(e, residual_capacity[e]-augment_value);
marci@762
  1330
	}
marci@762
  1331
      }
marci@762
  1332
marci@762
  1333
    }
marci@762
  1334
marci@762
  1335
    status=AFTER_AUGMENTING;
marci@762
  1336
    return _augment;
marci@762
  1337
  }
marci@762
  1338
marci@762
  1339
marci@762
  1340
marci@762
  1341
marci@762
  1342
  template <typename Graph, typename Num, typename CapMap, typename FlowMap>
marci@762
  1343
  bool AugmentingFlow<Graph, Num, CapMap, FlowMap>::augmentOnBlockingFlow2()
marci@762
  1344
  {
marci@762
  1345
    bool _augment=false;
marci@762
  1346
marci@762
  1347
    ResGW res_graph(*g, *capacity, *flow);
marci@762
  1348
marci@762
  1349
    //ReachedMap level(res_graph);
marci@762
  1350
    FOR_EACH_LOC(typename Graph::NodeIt, e, *g) level.set(e, 0);
marci@762
  1351
    BfsIterator<ResGW, ReachedMap> bfs(res_graph, level);
marci@762
  1352
marci@762
  1353
    bfs.pushAndSetReached(s);
marci@762
  1354
    DistanceMap<ResGW> dist(res_graph);
marci@762
  1355
    while ( !bfs.finished() ) {
marci@762
  1356
      ResGWOutEdgeIt e=bfs;
marci@762
  1357
      if (res_graph.valid(e) && bfs.isBNodeNewlyReached()) {
marci@762
  1358
	dist.set(res_graph.head(e), dist[res_graph.tail(e)]+1);
marci@762
  1359
      }
marci@762
  1360
      ++bfs;
marci@762
  1361
    } //computing distances from s in the residual graph
marci@762
  1362
marci@762
  1363
      //Subgraph containing the edges on some shortest paths
marci@762
  1364
    ConstMap<typename ResGW::Node, bool> true_map(true);
marci@762
  1365
    typedef SubGraphWrapper<ResGW, ConstMap<typename ResGW::Node, bool>,
marci@762
  1366
      DistanceMap<ResGW> > FilterResGW;
marci@762
  1367
    FilterResGW filter_res_graph(res_graph, true_map, dist);
marci@762
  1368
marci@762
  1369
    //Subgraph, which is able to delete edges which are already
marci@762
  1370
    //met by the dfs
marci@762
  1371
    typename FilterResGW::template NodeMap<typename FilterResGW::OutEdgeIt>
marci@762
  1372
      first_out_edges(filter_res_graph);
marci@762
  1373
    typename FilterResGW::NodeIt v;
marci@762
  1374
    for(filter_res_graph.first(v); filter_res_graph.valid(v);
marci@762
  1375
	filter_res_graph.next(v))
marci@762
  1376
      {
marci@762
  1377
 	typename FilterResGW::OutEdgeIt e;
marci@762
  1378
 	filter_res_graph.first(e, v);
marci@762
  1379
 	first_out_edges.set(v, e);
marci@762
  1380
      }
marci@762
  1381
    typedef ErasingFirstGraphWrapper<FilterResGW, typename FilterResGW::
marci@762
  1382
      template NodeMap<typename FilterResGW::OutEdgeIt> > ErasingResGW;
marci@762
  1383
    ErasingResGW erasing_res_graph(filter_res_graph, first_out_edges);
marci@762
  1384
marci@762
  1385
    bool __augment=true;
marci@762
  1386
marci@762
  1387
    while (__augment) {
marci@762
  1388
marci@762
  1389
      __augment=false;
marci@762
  1390
      //computing blocking flow with dfs
marci@762
  1391
      DfsIterator< ErasingResGW,
marci@762
  1392
	typename ErasingResGW::template NodeMap<bool> >
marci@762
  1393
	dfs(erasing_res_graph);
marci@762
  1394
      typename ErasingResGW::
marci@762
  1395
	template NodeMap<typename ErasingResGW::OutEdgeIt>
marci@762
  1396
	pred(erasing_res_graph);
marci@762
  1397
      pred.set(s, INVALID);
marci@762
  1398
      //invalid iterators for sources
marci@762
  1399
marci@762
  1400
      typename ErasingResGW::template NodeMap<Num>
marci@762
  1401
	free1(erasing_res_graph);
marci@762
  1402
marci@762
  1403
      dfs.pushAndSetReached
marci@762
  1404
	///\bug hugo 0.2
marci@762
  1405
	(typename ErasingResGW::Node
marci@762
  1406
	 (typename FilterResGW::Node
marci@762
  1407
	  (typename ResGW::Node(s)
marci@762
  1408
	   )
marci@762
  1409
	  )
marci@762
  1410
	 );
marci@762
  1411
      while (!dfs.finished()) {
marci@762
  1412
	++dfs;
marci@762
  1413
	if (erasing_res_graph.valid(typename ErasingResGW::OutEdgeIt(dfs)))
marci@762
  1414
 	  {
marci@762
  1415
  	    if (dfs.isBNodeNewlyReached()) {
marci@762
  1416
marci@762
  1417
 	      typename ErasingResGW::Node v=erasing_res_graph.aNode(dfs);
marci@762
  1418
 	      typename ErasingResGW::Node w=erasing_res_graph.bNode(dfs);
marci@762
  1419
marci@762
  1420
 	      pred.set(w, /*typename ErasingResGW::OutEdgeIt*/(dfs));
marci@762
  1421
 	      if (erasing_res_graph.valid(pred[v])) {
marci@762
  1422
 		free1.set
marci@762
  1423
		  (w, std::min(free1[v], res_graph.resCap
marci@762
  1424
			       (typename ErasingResGW::OutEdgeIt(dfs))));
marci@762
  1425
 	      } else {
marci@762
  1426
 		free1.set
marci@762
  1427
		  (w, res_graph.resCap
marci@762
  1428
		   (typename ErasingResGW::OutEdgeIt(dfs)));
marci@762
  1429
 	      }
marci@762
  1430
marci@762
  1431
 	      if (w==t) {
marci@762
  1432
 		__augment=true;
marci@762
  1433
 		_augment=true;
marci@762
  1434
 		break;
marci@762
  1435
 	      }
marci@762
  1436
 	    } else {
marci@762
  1437
 	      erasing_res_graph.erase(dfs);
marci@762
  1438
	    }
marci@762
  1439
	  }
marci@762
  1440
      }
marci@762
  1441
marci@762
  1442
      if (__augment) {
marci@762
  1443
	typename ErasingResGW::Node
marci@762
  1444
	  n=typename FilterResGW::Node(typename ResGW::Node(t));
marci@762
  1445
	// 	  typename ResGW::NodeMap<Num> a(res_graph);
marci@762
  1446
	// 	  typename ResGW::Node b;
marci@762
  1447
	// 	  Num j=a[b];
marci@762
  1448
	// 	  typename FilterResGW::NodeMap<Num> a1(filter_res_graph);
marci@762
  1449
	// 	  typename FilterResGW::Node b1;
marci@762
  1450
	// 	  Num j1=a1[b1];
marci@762
  1451
	// 	  typename ErasingResGW::NodeMap<Num> a2(erasing_res_graph);
marci@762
  1452
	// 	  typename ErasingResGW::Node b2;
marci@762
  1453
	// 	  Num j2=a2[b2];
marci@762
  1454
	Num augment_value=free1[n];
marci@762
  1455
	while (erasing_res_graph.valid(pred[n])) {
marci@762
  1456
	  typename ErasingResGW::OutEdgeIt e=pred[n];
marci@762
  1457
	  res_graph.augment(e, augment_value);
marci@762
  1458
	  n=erasing_res_graph.tail(e);
marci@762
  1459
	  if (res_graph.resCap(e)==0)
marci@762
  1460
	    erasing_res_graph.erase(e);
marci@762
  1461
	}
marci@762
  1462
      }
marci@762
  1463
marci@762
  1464
    } //while (__augment)
marci@762
  1465
marci@762
  1466
    status=AFTER_AUGMENTING;
marci@762
  1467
    return _augment;
marci@762
  1468
  }
marci@762
  1469
marci@762
  1470
marci@762
  1471
} //namespace hugo
marci@762
  1472
marci@762
  1473
#endif //HUGO_AUGMENTING_FLOW_H
marci@762
  1474
marci@762
  1475
marci@762
  1476
marci@762
  1477