RhodeCode

    HaoOrlin(const Digraph& graph, const CapacityMap& capacity,

             const Tolerance& tolerance = Tolerance()) :

      _graph(graph), _capacity(&capacity), _flow(0), _source(),

      _node_num(), _first(), _last(), _next(0), _prev(0),

      _active(0), _bucket(0), _dormant(), _sets(), _highest(),

      _excess(0), _source_set(0), _min_cut(), _min_cut_map(0),

      _tolerance(tolerance) {}

    ~HaoOrlin() {

      if (_min_cut_map) {

        delete _min_cut_map;

      }

      if (_source_set) {

        delete _source_set;

      }

      if (_excess) {

        delete _excess;

      }

      if (_next) {

        delete _next;

      }

      if (_prev) {

        delete _prev;

      }

      if (_active) {

        delete _active;

      }

      if (_bucket) {

        delete _bucket;

      }

      if (_flow) {

        delete _flow;

      }

    }

  private:

    void activate(const Node& i) {

      (*_active)[i] = true;

      int bucket = (*_bucket)[i];

      if ((*_prev)[i] == INVALID || (*_active)[(*_prev)[i]]) return;

      //unlace

      (*_next)[(*_prev)[i]] = (*_next)[i];

      if ((*_next)[i] != INVALID) {

        (*_prev)[(*_next)[i]] = (*_prev)[i];

      } else {

        _last[bucket] = (*_prev)[i];

      }

      //lace

      (*_next)[i] = _first[bucket];

      (*_prev)[_first[bucket]] = i;

      (*_prev)[i] = INVALID;

      _first[bucket] = i;

    }

    void deactivate(const Node& i) {

      (*_active)[i] = false;

      int bucket = (*_bucket)[i];

      if ((*_next)[i] == INVALID || !(*_active)[(*_next)[i]]) return;

      //unlace

      (*_prev)[(*_next)[i]] = (*_prev)[i];

      if ((*_prev)[i] != INVALID) {

        (*_next)[(*_prev)[i]] = (*_next)[i];

      } else {

        _first[bucket] = (*_next)[i];

      }

      //lace

      (*_prev)[i] = _last[bucket];

      (*_next)[_last[bucket]] = i;

      (*_next)[i] = INVALID;

      _last[bucket] = i;

    }

    void addItem(const Node& i, int bucket) {

      (*_bucket)[i] = bucket;

      if (_last[bucket] != INVALID) {

        (*_prev)[i] = _last[bucket];

        (*_next)[_last[bucket]] = i;

        (*_next)[i] = INVALID;

        _last[bucket] = i;

      } else {

        (*_prev)[i] = INVALID;

        _first[bucket] = i;

        (*_next)[i] = INVALID;

        _last[bucket] = i;

      }

    }

    void findMinCutOut() {

      for (NodeIt n(_graph); n != INVALID; ++n) {

        (*_excess)[n] = 0;

      }

      for (ArcIt a(_graph); a != INVALID; ++a) {

        (*_flow)[a] = 0;

      }

      int bucket_num = 0;

      std::vector<Node> queue(_node_num);

      int qfirst = 0, qlast = 0, qsep = 0;

      {

        typename Digraph::template NodeMap<bool> reached(_graph, false);

        reached[_source] = true;

        bool first_set = true;

        for (NodeIt t(_graph); t != INVALID; ++t) {

          if (reached[t]) continue;

          _sets.push_front(std::list<int>());

          queue[qlast++] = t;

          reached[t] = true;

          while (qfirst != qlast) {

            if (qsep == qfirst) {

              ++bucket_num;

              _sets.front().push_front(bucket_num);

              _dormant[bucket_num] = !first_set;

              _first[bucket_num] = _last[bucket_num] = INVALID;

              qsep = qlast;

            }

            Node n = queue[qfirst++];

            addItem(n, bucket_num);

            for (InArcIt a(_graph, n); a != INVALID; ++a) {

              Node u = _graph.source(a);

              if (!reached[u] && _tolerance.positive((*_capacity)[a])) {

                reached[u] = true;

                queue[qlast++] = u;

              }

            }

          }

          first_set = false;

        }

        ++bucket_num;

        (*_bucket)[_source] = 0;

        _dormant[0] = true;

      }

      (*_source_set)[_source] = true;

      Node target = _last[_sets.back().back()];

      {

        for (OutArcIt a(_graph, _source); a != INVALID; ++a) {

          if (_tolerance.positive((*_capacity)[a])) {

            Node u = _graph.target(a);

            (*_flow)[a] = (*_capacity)[a];

            (*_excess)[u] += (*_capacity)[a];

            if (!(*_active)[u] && u != _source) {

              activate(u);

            }

          }

        }

        if ((*_active)[target]) {

          deactivate(target);

        }

        _highest = _sets.back().begin();

        while (_highest != _sets.back().end() &&

               !(*_active)[_first[*_highest]]) {

          ++_highest;

        }

      }

      while (true) {

        while (_highest != _sets.back().end()) {

          Node n = _first[*_highest];

          Value excess = (*_excess)[n];

          int next_bucket = _node_num;

          int under_bucket;

          if (++std::list<int>::iterator(_highest) == _sets.back().end()) {

            under_bucket = -1;

          } else {

            under_bucket = *(++std::list<int>::iterator(_highest));

          }

          for (OutArcIt a(_graph, n); a != INVALID; ++a) {

            Node v = _graph.target(a);

            if (_dormant[(*_bucket)[v]]) continue;

            Value rem = (*_capacity)[a] - (*_flow)[a];

            if (!_tolerance.positive(rem)) continue;

            if ((*_bucket)[v] == under_bucket) {

              if (!(*_active)[v] && v != target) {

              ++_highest;

              if (_highest != _sets.back().end() &&

                  !(*_active)[_first[*_highest]]) {

                _highest = _sets.back().end();

              }

            }

          }

        }

        if ((*_excess)[target] < _min_cut) {

          _min_cut = (*_excess)[target];

          for (NodeIt i(_graph); i != INVALID; ++i) {

            (*_min_cut_map)[i] = true;

          }

          for (std::list<int>::iterator it = _sets.back().begin();

               it != _sets.back().end(); ++it) {

            Node n = _first[*it];

            while (n != INVALID) {

              (*_min_cut_map)[n] = false;

              n = (*_next)[n];

            }

          }

        }

        {

          Node new_target;

          if ((*_prev)[target] != INVALID || (*_next)[target] != INVALID) {

            if ((*_next)[target] == INVALID) {

              _last[(*_bucket)[target]] = (*_prev)[target];

              new_target = (*_prev)[target];

            } else {

              (*_prev)[(*_next)[target]] = (*_prev)[target];

              new_target = (*_next)[target];

            }

            if ((*_prev)[target] == INVALID) {

              _first[(*_bucket)[target]] = (*_next)[target];

            } else {

              (*_next)[(*_prev)[target]] = (*_next)[target];

            }

          } else {

            _sets.back().pop_back();

            if (_sets.back().empty()) {

              _sets.pop_back();

              if (_sets.empty())

                break;

              for (std::list<int>::iterator it = _sets.back().begin();

                   it != _sets.back().end(); ++it) {

                _dormant[*it] = false;

              }

            }

            new_target = _last[_sets.back().back()];

          }

          (*_bucket)[target] = 0;

          (*_source_set)[target] = true;

          for (OutArcIt a(_graph, target); a != INVALID; ++a) {

            Value rem = (*_capacity)[a] - (*_flow)[a];

            if (!_tolerance.positive(rem)) continue;

            Node v = _graph.target(a);

            if (!(*_active)[v] && !(*_source_set)[v]) {

              activate(v);

            }

            (*_excess)[v] += rem;

            (*_flow)[a] = (*_capacity)[a];

          }

          for (InArcIt a(_graph, target); a != INVALID; ++a) {

            Value rem = (*_flow)[a];

            if (!_tolerance.positive(rem)) continue;

            Node v = _graph.source(a);

            if (!(*_active)[v] && !(*_source_set)[v]) {

              activate(v);

            }

            (*_excess)[v] += rem;

            (*_flow)[a] = 0;

          }

          target = new_target;

          if ((*_active)[target]) {

            deactivate(target);

          }

          _highest = _sets.back().begin();

          while (_highest != _sets.back().end() &&

                 !(*_active)[_first[*_highest]]) {

            ++_highest;

          }

        }

      }

    }

    void findMinCutIn() {

      for (NodeIt n(_graph); n != INVALID; ++n) {

        (*_excess)[n] = 0;

      }

      for (ArcIt a(_graph); a != INVALID; ++a) {

        (*_flow)[a] = 0;

      }

      int bucket_num = 0;

      std::vector<Node> queue(_node_num);

      int qfirst = 0, qlast = 0, qsep = 0;

      {

        typename Digraph::template NodeMap<bool> reached(_graph, false);

        reached[_source] = true;

        bool first_set = true;

        for (NodeIt t(_graph); t != INVALID; ++t) {

          if (reached[t]) continue;

          _sets.push_front(std::list<int>());

          queue[qlast++] = t;

          reached[t] = true;

          while (qfirst != qlast) {

            if (qsep == qfirst) {

              ++bucket_num;

              _sets.front().push_front(bucket_num);

              _dormant[bucket_num] = !first_set;

              _first[bucket_num] = _last[bucket_num] = INVALID;

              qsep = qlast;

            }

            Node n = queue[qfirst++];

            addItem(n, bucket_num);

            for (OutArcIt a(_graph, n); a != INVALID; ++a) {

              Node u = _graph.target(a);

              if (!reached[u] && _tolerance.positive((*_capacity)[a])) {

                reached[u] = true;

                queue[qlast++] = u;

              }

            }

          }

          first_set = false;

        }

        ++bucket_num;

        (*_bucket)[_source] = 0;

        _dormant[0] = true;

      }

      (*_source_set)[_source] = true;

      Node target = _last[_sets.back().back()];

      {

        for (InArcIt a(_graph, _source); a != INVALID; ++a) {

          if (_tolerance.positive((*_capacity)[a])) {

            Node u = _graph.source(a);

            (*_flow)[a] = (*_capacity)[a];

            (*_excess)[u] += (*_capacity)[a];

            if (!(*_active)[u] && u != _source) {

              activate(u);

            }

          }

        }

        if ((*_active)[target]) {

          deactivate(target);

        }

        _highest = _sets.back().begin();

        while (_highest != _sets.back().end() &&

               !(*_active)[_first[*_highest]]) {

          ++_highest;

        }

      }

      while (true) {

        while (_highest != _sets.back().end()) {

          Node n = _first[*_highest];

          Value excess = (*_excess)[n];

          int next_bucket = _node_num;

          int under_bucket;

          if (++std::list<int>::iterator(_highest) == _sets.back().end()) {

            under_bucket = -1;

          } else {

            under_bucket = *(++std::list<int>::iterator(_highest));

          }

          for (InArcIt a(_graph, n); a != INVALID; ++a) {

            Node v = _graph.source(a);

            if (_dormant[(*_bucket)[v]]) continue;

            Value rem = (*_capacity)[a] - (*_flow)[a];

            if (!_tolerance.positive(rem)) continue;

            if ((*_bucket)[v] == under_bucket) {

#include <sstream>

#include <lemon/smart_graph.h>

#include <lemon/adaptors.h>

#include <lemon/concepts/digraph.h>

#include <lemon/concepts/maps.h>

#include <lemon/lgf_reader.h>

#include <lemon/hao_orlin.h>

#include "test_tools.h"

using namespace lemon;

using namespace std;

const std::string lgf =

  "@nodes\n"

  "label\n"

  "0\n"

  "1\n"

  "2\n"

  "3\n"

  "4\n"

  "5\n"

  "@edges\n"

  "     cap1 cap2 cap3\n"

  "0 1  1    1    1   \n"

  "0 2  2    2    4   \n"

  "1 2  4    4    4   \n"

  "3 4  1    1    1   \n"

  "3 5  2    2    4   \n"

  "4 5  4    4    4   \n"

  "5 4  4    4    4   \n"

  "2 3  1    6    6   \n"

  "4 0  1    6    6   \n";

void checkHaoOrlinCompile()

{

  typedef int Value;

  typedef concepts::Digraph Digraph;

  typedef Digraph::Node Node;

  typedef Digraph::Arc Arc;

  typedef concepts::ReadMap<Arc, Value> CapMap;

  typedef concepts::WriteMap<Node, bool> CutMap;

  Digraph g;

  Node n;

  CapMap cap;

  CutMap cut;

  Value v;

  HaoOrlin<Digraph, CapMap> ho_test(g, cap);

  const HaoOrlin<Digraph, CapMap>&

    const_ho_test = ho_test;

  ho_test.init();

  ho_test.init(n);

  ho_test.calculateOut();

  ho_test.calculateIn();

  ho_test.run();

  ho_test.run(n);

  v = const_ho_test.minCutValue();

  v = const_ho_test.minCutMap(cut);

}

template <typename Graph, typename CapMap, typename CutMap>

typename CapMap::Value 

  cutValue(const Graph& graph, const CapMap& cap, const CutMap& cut)

{

  typename CapMap::Value sum = 0;

  for (typename Graph::ArcIt a(graph); a != INVALID; ++a) {

    if (cut[graph.source(a)] && !cut[graph.target(a)])

      sum += cap[a];

  }

  return sum;

}

int main() {

  SmartDigraph graph;

  SmartDigraph::ArcMap<int> cap1(graph), cap2(graph), cap3(graph);

  SmartDigraph::NodeMap<bool> cut(graph);

  istringstream input(lgf);

  digraphReader(graph, input)

    .arcMap("cap1", cap1)

    .arcMap("cap2", cap2)

    .arcMap("cap3", cap3)

    .run();

  {

    HaoOrlin<SmartDigraph> ho(graph, cap1);

    ho.run();

    ho.minCutMap(cut);

  }

  {

    HaoOrlin<SmartDigraph> ho(graph, cap2);

    ho.run();

    ho.minCutMap(cut);

  }

  {

    HaoOrlin<SmartDigraph> ho(graph, cap3);

    ho.run();

    ho.minCutMap(cut);

  }

  typedef Undirector<SmartDigraph> UGraph;

  UGraph ugraph(graph);

  {

    HaoOrlin<UGraph, SmartDigraph::ArcMap<int> > ho(ugraph, cap1);

    ho.run();

    ho.minCutMap(cut);

  }

  {

    HaoOrlin<UGraph, SmartDigraph::ArcMap<int> > ho(ugraph, cap2);

    ho.run();

    ho.minCutMap(cut);

  }

  {

    HaoOrlin<UGraph, SmartDigraph::ArcMap<int> > ho(ugraph, cap3);

    ho.run();

    ho.minCutMap(cut);

    check(ho.minCutValue() == 5, "Wrong cut value");

  }

  return 0;

}