// -*- C++ -*- //

 /*

 Union-Find structure

 * Ehhez nagyon fontos, hogy az atadott property map inicializalva legyen

 * minden szobajovo kulcs ertekre, -1 -es ertekkel!

 */

 #ifndef UNION_FIND_H

 #define UNION_FIND_H

 #include <vector>

 //#include <map>

 namespace hugo {

   template <typename KeyType, typename KeyIntMap>

     class UnionFind {

     KeyIntMap& pointmap;

     struct VectorElementType {

       int boss;

       int count;

       VectorElementType(int _boss, int _count){

 	boss = _boss;

 	count = _count;

       }

     };

     std::vector<VectorElementType> container;

     public:

     UnionFind(KeyIntMap& _pointmap): pointmap(_pointmap){

     } 

     //Give a component of one point to the structure

     int addPoint(KeyType u){

       int _index = container.size();

       VectorElementType buf(_index,1);

       container.push_back(buf);

       return _index;

     }

     //Finds the big boss of u

     int find(KeyType u){

       if (pointmap.get(u)==-1){

 	int whoami = addPoint(u);

 	pointmap.set(u, whoami);

 	return whoami;

       }

       else{

 	int emp = pointmap.get(u);

 	int boss = container[emp].boss;

 	while(emp != boss){

 	  emp = boss;

 	  boss = container[emp].boss;

 	}

 	return boss;

       }

     }

     //Finds u and v in the structures and merges the comopnents, if not equal

     bool findAndMerge(KeyType u,KeyType v){

       int bu = find(u);

       int bv = find(v);

       if (bu != bv){

 	unio(bu,bv);

 	return true;

       }

       else{

 	return false;

       }

     }

     //Merges a into b

     void mergeInto(int a, int b){

       container[a].boss = b;

       container[b].count +=  container[a].count;

     }

     //Makes the union

     void unio(int b1, int b2){

       if (container[b1].count>container[b2].count){

 	mergeInto(b2,b1);

       }

       else{

 	mergeInto(b1,b2);

       }

     }//unio

   };

 }//namespace hugo

 #endif