/* -*- C++ -*-

  *

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

  *

  * Copyright (C) 2003-2006

  * 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_UNION_FIND_H

 #define LEMON_UNION_FIND_H

 //!\ingroup auxdat

 //!\file

 //!\brief Union-Find data structures.

 //!

 #include <vector>

 #include <list>

 #include <utility>

 #include <algorithm>

 #include <lemon/invalid.h>

 namespace lemon {

   //! \addtogroup auxdat

   //! @{

   /**

    * \brief A \e Union-Find data structure implementation

    *

    * The class implements the \e Union-Find data structure. 

    * The union operation uses rank heuristic, while

    * the find operation uses path compression.

    * This is a very simple but efficient implementation, providing 

    * only four methods: join (union), find, insert and size.

    * For more features see the \ref UnionFindEnum class.

    *

    * It is primarily used in Kruskal algorithm for finding minimal

    * cost spanning tree in a graph.

    * \sa kruskal()

    *

    * \pre The elements are automatically added only if the map 

    * given to the constructor was filled with -1's. Otherwise you

    * need to add all the elements by the \ref insert() method.

    * \bug It is not clear what the constructor parameter is used for.

    */

   template <typename T, typename TIntMap>

   class UnionFind {

   public:

     typedef T ElementType;

     typedef std::pair<int,int> PairType;

   private:

     std::vector<PairType> data;

     TIntMap& map;

   public:

     UnionFind(TIntMap& m) : map(m) {}

     /**

      * \brief Returns the index of the element's component.

      *

      * The method returns the index of the element's component.

      * This is an integer between zero and the number of inserted elements.

      */

     int find(T a)

     {

       int comp0 = map[a];

       if (comp0 < 0) {

 	return insert(a);

       }

       int comp = comp0;

       int next;

       while ( (next = data[comp].first) != comp) {

 	comp = next;

       }

       while ( (next = data[comp0].first) != comp) {

 	data[comp0].first = comp;

 	comp0 = next;

       }

       return comp;

     }

     /**

      * \brief Inserts a new element into the structure.

      *

      * This method inserts a new element into the data structure. 

      *

      * It is not required to use this method:

      * if the map given to the constructor was filled 

      * with -1's then it is called automatically

      * on the first \ref find or \ref join.

      *

      * The method returns the index of the new component.

      */

     int insert(T a)

     {

       int n = data.size();

       data.push_back(std::make_pair(n, 1));

       map.set(a,n);

       return n;

     }

     /**

      * \brief Joining the components of element \e a and element \e b.

      *

      * This is the \e union operation of the Union-Find structure. 

      * Joins the component of element \e a and component of

      * element \e b. If \e a and \e b are in the same component then

      * it returns false otherwise it returns true.

      */

     bool join(T a, T b)

     {

       int ca = find(a);

       int cb = find(b);

       if ( ca == cb ) 

 	return false;

       if ( data[ca].second > data[cb].second ) {

 	data[cb].first = ca;

 	data[ca].second += data[cb].second;

       }

       else {

 	data[ca].first = cb;

 	data[cb].second += data[ca].second;

       }

       return true;

     }

     /**

      * \brief Returns the size of the component of element \e a.

      *

      * Returns the size of the component of element \e a.

      */

     int size(T a)

     {

       int ca = find(a);

       return data[ca].second;

     }

   };

   /*******************************************************/

 #ifdef DEVELOPMENT_DOCS

   /**

    * \brief The auxiliary class for the \ref UnionFindEnum class.

    *

    * In the \ref UnionFindEnum class all components are represented as

    * a std::list. 

    * Items of these lists are UnionFindEnumItem structures.

    *

    * The class has four fields:

    *  - T me - the actual element 

    *  - IIter parent - the parent of the element in the union-find structure

    *  - int size - the size of the component of the element. 

    *            Only valid if the element

    *            is the leader of the component.

    *  - CIter my_class - pointer into the list of components 

    *            pointing to the component of the element.

    *            Only valid if the element is the leader of the component.

    */

 #endif

   template <typename T>

   struct UnionFindEnumItem {

     typedef std::list<UnionFindEnumItem> ItemList;

     typedef std::list<ItemList> ClassList;

     typedef typename ItemList::iterator IIter;

     typedef typename ClassList::iterator CIter;

     T me;

     IIter parent;

     int size;

     CIter my_class;

     UnionFindEnumItem() {}

     UnionFindEnumItem(const T &_me, CIter _my_class): 

       me(_me), size(1), my_class(_my_class) {}

   };

   /**

    * \brief A \e Union-Find data structure implementation which

    * is able to enumerate the components.

    *

    * The class implements a \e Union-Find data structure

    * which is able to enumerate the components and the items in

    * a component. If you don't need this feature then perhaps it's

    * better to use the \ref UnionFind class which is more efficient.

    *

    * The union operation uses rank heuristic, while

    * the find operation uses path compression.

    *

    * \pre You

    * need to add all the elements by the \ref insert() method.

    */

   template <typename T, template <typename Item> class Map>

   class UnionFindEnum {

     typedef std::list<UnionFindEnumItem<T> > ItemList;

     typedef std::list<ItemList> ClassList;

     typedef typename ItemList::iterator IIter;

     typedef typename ItemList::const_iterator IcIter;

     typedef typename ClassList::iterator CIter;

     typedef typename ClassList::const_iterator CcIter;

   public:

     typedef T ElementType;

     typedef UnionFindEnumItem<T> ItemType;

     typedef Map< IIter > MapType;

   private:

     MapType& m;

     ClassList classes;

     IIter _find(IIter a) const {

       IIter comp = a;

       IIter next;

       while( (next = comp->parent) != comp ) {

 	comp = next;

       }

       IIter comp1 = a;

       while( (next = comp1->parent) != comp ) {

 	comp1->parent = comp->parent;

 	comp1 = next;

       }

       return comp;

     }

   public:

     UnionFindEnum(MapType& _m) : m(_m) {}

     /**

      * \brief Inserts the given element into a new component.

      *

      * This method creates a new component consisting only of the

      * given element.

      */

     void insert(const T &a)

     {

       classes.push_back(ItemList());

       CIter aclass = classes.end();

       --aclass;

       ItemList &alist = *aclass;

       alist.push_back(ItemType(a, aclass));

       IIter ai = alist.begin();

       ai->parent = ai;

       m.set(a, ai);

     }

     /**

      * \brief Inserts the given element into the component of the others.

      *

      * This methods inserts the element \e a into the component of the

      * element \e comp. 

      */

     void insert(const T &a, const T &comp) {

       IIter clit = _find(m[comp]);

       ItemList &c = *clit->my_class;

       c.push_back(ItemType(a,0));

       IIter ai = c.end();

       --ai;

       ai->parent = clit;

       m.set(a, ai);

       ++clit->size;

     }

     /**

      * \brief Finds the leader of the component of the given element.

      *

      * The method returns the leader of the component of the given element.

      */

     T find(const T &a) const {

       return _find(m[a])->me;

     }

     /**

      * \brief Joining the component of element \e a and element \e b.

      *

      * This is the \e union operation of the Union-Find structure. 

      * Joins the component of element \e a and component of

      * element \e b. If \e a and \e b are in the same component then

      * returns false else returns true.

      */

     bool join(T a, T b) {

       IIter ca = _find(m[a]);

       IIter cb = _find(m[b]);

       if ( ca == cb ) {

 	return false;

       }

       if ( ca->size > cb->size ) {

 	cb->parent = ca->parent;

 	ca->size += cb->size;

 	ItemList &alist = *ca->my_class;

 	alist.splice(alist.end(),*cb->my_class);

 	classes.erase(cb->my_class);

 	cb->my_class = 0;

       }

       else {

 	ca->parent = cb->parent;

 	cb->size += ca->size;

 	ItemList &blist = *cb->my_class;

 	blist.splice(blist.end(),*ca->my_class);

 	classes.erase(ca->my_class);

 	ca->my_class = 0;

       }

       return true;

     }

     /**

      * \brief Returns the size of the component of element \e a.

      *

      * Returns the size of the component of element \e a.

      */

     int size(const T &a) const {

       return _find(m[a])->size;

     }

     /**

      * \brief Splits up the component of the element. 

      *

      * Splitting the component of the element into sigleton

      * components (component of size one).

      */

     void split(const T &a) {

       IIter ca = _find(m[a]);

       if ( ca->size == 1 )

 	return;

       CIter aclass = ca->my_class;

       for(IIter curr = ca; ++curr != aclass->end(); curr=ca) {

 	classes.push_back(ItemList());

 	CIter nl = --classes.end();

 	nl->splice(nl->end(), *aclass, curr);

 	curr->size=1;

 	curr->parent=curr;

 	curr->my_class = nl;

       }

       ca->size=1;

       return;

     }

     /**

      * \brief Sets the given element to the leader element of its component.

      *

      * Sets the given element to the leader element of its component.

      */

     void makeRep(const T &a) {

       IIter ia = m[a];

       IIter la = _find(ia);

       if (la == ia) return;

       ia->my_class = la->my_class;

       la->my_class = 0;

       ia->size = la->size;

       CIter l = ia->my_class;

       l->splice(l->begin(),*l,ia);

       ia->parent = ia;

       la->parent = ia;

     }

     /**

      * \brief Moves the given element to an other component.

      *

      * This method moves the element \e a from its component

      * to the component of \e comp.

      * If \e a and \e comp are in the same component then

      * it returns false otherwise it returns true.

      */

     bool move(const T &a, const T &comp) {

       IIter ai = m[a];

       IIter lai = _find(ai);

       IIter clit = _find(m[comp]);

       if (lai == clit)

 	return false;

       ItemList &cl = *clit->my_class,

 	&al = *lai->my_class;

       bool is_leader = (lai == ai);

       bool singleton = false;

       if (is_leader) {

 	++lai;

       }

       cl.splice(cl.end(), al, ai);

       if (is_leader) {

 	if (ai->size == 1) {

 	  classes.erase(ai->my_class);

 	  singleton = true;

 	}

 	else {

 	  lai->size = ai->size; 

 	  lai->my_class = ai->my_class;	

 	}

       }

       if (!singleton) {

 	for (IIter i = lai; i != al.end(); ++i)

 	  i->parent = lai;

 	--lai->size;

       }

       ai->parent = clit;

       ai->my_class = 0;

       ++clit->size;

       return true;

     }

     /**

      * \brief Removes the given element from the structure.

      *

      * Removes the given element from the structure.

      *

      * Removes the element from its component and if the component becomes

      * empty then removes that component from the component list.

      */

     void erase(const T &a) {

       IIter ma = m[a];

       if (ma == 0) return;

       IIter la = _find(ma);

       if (la == ma) {

 	if (ma -> size == 1){

 	  classes.erase(ma->my_class);

 	  m.set(a,0);

 	  return;

 	}

 	++la;

 	la->size = ma->size; 

 	la->my_class = ma->my_class;	

       }

       for (IIter i = la; i != la->my_class->end(); ++i) {

 	i->parent = la;

       }

       la->size--;

       la->my_class->erase(ma);

       m.set(a,0);

     }

     /**

      * \brief Removes the component of the given element from the structure.

      *

      * Removes the component of the given element from the structure.

      */

     void eraseClass(const T &a) {

       IIter ma = m[a];

       if (ma == 0) return;

 #     ifdef DEBUG

       CIter c = _find(ma)->my_class;

       for (IIter i=c->begin(); i!=c->end(); ++i)

 	m.set(i->me, 0);

 #     endif

       classes.erase(_find(ma)->my_class);

     }

     class ClassIt {

       friend class UnionFindEnum;

       CcIter i;

     public:

       ClassIt(Invalid): i(0) {}

       ClassIt() {}

       operator const T& () const { 

 	ItemList const &ll = *i;

 	return (ll.begin())->me; }

       bool operator == (ClassIt it) const {

 	return (i == it.i);

       }

       bool operator != (ClassIt it) const {

 	return (i != it.i);

       }

       bool operator < (ClassIt it) const {

 	return (i < it.i);

       }

       bool valid() const { return i != 0; }

     private:

       void first(const ClassList &l) { i = l.begin(); validate(l); }

       void next(const ClassList &l) {

 	++i; 

 	validate(l);

       }

       void validate(const ClassList &l) {

 	if ( i == l.end() ) 

 	  i = 0;

       }

     };

     /**

      * \brief Sets the iterator to point to the first component.

      * 

      * Sets the iterator to point to the first component.

      *

      * With the \ref first, \ref valid and \ref next methods you can

      * iterate through the components. For example:

      * \code

      * UnionFindEnum<Graph::Node, Graph::NodeMap>::MapType map(G);

      * UnionFindEnum<Graph::Node, Graph::NodeMap> U(map);

      * UnionFindEnum<Graph::Node, Graph::NodeMap>::ClassIt iter;

      *  for (U.first(iter); U.valid(iter); U.next(iter)) {

      *    // iter is convertible to Graph::Node

      *    cout << iter << endl;

      *  }

      * \endcode

      */

     ClassIt& first(ClassIt& it) const {

       it.first(classes);

       return it;

     }

     /**

      * \brief Returns whether the iterator is valid.

      *

      * Returns whether the iterator is valid.

      *

      * With the \ref first, \ref valid and \ref next methods you can

      * iterate through the components. See the example here: \ref first.

      */

     bool valid(ClassIt const &it) const {

       return it.valid(); 

     }

     /**

      * \brief Steps the iterator to the next component. 

      *

      * Steps the iterator to the next component.

      *

      * With the \ref first, \ref valid and \ref next methods you can

      * iterate through the components. See the example here: \ref first.

      */

     ClassIt& next(ClassIt& it) const {

       it.next(classes);

       return it;

     }

     class ItemIt {

       friend class UnionFindEnum;

       IcIter i;

       const ItemList *l;

     public:

       ItemIt(Invalid): i(0) {}

       ItemIt() {}

       operator const T& () const { return i->me; }

       bool operator == (ItemIt it) const {

 	return (i == it.i);

       }

       bool operator != (ItemIt it) const {

 	return (i != it.i);

       }

       bool operator < (ItemIt it) const {

 	return (i < it.i);

       }

       bool valid() const { return i != 0; }

     private:

       void first(const ItemList &il) { l=&il; i = l->begin(); validate(); }

       void next() {

 	++i; 

 	validate();

       }

       void validate() {

 	if ( i == l->end() ) 

 	  i = 0;

       }

     };

     /**

      * \brief Sets the iterator to point to the first element of the component.

      * 

      * \anchor first2 

      * Sets the iterator to point to the first element of the component.

      *

      * With the \ref first2 "first", \ref valid2 "valid" 

      * and \ref next2 "next" methods you can

      * iterate through the elements of a component. For example

      * (iterating through the component of the node \e node):

      * \code

      * Graph::Node node = ...;

      * UnionFindEnum<Graph::Node, Graph::NodeMap>::MapType map(G);

      * UnionFindEnum<Graph::Node, Graph::NodeMap> U(map);

      * UnionFindEnum<Graph::Node, Graph::NodeMap>::ItemIt iiter;

      *   for (U.first(iiter, node); U.valid(iiter); U.next(iiter)) {

      *     // iiter is convertible to Graph::Node

      *     cout << iiter << endl;

      *   }

      * \endcode

      */

     ItemIt& first(ItemIt& it, const T& a) const {

       it.first( * _find(m[a])->my_class );

       return it;

     }

     /**

      * \brief Returns whether the iterator is valid.

      *

      * \anchor valid2

      * Returns whether the iterator is valid.

      *

      * With the \ref first2 "first", \ref valid2 "valid" 

      * and \ref next2 "next" methods you can

      * iterate through the elements of a component.

      * See the example here: \ref first2 "first".

      */

     bool valid(ItemIt const &it) const {

       return it.valid(); 

     }

     /**

      * \brief Steps the iterator to the next component. 

      *

      * \anchor next2

      * Steps the iterator to the next component.

      *

      * With the \ref first2 "first", \ref valid2 "valid" 

      * and \ref next2 "next" methods you can

      * iterate through the elements of a component.

      * See the example here: \ref first2 "first".

      */

     ItemIt& next(ItemIt& it) const {

       it.next();

       return it;

     }

   };

   //! @}

 } //namespace lemon

 #endif //LEMON_UNION_FIND_H