/* -*- C++ -*-

  *

  * This file is a part of LEMON, a generic C++ optimization library

  *

  * Copyright (C) 2003-2007

  * Egervary Jeno Kombinatorikus Optimalizalasi Kutatocsoport

  * (Egervary Research Group on Combinatorial Optimization, EGRES).

  *

  * Permission to use, modify and distribute this software is granted

  * provided that this copyright notice appears in all copies. For

  * precise terms see the accompanying LICENSE file.

  *

  * This software is provided "AS IS" with no warranty of any kind,

  * express or implied, and with no claim as to its suitability for any

  * purpose.

  *

  */

 #ifndef LEMON_MIN_MEAN_CYCLE_H

 #define LEMON_MIN_MEAN_CYCLE_H

 /// \ingroup min_cost_flow

 ///

 /// \file 

 /// \brief Karp algorithm for finding a minimum mean cycle.

 #include <lemon/graph_utils.h>

 #include <lemon/topology.h>

 #include <lemon/path.h>

 namespace lemon {

   /// \addtogroup min_cost_flow

   /// @{

   /// \brief Implementation of Karp's algorithm for finding a

   /// minimum mean (directed) cycle.

   ///

   /// The \ref lemon::MinMeanCycle "MinMeanCycle" implements Karp's

   /// algorithm for finding a minimum mean (directed) cycle.

   ///

   /// \param Graph The directed graph type the algorithm runs on.

   /// \param LengthMap The type of the length (cost) map.

   ///

   /// \author Peter Kovacs

 #ifdef DOXYGEN

   template <typename Graph, typename LengthMap>

 #else

   template <typename Graph,

     typename LengthMap = typename Graph::template EdgeMap<int> >

 #endif

   class MinMeanCycle

   {

     typedef typename Graph::Node Node;

     typedef typename Graph::NodeIt NodeIt;

     typedef typename Graph::Edge Edge;

     typedef typename Graph::EdgeIt EdgeIt;

     typedef typename Graph::OutEdgeIt OutEdgeIt;

     typedef typename LengthMap::Value Length;

     typedef typename Graph::template NodeMap<int> IntNodeMap;

     typedef typename Graph::template NodeMap<Edge> PredNodeMap;

     typedef Path<Graph> Path;

     typedef std::vector<Node> NodeVector;

     typedef typename NodeVector::iterator NodeVectorIt;

   protected:

     /// \brief Data sturcture for path data.

     struct PathData

     {

       bool found;

       Length dist;

       Edge pred;

       PathData(bool _found = false, Length _dist = 0) :

 	found(_found), dist(_dist), pred(INVALID) {}

       PathData(bool _found, Length _dist, Edge _pred) :

 	found(_found), dist(_dist), pred(_pred) {}

     };

   private:

     typedef typename Graph::template NodeMap<std::vector<PathData> >

       PathDataNodeMap;

   protected:

     /// \brief Node map for storing path data.

     /// 

     /// Node map for storing path data of all nodes in the current

     /// component. dmap[v][k] is the length of a shortest directed walk 

     /// to node v from the starting node containing exactly k edges.

     PathDataNodeMap dmap;

     /// \brief The directed graph the algorithm runs on.

     const Graph& graph;

     /// \brief The length of the edges.

     const LengthMap& length;

     /// \brief The total length of the found cycle.

     Length cycle_length;

     /// \brief The number of edges in the found cycle.

     int cycle_size;

     /// \brief A node for obtaining a minimum mean cycle. 

     Node cycle_node;

     /// \brief The found cycle.

     Path *cycle_path;

     /// \brief The algorithm uses local \ref lemon::Path "Path" 

     /// structure to store the found cycle.

     bool local_path;

     /// \brief Node map for identifying strongly connected components.

     IntNodeMap comp;

     /// \brief The number of strongly connected components.

     int comp_num;

     /// \brief Counter for identifying the current component.

     int comp_cnt;

     /// \brief Nodes of the current component.

     NodeVector nodes;

     /// \brief The processed nodes in the last round.

     NodeVector process;

   public :

     /// \brief The constructor of the class.

     ///

     /// The constructor of the class.

     ///

     /// \param _graph The directed graph the algorithm runs on.

     /// \param _length The length (cost) of the edges.

     MinMeanCycle( const Graph& _graph,

 		  const LengthMap& _length ) :

       graph(_graph), length(_length), dmap(_graph), comp(_graph),

       cycle_length(0), cycle_size(-1), cycle_node(INVALID),

       cycle_path(NULL), local_path(false)

     { }

     /// \brief The destructor of the class.

     ~MinMeanCycle() { 

       if (local_path) delete cycle_path;

     }

   protected:

     /// \brief Initializes the internal data structures for the current

     /// component.

     void initCurrent() {

       nodes.clear();

       // Finding the nodes of the current component

       for (NodeIt v(graph); v != INVALID; ++v) {

 	if (comp[v] == comp_cnt) nodes.push_back(v);

       }

       // Creating vectors for all nodes

       int n = nodes.size();

       for (NodeVectorIt vi = nodes.begin(); vi != nodes.end(); ++vi) {

 	dmap[*vi].resize(n + 1);

       }

     }

     /// \brief Processes all rounds of computing required path data for 

     /// the current component.

     void processRounds() {

       dmap[nodes[0]][0] = PathData(true, 0);

       process.clear();

       // Processing the first round

       for (OutEdgeIt e(graph, nodes[0]); e != INVALID; ++e) {

 	Node v = graph.target(e);

 	if (comp[v] != comp_cnt || v == nodes[0]) continue;

 	dmap[v][1] = PathData(true, length[e], e);

 	process.push_back(v);

       }

       // Processing other rounds 

       int n = nodes.size(), k;

       for (k = 2; k <= n && process.size() < n; ++k) {

 	processNextBuildRound(k);

       }

       for ( ; k <= n; ++k) {

 	processNextFullRound(k);

       }

     }

     /// \brief Processes one round of computing required path data and

     /// rebuilds \ref process vector.

     void processNextBuildRound(int k) {

       NodeVector next;

       for (NodeVectorIt ui = process.begin(); ui != process.end(); ++ui) {

 	for (OutEdgeIt e(graph, *ui); e != INVALID; ++e) {

 	  Node v = graph.target(e);

 	  if (comp[v] != comp_cnt) continue;

 	  if (!dmap[v][k].found) {

 	    next.push_back(v);

 	    dmap[v][k] = PathData(true, dmap[*ui][k-1].dist + length[e], e);

 	  }

 	  else if (dmap[*ui][k-1].dist + length[e] < dmap[v][k].dist) {

 	    dmap[v][k] = PathData(true, dmap[*ui][k-1].dist + length[e], e);

 	  }

 	}

       }

       process.swap(next);

     }

     /// \brief Processes one round of computing required path data

     /// using \ref nodes vector instead of \ref process vector.

     void processNextFullRound(int k) {

       for (NodeVectorIt ui = nodes.begin(); ui != nodes.end(); ++ui) {

 	for (OutEdgeIt e(graph, *ui); e != INVALID; ++e) {

 	  Node v = graph.target(e);

 	  if (comp[v] != comp_cnt) continue;

 	  if ( !dmap[v][k].found || 

 	       dmap[*ui][k-1].dist + length[e] < dmap[v][k].dist ) {

 	    dmap[v][k] = PathData(true, dmap[*ui][k-1].dist + length[e], e);

 	  }

 	}

       }

     }

     /// \brief Finds the minimum cycle mean value in the current 

     /// component.

     bool findCurrentMin(Length &min_length, int &min_size, Node &min_node) {

       bool found_min = false;

       for (NodeVectorIt vi = nodes.begin(); vi != nodes.end(); ++vi) {

 	int n = nodes.size();

 	if (!dmap[*vi][n].found) continue;

 	Length len;

 	int size;

 	bool found_one = false;

 	for (int k = 0; k < n; ++k) {

 	  if (!dmap[*vi][k].found) continue;

 	  Length _len = dmap[*vi][n].dist - dmap[*vi][k].dist;

 	  int _size = n - k; 

 	  if (!found_one || len * _size < _len * size) {

 	    found_one = true;

 	    len = _len;

 	    size = _size;

 	  }

 	}

 	if ( found_one && 

 	     (!found_min || len * min_size < min_length * size) ) {

 	  found_min = true;

 	  min_length = len;

 	  min_size = size;

 	  min_node = *vi;

 	}

       }

       return found_min;

     }

   public:  

     /// \brief Runs the algorithm.

     ///

     /// Runs the algorithm.

     ///

     /// \return \c true if a cycle exists in the graph.

     ///

     /// \note Apart from the return value, m.run() is just a shortcut 

     /// of the following code.

     /// \code

     ///   m.init();

     ///   m.findMinMean();

     ///   m.findCycle();

     /// \endcode

     bool run() {

       init();

       findMinMean();

       return findCycle();

     }

     /// \brief Initializes the internal data structures.

     void init() {

       comp_num = stronglyConnectedComponents(graph, comp);

       if (!cycle_path) {

 	local_path = true;

 	cycle_path = new Path;

       }

     }

     /// \brief Finds the minimum cycle mean value in the graph.

     ///

     /// Computes all the required path data and finds the minimum cycle

     /// mean value in the graph.

     ///

     /// \return \c true if a cycle exists in the graph.

     /// 

     /// \pre \ref init() must be called before using this function.

     bool findMinMean() {

       cycle_node = INVALID;

       for (comp_cnt = 0; comp_cnt < comp_num; ++comp_cnt) {

 	initCurrent();

 	processRounds();

 	Length min_length;

 	int min_size;

 	Node min_node;

 	bool found_min = findCurrentMin(min_length, min_size, min_node);

 	if ( found_min && (cycle_node == INVALID || 

 	     min_length * cycle_size < cycle_length * min_size) ) {

 	  cycle_length = min_length;

 	  cycle_size = min_size;

 	  cycle_node = min_node;

 	}

       }

       return (cycle_node != INVALID);

     }

     /// \brief Finds a critical (minimum mean) cycle.

     ///

     /// Finds a critical (minimum mean) cycle using the path data

     /// stored in \ref dmap.

     ///

     /// \return \c true if a cycle exists in the graph.

     /// 

     /// \pre \ref init() and \ref findMinMean() must be called before 

     /// using this function.

     bool findCycle() {

       if (cycle_node == INVALID) return false;

       cycle_length = 0;

       cycle_size = 0;

       IntNodeMap reached(graph, -1);

       int r = reached[cycle_node] = dmap[cycle_node].size() - 1;

       Node u = graph.source(dmap[cycle_node][r].pred);

       while (reached[u] < 0) {

 	reached[u] = --r;

 	u = graph.source(dmap[u][r].pred);

       }

       r = reached[u];

       Edge e = dmap[u][r].pred;

       cycle_path->addFront(e);

       cycle_length = cycle_length + length[e];

       ++cycle_size;

       Node v;

       while ((v = graph.source(e)) != u) {

 	e = dmap[v][--r].pred;

 	cycle_path->addFront(e);

 	cycle_length = cycle_length + length[e];

 	++cycle_size;

       }

       return true;

     }

     /// \brief Resets the internal data structures.

     ///

     /// Resets the internal data structures so that \ref findMinMean() 

     /// and \ref findCycle() can be called again (e.g. when the 

     /// underlaying graph has been modified).

     void reset() {

       for (NodeIt u(graph); u != INVALID; ++u)

 	dmap[u].clear();

       cycle_node = INVALID;

       if (cycle_path) cycle_path->clear();

       comp_num = stronglyConnectedComponents(graph, comp);

     }

     /// \brief Returns the total length of the found cycle.

     ///

     /// Returns the total length of the found cycle.

     ///

     /// \pre \ref run() must be called before using this function.

     Length cycleLength() const { 

       return cycle_length;

     }

     /// \brief Returns the number of edges in the found cycle.

     ///

     /// Returns the number of edges in the found cycle.

     ///

     /// \pre \ref run() must be called before using this function.

     int cycleEdgeNum() const { 

       return cycle_size;

     }

     /// \brief Returns the mean length of the found cycle.

     ///

     /// Returns the mean length of the found cycle.

     ///

     /// \pre \ref run() must be called before using this function.

     ///

     /// \warning LengthMap::Value must be convertible to double.

     ///

     /// \note m.minMean() is just a shortcut of the following code.

     /// \code

     ///   return m.cycleEdgeNum() / double(m.cycleLength());

     /// \endcode

     double minMean() const { 

       return cycle_length / (double)cycle_size;

     }

     /// \brief Returns a const reference to the \ref lemon::Path "Path"

     /// structure of the found cycle.

     ///

     /// Returns a const reference to the \ref lemon::Path "Path"

     /// structure of the found cycle.

     ///

     /// \pre \ref run() must be called before using this function.

     ///

     /// \sa \ref cyclePath()

     const Path &cycle() const {

       return *cycle_path;

     }

     /// \brief Sets the \ref lemon::Path "Path" structure storing the 

     /// found cycle.

     /// 

     /// Sets the \ref lemon::Path "Path" structure storing the found 

     /// cycle. If you don't use this function before calling 

     /// \ref run(), it will allocate one. The destuctor deallocates 

     /// this automatically allocated map, of course.

     ///

     /// \note The algorithm calls only the \ref lemon::Path::addFront()

     /// "addFront()" method of the given \ref lemon::Path "Path" 

     /// structure.

     /// 

     /// \return \c (*this)

     MinMeanCycle &cyclePath(Path& path) {

       if (local_path) {

 	delete cycle_path;

 	local_path = false;

       }

       cycle_path = &path;

       return *this;

     }

   }; //class MinMeanCycle

   ///@}

 } //namespace lemon

 #endif //LEMON_MIN_MEAN_CYCLE_H