lemon/hao_orlin.h
author Balazs Dezso <deba@inf.elte.hu>
Mon, 01 Dec 2008 23:12:16 +0100
changeset 425 b8ce15103485
child 427 01c443515ad2
permissions -rw-r--r--
Port Hao-Orlin algorithm from SVN -r3509 (#58)
deba@425
     1
/* -*- mode: C++; indent-tabs-mode: nil; -*-
deba@425
     2
 *
deba@425
     3
 * This file is a part of LEMON, a generic C++ optimization library.
deba@425
     4
 *
deba@425
     5
 * Copyright (C) 2003-2008
deba@425
     6
 * Egervary Jeno Kombinatorikus Optimalizalasi Kutatocsoport
deba@425
     7
 * (Egervary Research Group on Combinatorial Optimization, EGRES).
deba@425
     8
 *
deba@425
     9
 * Permission to use, modify and distribute this software is granted
deba@425
    10
 * provided that this copyright notice appears in all copies. For
deba@425
    11
 * precise terms see the accompanying LICENSE file.
deba@425
    12
 *
deba@425
    13
 * This software is provided "AS IS" with no warranty of any kind,
deba@425
    14
 * express or implied, and with no claim as to its suitability for any
deba@425
    15
 * purpose.
deba@425
    16
 *
deba@425
    17
 */
deba@425
    18
deba@425
    19
#ifndef LEMON_HAO_ORLIN_H
deba@425
    20
#define LEMON_HAO_ORLIN_H
deba@425
    21
deba@425
    22
#include <vector>
deba@425
    23
#include <list>
deba@425
    24
#include <limits>
deba@425
    25
deba@425
    26
#include <lemon/maps.h>
deba@425
    27
#include <lemon/core.h>
deba@425
    28
#include <lemon/tolerance.h>
deba@425
    29
deba@425
    30
/// \file
deba@425
    31
/// \ingroup min_cut
deba@425
    32
/// \brief Implementation of the Hao-Orlin algorithm.
deba@425
    33
///
deba@425
    34
/// Implementation of the Hao-Orlin algorithm class for testing network
deba@425
    35
/// reliability.
deba@425
    36
deba@425
    37
namespace lemon {
deba@425
    38
deba@425
    39
  /// \ingroup min_cut
deba@425
    40
  ///
deba@425
    41
  /// \brief %Hao-Orlin algorithm to find a minimum cut in directed graphs.
deba@425
    42
  ///
deba@425
    43
  /// Hao-Orlin calculates a minimum cut in a directed graph
deba@425
    44
  /// \f$D=(V,A)\f$. It takes a fixed node \f$ source \in V \f$ and
deba@425
    45
  /// consists of two phases: in the first phase it determines a
deba@425
    46
  /// minimum cut with \f$ source \f$ on the source-side (i.e. a set
deba@425
    47
  /// \f$ X\subsetneq V \f$ with \f$ source \in X \f$ and minimal
deba@425
    48
  /// out-degree) and in the second phase it determines a minimum cut
deba@425
    49
  /// with \f$ source \f$ on the sink-side (i.e. a set
deba@425
    50
  /// \f$ X\subsetneq V \f$ with \f$ source \notin X \f$ and minimal
deba@425
    51
  /// out-degree). Obviously, the smaller of these two cuts will be a
deba@425
    52
  /// minimum cut of \f$ D \f$. The algorithm is a modified
deba@425
    53
  /// push-relabel preflow algorithm and our implementation calculates
deba@425
    54
  /// the minimum cut in \f$ O(n^2\sqrt{m}) \f$ time (we use the
deba@425
    55
  /// highest-label rule), or in \f$O(nm)\f$ for unit capacities. The
deba@425
    56
  /// purpose of such algorithm is testing network reliability. For an
deba@425
    57
  /// undirected graph you can run just the first phase of the
deba@425
    58
  /// algorithm or you can use the algorithm of Nagamochi and Ibaraki
deba@425
    59
  /// which solves the undirected problem in
deba@425
    60
  /// \f$ O(nm + n^2 \log(n)) \f$ time: it is implemented in the
deba@425
    61
  /// NagamochiIbaraki algorithm class.
deba@425
    62
  ///
deba@425
    63
  /// \param _Digraph is the graph type of the algorithm.
deba@425
    64
  /// \param _CapacityMap is an edge map of capacities which should
deba@425
    65
  /// be any numreric type. The default type is _Digraph::ArcMap<int>.
deba@425
    66
  /// \param _Tolerance is the handler of the inexact computation. The
deba@425
    67
  /// default type for this is Tolerance<CapacityMap::Value>.
deba@425
    68
#ifdef DOXYGEN
deba@425
    69
  template <typename _Digraph, typename _CapacityMap, typename _Tolerance>
deba@425
    70
#else
deba@425
    71
  template <typename _Digraph,
deba@425
    72
            typename _CapacityMap = typename _Digraph::template ArcMap<int>,
deba@425
    73
            typename _Tolerance = Tolerance<typename _CapacityMap::Value> >
deba@425
    74
#endif
deba@425
    75
  class HaoOrlin {
deba@425
    76
  private:
deba@425
    77
deba@425
    78
    typedef _Digraph Digraph;
deba@425
    79
    typedef _CapacityMap CapacityMap;
deba@425
    80
    typedef _Tolerance Tolerance;
deba@425
    81
deba@425
    82
    typedef typename CapacityMap::Value Value;
deba@425
    83
deba@425
    84
    TEMPLATE_GRAPH_TYPEDEFS(Digraph);
deba@425
    85
deba@425
    86
    const Digraph& _graph;
deba@425
    87
    const CapacityMap* _capacity;
deba@425
    88
deba@425
    89
    typedef typename Digraph::template ArcMap<Value> FlowMap;
deba@425
    90
    FlowMap* _flow;
deba@425
    91
deba@425
    92
    Node _source;
deba@425
    93
deba@425
    94
    int _node_num;
deba@425
    95
deba@425
    96
    // Bucketing structure
deba@425
    97
    std::vector<Node> _first, _last;
deba@425
    98
    typename Digraph::template NodeMap<Node>* _next;
deba@425
    99
    typename Digraph::template NodeMap<Node>* _prev;
deba@425
   100
    typename Digraph::template NodeMap<bool>* _active;
deba@425
   101
    typename Digraph::template NodeMap<int>* _bucket;
deba@425
   102
deba@425
   103
    std::vector<bool> _dormant;
deba@425
   104
deba@425
   105
    std::list<std::list<int> > _sets;
deba@425
   106
    std::list<int>::iterator _highest;
deba@425
   107
deba@425
   108
    typedef typename Digraph::template NodeMap<Value> ExcessMap;
deba@425
   109
    ExcessMap* _excess;
deba@425
   110
deba@425
   111
    typedef typename Digraph::template NodeMap<bool> SourceSetMap;
deba@425
   112
    SourceSetMap* _source_set;
deba@425
   113
deba@425
   114
    Value _min_cut;
deba@425
   115
deba@425
   116
    typedef typename Digraph::template NodeMap<bool> MinCutMap;
deba@425
   117
    MinCutMap* _min_cut_map;
deba@425
   118
deba@425
   119
    Tolerance _tolerance;
deba@425
   120
deba@425
   121
  public:
deba@425
   122
deba@425
   123
    /// \brief Constructor
deba@425
   124
    ///
deba@425
   125
    /// Constructor of the algorithm class.
deba@425
   126
    HaoOrlin(const Digraph& graph, const CapacityMap& capacity,
deba@425
   127
             const Tolerance& tolerance = Tolerance()) :
deba@425
   128
      _graph(graph), _capacity(&capacity), _flow(0), _source(),
deba@425
   129
      _node_num(), _first(), _last(), _next(0), _prev(0),
deba@425
   130
      _active(0), _bucket(0), _dormant(), _sets(), _highest(),
deba@425
   131
      _excess(0), _source_set(0), _min_cut(), _min_cut_map(0),
deba@425
   132
      _tolerance(tolerance) {}
deba@425
   133
deba@425
   134
    ~HaoOrlin() {
deba@425
   135
      if (_min_cut_map) {
deba@425
   136
        delete _min_cut_map;
deba@425
   137
      }
deba@425
   138
      if (_source_set) {
deba@425
   139
        delete _source_set;
deba@425
   140
      }
deba@425
   141
      if (_excess) {
deba@425
   142
        delete _excess;
deba@425
   143
      }
deba@425
   144
      if (_next) {
deba@425
   145
        delete _next;
deba@425
   146
      }
deba@425
   147
      if (_prev) {
deba@425
   148
        delete _prev;
deba@425
   149
      }
deba@425
   150
      if (_active) {
deba@425
   151
        delete _active;
deba@425
   152
      }
deba@425
   153
      if (_bucket) {
deba@425
   154
        delete _bucket;
deba@425
   155
      }
deba@425
   156
      if (_flow) {
deba@425
   157
        delete _flow;
deba@425
   158
      }
deba@425
   159
    }
deba@425
   160
deba@425
   161
  private:
deba@425
   162
deba@425
   163
    void activate(const Node& i) {
deba@425
   164
      _active->set(i, true);
deba@425
   165
deba@425
   166
      int bucket = (*_bucket)[i];
deba@425
   167
deba@425
   168
      if ((*_prev)[i] == INVALID || (*_active)[(*_prev)[i]]) return;
deba@425
   169
      //unlace
deba@425
   170
      _next->set((*_prev)[i], (*_next)[i]);
deba@425
   171
      if ((*_next)[i] != INVALID) {
deba@425
   172
        _prev->set((*_next)[i], (*_prev)[i]);
deba@425
   173
      } else {
deba@425
   174
        _last[bucket] = (*_prev)[i];
deba@425
   175
      }
deba@425
   176
      //lace
deba@425
   177
      _next->set(i, _first[bucket]);
deba@425
   178
      _prev->set(_first[bucket], i);
deba@425
   179
      _prev->set(i, INVALID);
deba@425
   180
      _first[bucket] = i;
deba@425
   181
    }
deba@425
   182
deba@425
   183
    void deactivate(const Node& i) {
deba@425
   184
      _active->set(i, false);
deba@425
   185
      int bucket = (*_bucket)[i];
deba@425
   186
deba@425
   187
      if ((*_next)[i] == INVALID || !(*_active)[(*_next)[i]]) return;
deba@425
   188
deba@425
   189
      //unlace
deba@425
   190
      _prev->set((*_next)[i], (*_prev)[i]);
deba@425
   191
      if ((*_prev)[i] != INVALID) {
deba@425
   192
        _next->set((*_prev)[i], (*_next)[i]);
deba@425
   193
      } else {
deba@425
   194
        _first[bucket] = (*_next)[i];
deba@425
   195
      }
deba@425
   196
      //lace
deba@425
   197
      _prev->set(i, _last[bucket]);
deba@425
   198
      _next->set(_last[bucket], i);
deba@425
   199
      _next->set(i, INVALID);
deba@425
   200
      _last[bucket] = i;
deba@425
   201
    }
deba@425
   202
deba@425
   203
    void addItem(const Node& i, int bucket) {
deba@425
   204
      (*_bucket)[i] = bucket;
deba@425
   205
      if (_last[bucket] != INVALID) {
deba@425
   206
        _prev->set(i, _last[bucket]);
deba@425
   207
        _next->set(_last[bucket], i);
deba@425
   208
        _next->set(i, INVALID);
deba@425
   209
        _last[bucket] = i;
deba@425
   210
      } else {
deba@425
   211
        _prev->set(i, INVALID);
deba@425
   212
        _first[bucket] = i;
deba@425
   213
        _next->set(i, INVALID);
deba@425
   214
        _last[bucket] = i;
deba@425
   215
      }
deba@425
   216
    }
deba@425
   217
deba@425
   218
    void findMinCutOut() {
deba@425
   219
deba@425
   220
      for (NodeIt n(_graph); n != INVALID; ++n) {
deba@425
   221
        _excess->set(n, 0);
deba@425
   222
      }
deba@425
   223
deba@425
   224
      for (ArcIt a(_graph); a != INVALID; ++a) {
deba@425
   225
        _flow->set(a, 0);
deba@425
   226
      }
deba@425
   227
deba@425
   228
      int bucket_num = 1;
deba@425
   229
deba@425
   230
      {
deba@425
   231
        typename Digraph::template NodeMap<bool> reached(_graph, false);
deba@425
   232
deba@425
   233
        reached.set(_source, true);
deba@425
   234
deba@425
   235
        bool first_set = true;
deba@425
   236
deba@425
   237
        for (NodeIt t(_graph); t != INVALID; ++t) {
deba@425
   238
          if (reached[t]) continue;
deba@425
   239
          _sets.push_front(std::list<int>());
deba@425
   240
          _sets.front().push_front(bucket_num);
deba@425
   241
          _dormant[bucket_num] = !first_set;
deba@425
   242
deba@425
   243
          _bucket->set(t, bucket_num);
deba@425
   244
          _first[bucket_num] = _last[bucket_num] = t;
deba@425
   245
          _next->set(t, INVALID);
deba@425
   246
          _prev->set(t, INVALID);
deba@425
   247
deba@425
   248
          ++bucket_num;
deba@425
   249
deba@425
   250
          std::vector<Node> queue;
deba@425
   251
          queue.push_back(t);
deba@425
   252
          reached.set(t, true);
deba@425
   253
deba@425
   254
          while (!queue.empty()) {
deba@425
   255
            _sets.front().push_front(bucket_num);
deba@425
   256
            _dormant[bucket_num] = !first_set;
deba@425
   257
            _first[bucket_num] = _last[bucket_num] = INVALID;
deba@425
   258
deba@425
   259
            std::vector<Node> nqueue;
deba@425
   260
            for (int i = 0; i < int(queue.size()); ++i) {
deba@425
   261
              Node n = queue[i];
deba@425
   262
              for (InArcIt a(_graph, n); a != INVALID; ++a) {
deba@425
   263
                Node u = _graph.source(a);
deba@425
   264
                if (!reached[u] && _tolerance.positive((*_capacity)[a])) {
deba@425
   265
                  reached.set(u, true);
deba@425
   266
                  addItem(u, bucket_num);
deba@425
   267
                  nqueue.push_back(u);
deba@425
   268
                }
deba@425
   269
              }
deba@425
   270
            }
deba@425
   271
            queue.swap(nqueue);
deba@425
   272
            ++bucket_num;
deba@425
   273
          }
deba@425
   274
          _sets.front().pop_front();
deba@425
   275
          --bucket_num;
deba@425
   276
          first_set = false;
deba@425
   277
        }
deba@425
   278
deba@425
   279
        _bucket->set(_source, 0);
deba@425
   280
        _dormant[0] = true;
deba@425
   281
      }
deba@425
   282
      _source_set->set(_source, true);
deba@425
   283
deba@425
   284
      Node target = _last[_sets.back().back()];
deba@425
   285
      {
deba@425
   286
        for (OutArcIt a(_graph, _source); a != INVALID; ++a) {
deba@425
   287
          if (_tolerance.positive((*_capacity)[a])) {
deba@425
   288
            Node u = _graph.target(a);
deba@425
   289
            _flow->set(a, (*_capacity)[a]);
deba@425
   290
            _excess->set(u, (*_excess)[u] + (*_capacity)[a]);
deba@425
   291
            if (!(*_active)[u] && u != _source) {
deba@425
   292
              activate(u);
deba@425
   293
            }
deba@425
   294
          }
deba@425
   295
        }
deba@425
   296
deba@425
   297
        if ((*_active)[target]) {
deba@425
   298
          deactivate(target);
deba@425
   299
        }
deba@425
   300
deba@425
   301
        _highest = _sets.back().begin();
deba@425
   302
        while (_highest != _sets.back().end() &&
deba@425
   303
               !(*_active)[_first[*_highest]]) {
deba@425
   304
          ++_highest;
deba@425
   305
        }
deba@425
   306
      }
deba@425
   307
deba@425
   308
      while (true) {
deba@425
   309
        while (_highest != _sets.back().end()) {
deba@425
   310
          Node n = _first[*_highest];
deba@425
   311
          Value excess = (*_excess)[n];
deba@425
   312
          int next_bucket = _node_num;
deba@425
   313
deba@425
   314
          int under_bucket;
deba@425
   315
          if (++std::list<int>::iterator(_highest) == _sets.back().end()) {
deba@425
   316
            under_bucket = -1;
deba@425
   317
          } else {
deba@425
   318
            under_bucket = *(++std::list<int>::iterator(_highest));
deba@425
   319
          }
deba@425
   320
deba@425
   321
          for (OutArcIt a(_graph, n); a != INVALID; ++a) {
deba@425
   322
            Node v = _graph.target(a);
deba@425
   323
            if (_dormant[(*_bucket)[v]]) continue;
deba@425
   324
            Value rem = (*_capacity)[a] - (*_flow)[a];
deba@425
   325
            if (!_tolerance.positive(rem)) continue;
deba@425
   326
            if ((*_bucket)[v] == under_bucket) {
deba@425
   327
              if (!(*_active)[v] && v != target) {
deba@425
   328
                activate(v);
deba@425
   329
              }
deba@425
   330
              if (!_tolerance.less(rem, excess)) {
deba@425
   331
                _flow->set(a, (*_flow)[a] + excess);
deba@425
   332
                _excess->set(v, (*_excess)[v] + excess);
deba@425
   333
                excess = 0;
deba@425
   334
                goto no_more_push;
deba@425
   335
              } else {
deba@425
   336
                excess -= rem;
deba@425
   337
                _excess->set(v, (*_excess)[v] + rem);
deba@425
   338
                _flow->set(a, (*_capacity)[a]);
deba@425
   339
              }
deba@425
   340
            } else if (next_bucket > (*_bucket)[v]) {
deba@425
   341
              next_bucket = (*_bucket)[v];
deba@425
   342
            }
deba@425
   343
          }
deba@425
   344
deba@425
   345
          for (InArcIt a(_graph, n); a != INVALID; ++a) {
deba@425
   346
            Node v = _graph.source(a);
deba@425
   347
            if (_dormant[(*_bucket)[v]]) continue;
deba@425
   348
            Value rem = (*_flow)[a];
deba@425
   349
            if (!_tolerance.positive(rem)) continue;
deba@425
   350
            if ((*_bucket)[v] == under_bucket) {
deba@425
   351
              if (!(*_active)[v] && v != target) {
deba@425
   352
                activate(v);
deba@425
   353
              }
deba@425
   354
              if (!_tolerance.less(rem, excess)) {
deba@425
   355
                _flow->set(a, (*_flow)[a] - excess);
deba@425
   356
                _excess->set(v, (*_excess)[v] + excess);
deba@425
   357
                excess = 0;
deba@425
   358
                goto no_more_push;
deba@425
   359
              } else {
deba@425
   360
                excess -= rem;
deba@425
   361
                _excess->set(v, (*_excess)[v] + rem);
deba@425
   362
                _flow->set(a, 0);
deba@425
   363
              }
deba@425
   364
            } else if (next_bucket > (*_bucket)[v]) {
deba@425
   365
              next_bucket = (*_bucket)[v];
deba@425
   366
            }
deba@425
   367
          }
deba@425
   368
deba@425
   369
        no_more_push:
deba@425
   370
deba@425
   371
          _excess->set(n, excess);
deba@425
   372
deba@425
   373
          if (excess != 0) {
deba@425
   374
            if ((*_next)[n] == INVALID) {
deba@425
   375
              typename std::list<std::list<int> >::iterator new_set =
deba@425
   376
                _sets.insert(--_sets.end(), std::list<int>());
deba@425
   377
              new_set->splice(new_set->end(), _sets.back(),
deba@425
   378
                              _sets.back().begin(), ++_highest);
deba@425
   379
              for (std::list<int>::iterator it = new_set->begin();
deba@425
   380
                   it != new_set->end(); ++it) {
deba@425
   381
                _dormant[*it] = true;
deba@425
   382
              }
deba@425
   383
              while (_highest != _sets.back().end() &&
deba@425
   384
                     !(*_active)[_first[*_highest]]) {
deba@425
   385
                ++_highest;
deba@425
   386
              }
deba@425
   387
            } else if (next_bucket == _node_num) {
deba@425
   388
              _first[(*_bucket)[n]] = (*_next)[n];
deba@425
   389
              _prev->set((*_next)[n], INVALID);
deba@425
   390
deba@425
   391
              std::list<std::list<int> >::iterator new_set =
deba@425
   392
                _sets.insert(--_sets.end(), std::list<int>());
deba@425
   393
deba@425
   394
              new_set->push_front(bucket_num);
deba@425
   395
              _bucket->set(n, bucket_num);
deba@425
   396
              _first[bucket_num] = _last[bucket_num] = n;
deba@425
   397
              _next->set(n, INVALID);
deba@425
   398
              _prev->set(n, INVALID);
deba@425
   399
              _dormant[bucket_num] = true;
deba@425
   400
              ++bucket_num;
deba@425
   401
deba@425
   402
              while (_highest != _sets.back().end() &&
deba@425
   403
                     !(*_active)[_first[*_highest]]) {
deba@425
   404
                ++_highest;
deba@425
   405
              }
deba@425
   406
            } else {
deba@425
   407
              _first[*_highest] = (*_next)[n];
deba@425
   408
              _prev->set((*_next)[n], INVALID);
deba@425
   409
deba@425
   410
              while (next_bucket != *_highest) {
deba@425
   411
                --_highest;
deba@425
   412
              }
deba@425
   413
deba@425
   414
              if (_highest == _sets.back().begin()) {
deba@425
   415
                _sets.back().push_front(bucket_num);
deba@425
   416
                _dormant[bucket_num] = false;
deba@425
   417
                _first[bucket_num] = _last[bucket_num] = INVALID;
deba@425
   418
                ++bucket_num;
deba@425
   419
              }
deba@425
   420
              --_highest;
deba@425
   421
deba@425
   422
              _bucket->set(n, *_highest);
deba@425
   423
              _next->set(n, _first[*_highest]);
deba@425
   424
              if (_first[*_highest] != INVALID) {
deba@425
   425
                _prev->set(_first[*_highest], n);
deba@425
   426
              } else {
deba@425
   427
                _last[*_highest] = n;
deba@425
   428
              }
deba@425
   429
              _first[*_highest] = n;
deba@425
   430
            }
deba@425
   431
          } else {
deba@425
   432
deba@425
   433
            deactivate(n);
deba@425
   434
            if (!(*_active)[_first[*_highest]]) {
deba@425
   435
              ++_highest;
deba@425
   436
              if (_highest != _sets.back().end() &&
deba@425
   437
                  !(*_active)[_first[*_highest]]) {
deba@425
   438
                _highest = _sets.back().end();
deba@425
   439
              }
deba@425
   440
            }
deba@425
   441
          }
deba@425
   442
        }
deba@425
   443
deba@425
   444
        if ((*_excess)[target] < _min_cut) {
deba@425
   445
          _min_cut = (*_excess)[target];
deba@425
   446
          for (NodeIt i(_graph); i != INVALID; ++i) {
deba@425
   447
            _min_cut_map->set(i, true);
deba@425
   448
          }
deba@425
   449
          for (std::list<int>::iterator it = _sets.back().begin();
deba@425
   450
               it != _sets.back().end(); ++it) {
deba@425
   451
            Node n = _first[*it];
deba@425
   452
            while (n != INVALID) {
deba@425
   453
              _min_cut_map->set(n, false);
deba@425
   454
              n = (*_next)[n];
deba@425
   455
            }
deba@425
   456
          }
deba@425
   457
        }
deba@425
   458
deba@425
   459
        {
deba@425
   460
          Node new_target;
deba@425
   461
          if ((*_prev)[target] != INVALID || (*_next)[target] != INVALID) {
deba@425
   462
            if ((*_next)[target] == INVALID) {
deba@425
   463
              _last[(*_bucket)[target]] = (*_prev)[target];
deba@425
   464
              new_target = (*_prev)[target];
deba@425
   465
            } else {
deba@425
   466
              _prev->set((*_next)[target], (*_prev)[target]);
deba@425
   467
              new_target = (*_next)[target];
deba@425
   468
            }
deba@425
   469
            if ((*_prev)[target] == INVALID) {
deba@425
   470
              _first[(*_bucket)[target]] = (*_next)[target];
deba@425
   471
            } else {
deba@425
   472
              _next->set((*_prev)[target], (*_next)[target]);
deba@425
   473
            }
deba@425
   474
          } else {
deba@425
   475
            _sets.back().pop_back();
deba@425
   476
            if (_sets.back().empty()) {
deba@425
   477
              _sets.pop_back();
deba@425
   478
              if (_sets.empty())
deba@425
   479
                break;
deba@425
   480
              for (std::list<int>::iterator it = _sets.back().begin();
deba@425
   481
                   it != _sets.back().end(); ++it) {
deba@425
   482
                _dormant[*it] = false;
deba@425
   483
              }
deba@425
   484
            }
deba@425
   485
            new_target = _last[_sets.back().back()];
deba@425
   486
          }
deba@425
   487
deba@425
   488
          _bucket->set(target, 0);
deba@425
   489
deba@425
   490
          _source_set->set(target, true);
deba@425
   491
          for (OutArcIt a(_graph, target); a != INVALID; ++a) {
deba@425
   492
            Value rem = (*_capacity)[a] - (*_flow)[a];
deba@425
   493
            if (!_tolerance.positive(rem)) continue;
deba@425
   494
            Node v = _graph.target(a);
deba@425
   495
            if (!(*_active)[v] && !(*_source_set)[v]) {
deba@425
   496
              activate(v);
deba@425
   497
            }
deba@425
   498
            _excess->set(v, (*_excess)[v] + rem);
deba@425
   499
            _flow->set(a, (*_capacity)[a]);
deba@425
   500
          }
deba@425
   501
deba@425
   502
          for (InArcIt a(_graph, target); a != INVALID; ++a) {
deba@425
   503
            Value rem = (*_flow)[a];
deba@425
   504
            if (!_tolerance.positive(rem)) continue;
deba@425
   505
            Node v = _graph.source(a);
deba@425
   506
            if (!(*_active)[v] && !(*_source_set)[v]) {
deba@425
   507
              activate(v);
deba@425
   508
            }
deba@425
   509
            _excess->set(v, (*_excess)[v] + rem);
deba@425
   510
            _flow->set(a, 0);
deba@425
   511
          }
deba@425
   512
deba@425
   513
          target = new_target;
deba@425
   514
          if ((*_active)[target]) {
deba@425
   515
            deactivate(target);
deba@425
   516
          }
deba@425
   517
deba@425
   518
          _highest = _sets.back().begin();
deba@425
   519
          while (_highest != _sets.back().end() &&
deba@425
   520
                 !(*_active)[_first[*_highest]]) {
deba@425
   521
            ++_highest;
deba@425
   522
          }
deba@425
   523
        }
deba@425
   524
      }
deba@425
   525
    }
deba@425
   526
deba@425
   527
    void findMinCutIn() {
deba@425
   528
deba@425
   529
      for (NodeIt n(_graph); n != INVALID; ++n) {
deba@425
   530
        _excess->set(n, 0);
deba@425
   531
      }
deba@425
   532
deba@425
   533
      for (ArcIt a(_graph); a != INVALID; ++a) {
deba@425
   534
        _flow->set(a, 0);
deba@425
   535
      }
deba@425
   536
deba@425
   537
      int bucket_num = 1;
deba@425
   538
deba@425
   539
      {
deba@425
   540
        typename Digraph::template NodeMap<bool> reached(_graph, false);
deba@425
   541
deba@425
   542
        reached.set(_source, true);
deba@425
   543
deba@425
   544
        bool first_set = true;
deba@425
   545
deba@425
   546
        for (NodeIt t(_graph); t != INVALID; ++t) {
deba@425
   547
          if (reached[t]) continue;
deba@425
   548
          _sets.push_front(std::list<int>());
deba@425
   549
          _sets.front().push_front(bucket_num);
deba@425
   550
          _dormant[bucket_num] = !first_set;
deba@425
   551
deba@425
   552
          _bucket->set(t, bucket_num);
deba@425
   553
          _first[bucket_num] = _last[bucket_num] = t;
deba@425
   554
          _next->set(t, INVALID);
deba@425
   555
          _prev->set(t, INVALID);
deba@425
   556
deba@425
   557
          ++bucket_num;
deba@425
   558
deba@425
   559
          std::vector<Node> queue;
deba@425
   560
          queue.push_back(t);
deba@425
   561
          reached.set(t, true);
deba@425
   562
deba@425
   563
          while (!queue.empty()) {
deba@425
   564
            _sets.front().push_front(bucket_num);
deba@425
   565
            _dormant[bucket_num] = !first_set;
deba@425
   566
            _first[bucket_num] = _last[bucket_num] = INVALID;
deba@425
   567
deba@425
   568
            std::vector<Node> nqueue;
deba@425
   569
            for (int i = 0; i < int(queue.size()); ++i) {
deba@425
   570
              Node n = queue[i];
deba@425
   571
              for (OutArcIt a(_graph, n); a != INVALID; ++a) {
deba@425
   572
                Node u = _graph.target(a);
deba@425
   573
                if (!reached[u] && _tolerance.positive((*_capacity)[a])) {
deba@425
   574
                  reached.set(u, true);
deba@425
   575
                  addItem(u, bucket_num);
deba@425
   576
                  nqueue.push_back(u);
deba@425
   577
                }
deba@425
   578
              }
deba@425
   579
            }
deba@425
   580
            queue.swap(nqueue);
deba@425
   581
            ++bucket_num;
deba@425
   582
          }
deba@425
   583
          _sets.front().pop_front();
deba@425
   584
          --bucket_num;
deba@425
   585
          first_set = false;
deba@425
   586
        }
deba@425
   587
deba@425
   588
        _bucket->set(_source, 0);
deba@425
   589
        _dormant[0] = true;
deba@425
   590
      }
deba@425
   591
      _source_set->set(_source, true);
deba@425
   592
deba@425
   593
      Node target = _last[_sets.back().back()];
deba@425
   594
      {
deba@425
   595
        for (InArcIt a(_graph, _source); a != INVALID; ++a) {
deba@425
   596
          if (_tolerance.positive((*_capacity)[a])) {
deba@425
   597
            Node u = _graph.source(a);
deba@425
   598
            _flow->set(a, (*_capacity)[a]);
deba@425
   599
            _excess->set(u, (*_excess)[u] + (*_capacity)[a]);
deba@425
   600
            if (!(*_active)[u] && u != _source) {
deba@425
   601
              activate(u);
deba@425
   602
            }
deba@425
   603
          }
deba@425
   604
        }
deba@425
   605
        if ((*_active)[target]) {
deba@425
   606
          deactivate(target);
deba@425
   607
        }
deba@425
   608
deba@425
   609
        _highest = _sets.back().begin();
deba@425
   610
        while (_highest != _sets.back().end() &&
deba@425
   611
               !(*_active)[_first[*_highest]]) {
deba@425
   612
          ++_highest;
deba@425
   613
        }
deba@425
   614
      }
deba@425
   615
deba@425
   616
deba@425
   617
      while (true) {
deba@425
   618
        while (_highest != _sets.back().end()) {
deba@425
   619
          Node n = _first[*_highest];
deba@425
   620
          Value excess = (*_excess)[n];
deba@425
   621
          int next_bucket = _node_num;
deba@425
   622
deba@425
   623
          int under_bucket;
deba@425
   624
          if (++std::list<int>::iterator(_highest) == _sets.back().end()) {
deba@425
   625
            under_bucket = -1;
deba@425
   626
          } else {
deba@425
   627
            under_bucket = *(++std::list<int>::iterator(_highest));
deba@425
   628
          }
deba@425
   629
deba@425
   630
          for (InArcIt a(_graph, n); a != INVALID; ++a) {
deba@425
   631
            Node v = _graph.source(a);
deba@425
   632
            if (_dormant[(*_bucket)[v]]) continue;
deba@425
   633
            Value rem = (*_capacity)[a] - (*_flow)[a];
deba@425
   634
            if (!_tolerance.positive(rem)) continue;
deba@425
   635
            if ((*_bucket)[v] == under_bucket) {
deba@425
   636
              if (!(*_active)[v] && v != target) {
deba@425
   637
                activate(v);
deba@425
   638
              }
deba@425
   639
              if (!_tolerance.less(rem, excess)) {
deba@425
   640
                _flow->set(a, (*_flow)[a] + excess);
deba@425
   641
                _excess->set(v, (*_excess)[v] + excess);
deba@425
   642
                excess = 0;
deba@425
   643
                goto no_more_push;
deba@425
   644
              } else {
deba@425
   645
                excess -= rem;
deba@425
   646
                _excess->set(v, (*_excess)[v] + rem);
deba@425
   647
                _flow->set(a, (*_capacity)[a]);
deba@425
   648
              }
deba@425
   649
            } else if (next_bucket > (*_bucket)[v]) {
deba@425
   650
              next_bucket = (*_bucket)[v];
deba@425
   651
            }
deba@425
   652
          }
deba@425
   653
deba@425
   654
          for (OutArcIt a(_graph, n); a != INVALID; ++a) {
deba@425
   655
            Node v = _graph.target(a);
deba@425
   656
            if (_dormant[(*_bucket)[v]]) continue;
deba@425
   657
            Value rem = (*_flow)[a];
deba@425
   658
            if (!_tolerance.positive(rem)) continue;
deba@425
   659
            if ((*_bucket)[v] == under_bucket) {
deba@425
   660
              if (!(*_active)[v] && v != target) {
deba@425
   661
                activate(v);
deba@425
   662
              }
deba@425
   663
              if (!_tolerance.less(rem, excess)) {
deba@425
   664
                _flow->set(a, (*_flow)[a] - excess);
deba@425
   665
                _excess->set(v, (*_excess)[v] + excess);
deba@425
   666
                excess = 0;
deba@425
   667
                goto no_more_push;
deba@425
   668
              } else {
deba@425
   669
                excess -= rem;
deba@425
   670
                _excess->set(v, (*_excess)[v] + rem);
deba@425
   671
                _flow->set(a, 0);
deba@425
   672
              }
deba@425
   673
            } else if (next_bucket > (*_bucket)[v]) {
deba@425
   674
              next_bucket = (*_bucket)[v];
deba@425
   675
            }
deba@425
   676
          }
deba@425
   677
deba@425
   678
        no_more_push:
deba@425
   679
deba@425
   680
          _excess->set(n, excess);
deba@425
   681
deba@425
   682
          if (excess != 0) {
deba@425
   683
            if ((*_next)[n] == INVALID) {
deba@425
   684
              typename std::list<std::list<int> >::iterator new_set =
deba@425
   685
                _sets.insert(--_sets.end(), std::list<int>());
deba@425
   686
              new_set->splice(new_set->end(), _sets.back(),
deba@425
   687
                              _sets.back().begin(), ++_highest);
deba@425
   688
              for (std::list<int>::iterator it = new_set->begin();
deba@425
   689
                   it != new_set->end(); ++it) {
deba@425
   690
                _dormant[*it] = true;
deba@425
   691
              }
deba@425
   692
              while (_highest != _sets.back().end() &&
deba@425
   693
                     !(*_active)[_first[*_highest]]) {
deba@425
   694
                ++_highest;
deba@425
   695
              }
deba@425
   696
            } else if (next_bucket == _node_num) {
deba@425
   697
              _first[(*_bucket)[n]] = (*_next)[n];
deba@425
   698
              _prev->set((*_next)[n], INVALID);
deba@425
   699
deba@425
   700
              std::list<std::list<int> >::iterator new_set =
deba@425
   701
                _sets.insert(--_sets.end(), std::list<int>());
deba@425
   702
deba@425
   703
              new_set->push_front(bucket_num);
deba@425
   704
              _bucket->set(n, bucket_num);
deba@425
   705
              _first[bucket_num] = _last[bucket_num] = n;
deba@425
   706
              _next->set(n, INVALID);
deba@425
   707
              _prev->set(n, INVALID);
deba@425
   708
              _dormant[bucket_num] = true;
deba@425
   709
              ++bucket_num;
deba@425
   710
deba@425
   711
              while (_highest != _sets.back().end() &&
deba@425
   712
                     !(*_active)[_first[*_highest]]) {
deba@425
   713
                ++_highest;
deba@425
   714
              }
deba@425
   715
            } else {
deba@425
   716
              _first[*_highest] = (*_next)[n];
deba@425
   717
              _prev->set((*_next)[n], INVALID);
deba@425
   718
deba@425
   719
              while (next_bucket != *_highest) {
deba@425
   720
                --_highest;
deba@425
   721
              }
deba@425
   722
              if (_highest == _sets.back().begin()) {
deba@425
   723
                _sets.back().push_front(bucket_num);
deba@425
   724
                _dormant[bucket_num] = false;
deba@425
   725
                _first[bucket_num] = _last[bucket_num] = INVALID;
deba@425
   726
                ++bucket_num;
deba@425
   727
              }
deba@425
   728
              --_highest;
deba@425
   729
deba@425
   730
              _bucket->set(n, *_highest);
deba@425
   731
              _next->set(n, _first[*_highest]);
deba@425
   732
              if (_first[*_highest] != INVALID) {
deba@425
   733
                _prev->set(_first[*_highest], n);
deba@425
   734
              } else {
deba@425
   735
                _last[*_highest] = n;
deba@425
   736
              }
deba@425
   737
              _first[*_highest] = n;
deba@425
   738
            }
deba@425
   739
          } else {
deba@425
   740
deba@425
   741
            deactivate(n);
deba@425
   742
            if (!(*_active)[_first[*_highest]]) {
deba@425
   743
              ++_highest;
deba@425
   744
              if (_highest != _sets.back().end() &&
deba@425
   745
                  !(*_active)[_first[*_highest]]) {
deba@425
   746
                _highest = _sets.back().end();
deba@425
   747
              }
deba@425
   748
            }
deba@425
   749
          }
deba@425
   750
        }
deba@425
   751
deba@425
   752
        if ((*_excess)[target] < _min_cut) {
deba@425
   753
          _min_cut = (*_excess)[target];
deba@425
   754
          for (NodeIt i(_graph); i != INVALID; ++i) {
deba@425
   755
            _min_cut_map->set(i, false);
deba@425
   756
          }
deba@425
   757
          for (std::list<int>::iterator it = _sets.back().begin();
deba@425
   758
               it != _sets.back().end(); ++it) {
deba@425
   759
            Node n = _first[*it];
deba@425
   760
            while (n != INVALID) {
deba@425
   761
              _min_cut_map->set(n, true);
deba@425
   762
              n = (*_next)[n];
deba@425
   763
            }
deba@425
   764
          }
deba@425
   765
        }
deba@425
   766
deba@425
   767
        {
deba@425
   768
          Node new_target;
deba@425
   769
          if ((*_prev)[target] != INVALID || (*_next)[target] != INVALID) {
deba@425
   770
            if ((*_next)[target] == INVALID) {
deba@425
   771
              _last[(*_bucket)[target]] = (*_prev)[target];
deba@425
   772
              new_target = (*_prev)[target];
deba@425
   773
            } else {
deba@425
   774
              _prev->set((*_next)[target], (*_prev)[target]);
deba@425
   775
              new_target = (*_next)[target];
deba@425
   776
            }
deba@425
   777
            if ((*_prev)[target] == INVALID) {
deba@425
   778
              _first[(*_bucket)[target]] = (*_next)[target];
deba@425
   779
            } else {
deba@425
   780
              _next->set((*_prev)[target], (*_next)[target]);
deba@425
   781
            }
deba@425
   782
          } else {
deba@425
   783
            _sets.back().pop_back();
deba@425
   784
            if (_sets.back().empty()) {
deba@425
   785
              _sets.pop_back();
deba@425
   786
              if (_sets.empty())
deba@425
   787
                break;
deba@425
   788
              for (std::list<int>::iterator it = _sets.back().begin();
deba@425
   789
                   it != _sets.back().end(); ++it) {
deba@425
   790
                _dormant[*it] = false;
deba@425
   791
              }
deba@425
   792
            }
deba@425
   793
            new_target = _last[_sets.back().back()];
deba@425
   794
          }
deba@425
   795
deba@425
   796
          _bucket->set(target, 0);
deba@425
   797
deba@425
   798
          _source_set->set(target, true);
deba@425
   799
          for (InArcIt a(_graph, target); a != INVALID; ++a) {
deba@425
   800
            Value rem = (*_capacity)[a] - (*_flow)[a];
deba@425
   801
            if (!_tolerance.positive(rem)) continue;
deba@425
   802
            Node v = _graph.source(a);
deba@425
   803
            if (!(*_active)[v] && !(*_source_set)[v]) {
deba@425
   804
              activate(v);
deba@425
   805
            }
deba@425
   806
            _excess->set(v, (*_excess)[v] + rem);
deba@425
   807
            _flow->set(a, (*_capacity)[a]);
deba@425
   808
          }
deba@425
   809
deba@425
   810
          for (OutArcIt a(_graph, target); a != INVALID; ++a) {
deba@425
   811
            Value rem = (*_flow)[a];
deba@425
   812
            if (!_tolerance.positive(rem)) continue;
deba@425
   813
            Node v = _graph.target(a);
deba@425
   814
            if (!(*_active)[v] && !(*_source_set)[v]) {
deba@425
   815
              activate(v);
deba@425
   816
            }
deba@425
   817
            _excess->set(v, (*_excess)[v] + rem);
deba@425
   818
            _flow->set(a, 0);
deba@425
   819
          }
deba@425
   820
deba@425
   821
          target = new_target;
deba@425
   822
          if ((*_active)[target]) {
deba@425
   823
            deactivate(target);
deba@425
   824
          }
deba@425
   825
deba@425
   826
          _highest = _sets.back().begin();
deba@425
   827
          while (_highest != _sets.back().end() &&
deba@425
   828
                 !(*_active)[_first[*_highest]]) {
deba@425
   829
            ++_highest;
deba@425
   830
          }
deba@425
   831
        }
deba@425
   832
      }
deba@425
   833
    }
deba@425
   834
deba@425
   835
  public:
deba@425
   836
deba@425
   837
    /// \name Execution control
deba@425
   838
    /// The simplest way to execute the algorithm is to use
deba@425
   839
    /// one of the member functions called \c run(...).
deba@425
   840
    /// \n
deba@425
   841
    /// If you need more control on the execution,
deba@425
   842
    /// first you must call \ref init(), then the \ref calculateIn() or
deba@425
   843
    /// \ref calculateIn() functions.
deba@425
   844
deba@425
   845
    /// @{
deba@425
   846
deba@425
   847
    /// \brief Initializes the internal data structures.
deba@425
   848
    ///
deba@425
   849
    /// Initializes the internal data structures. It creates
deba@425
   850
    /// the maps, residual graph adaptors and some bucket structures
deba@425
   851
    /// for the algorithm.
deba@425
   852
    void init() {
deba@425
   853
      init(NodeIt(_graph));
deba@425
   854
    }
deba@425
   855
deba@425
   856
    /// \brief Initializes the internal data structures.
deba@425
   857
    ///
deba@425
   858
    /// Initializes the internal data structures. It creates
deba@425
   859
    /// the maps, residual graph adaptor and some bucket structures
deba@425
   860
    /// for the algorithm. Node \c source  is used as the push-relabel
deba@425
   861
    /// algorithm's source.
deba@425
   862
    void init(const Node& source) {
deba@425
   863
      _source = source;
deba@425
   864
deba@425
   865
      _node_num = countNodes(_graph);
deba@425
   866
deba@425
   867
      _first.resize(_node_num + 1);
deba@425
   868
      _last.resize(_node_num + 1);
deba@425
   869
deba@425
   870
      _dormant.resize(_node_num + 1);
deba@425
   871
deba@425
   872
      if (!_flow) {
deba@425
   873
        _flow = new FlowMap(_graph);
deba@425
   874
      }
deba@425
   875
      if (!_next) {
deba@425
   876
        _next = new typename Digraph::template NodeMap<Node>(_graph);
deba@425
   877
      }
deba@425
   878
      if (!_prev) {
deba@425
   879
        _prev = new typename Digraph::template NodeMap<Node>(_graph);
deba@425
   880
      }
deba@425
   881
      if (!_active) {
deba@425
   882
        _active = new typename Digraph::template NodeMap<bool>(_graph);
deba@425
   883
      }
deba@425
   884
      if (!_bucket) {
deba@425
   885
        _bucket = new typename Digraph::template NodeMap<int>(_graph);
deba@425
   886
      }
deba@425
   887
      if (!_excess) {
deba@425
   888
        _excess = new ExcessMap(_graph);
deba@425
   889
      }
deba@425
   890
      if (!_source_set) {
deba@425
   891
        _source_set = new SourceSetMap(_graph);
deba@425
   892
      }
deba@425
   893
      if (!_min_cut_map) {
deba@425
   894
        _min_cut_map = new MinCutMap(_graph);
deba@425
   895
      }
deba@425
   896
deba@425
   897
      _min_cut = std::numeric_limits<Value>::max();
deba@425
   898
    }
deba@425
   899
deba@425
   900
deba@425
   901
    /// \brief Calculates a minimum cut with \f$ source \f$ on the
deba@425
   902
    /// source-side.
deba@425
   903
    ///
deba@425
   904
    /// Calculates a minimum cut with \f$ source \f$ on the
deba@425
   905
    /// source-side (i.e. a set \f$ X\subsetneq V \f$ with \f$ source
deba@425
   906
    /// \in X \f$ and minimal out-degree).
deba@425
   907
    void calculateOut() {
deba@425
   908
      findMinCutOut();
deba@425
   909
    }
deba@425
   910
deba@425
   911
    /// \brief Calculates a minimum cut with \f$ source \f$ on the
deba@425
   912
    /// target-side.
deba@425
   913
    ///
deba@425
   914
    /// Calculates a minimum cut with \f$ source \f$ on the
deba@425
   915
    /// target-side (i.e. a set \f$ X\subsetneq V \f$ with \f$ source
deba@425
   916
    /// \in X \f$ and minimal out-degree).
deba@425
   917
    void calculateIn() {
deba@425
   918
      findMinCutIn();
deba@425
   919
    }
deba@425
   920
deba@425
   921
deba@425
   922
    /// \brief Runs the algorithm.
deba@425
   923
    ///
deba@425
   924
    /// Runs the algorithm. It finds nodes \c source and \c target
deba@425
   925
    /// arbitrarily and then calls \ref init(), \ref calculateOut()
deba@425
   926
    /// and \ref calculateIn().
deba@425
   927
    void run() {
deba@425
   928
      init();
deba@425
   929
      calculateOut();
deba@425
   930
      calculateIn();
deba@425
   931
    }
deba@425
   932
deba@425
   933
    /// \brief Runs the algorithm.
deba@425
   934
    ///
deba@425
   935
    /// Runs the algorithm. It uses the given \c source node, finds a
deba@425
   936
    /// proper \c target and then calls the \ref init(), \ref
deba@425
   937
    /// calculateOut() and \ref calculateIn().
deba@425
   938
    void run(const Node& s) {
deba@425
   939
      init(s);
deba@425
   940
      calculateOut();
deba@425
   941
      calculateIn();
deba@425
   942
    }
deba@425
   943
deba@425
   944
    /// @}
deba@425
   945
deba@425
   946
    /// \name Query Functions
deba@425
   947
    /// The result of the %HaoOrlin algorithm
deba@425
   948
    /// can be obtained using these functions.
deba@425
   949
    /// \n
deba@425
   950
    /// Before using these functions, either \ref run(), \ref
deba@425
   951
    /// calculateOut() or \ref calculateIn() must be called.
deba@425
   952
deba@425
   953
    /// @{
deba@425
   954
deba@425
   955
    /// \brief Returns the value of the minimum value cut.
deba@425
   956
    ///
deba@425
   957
    /// Returns the value of the minimum value cut.
deba@425
   958
    Value minCutValue() const {
deba@425
   959
      return _min_cut;
deba@425
   960
    }
deba@425
   961
deba@425
   962
deba@425
   963
    /// \brief Returns a minimum cut.
deba@425
   964
    ///
deba@425
   965
    /// Sets \c nodeMap to the characteristic vector of a minimum
deba@425
   966
    /// value cut: it will give a nonempty set \f$ X\subsetneq V \f$
deba@425
   967
    /// with minimal out-degree (i.e. \c nodeMap will be true exactly
deba@425
   968
    /// for the nodes of \f$ X \f$).  \pre nodeMap should be a
deba@425
   969
    /// bool-valued node-map.
deba@425
   970
    template <typename NodeMap>
deba@425
   971
    Value minCutMap(NodeMap& nodeMap) const {
deba@425
   972
      for (NodeIt it(_graph); it != INVALID; ++it) {
deba@425
   973
        nodeMap.set(it, (*_min_cut_map)[it]);
deba@425
   974
      }
deba@425
   975
      return _min_cut;
deba@425
   976
    }
deba@425
   977
deba@425
   978
    /// @}
deba@425
   979
deba@425
   980
  }; //class HaoOrlin
deba@425
   981
deba@425
   982
deba@425
   983
} //namespace lemon
deba@425
   984
deba@425
   985
#endif //LEMON_HAO_ORLIN_H