并查集

来源:互联网 发布:windows找不到cdm 编辑:程序博客网 时间:2024/06/08 00:30
在一个有N个元素的集合问题中,我们通常是让每个元素构成一个单元素的集合,然后按一定顺序将属于同一集合的元素进行合并。在此过程中要不断的查询某个元素归属于哪个集合,适合于描述这类问题的抽象数据类型我们称之为并查集。
例:
 已知有n个人和m对好友关系,并且这些好友关系存储在r里面,如果两个人是直接或间接的好友,则认为他们属于同一个朋友圈。
  n=5,m=3,r={{1,2},{2,3},{4,5}}求出这n个人里面一共有多少个朋友圈。


 要求上面题中朋友圈的个数,只需要求出并查集中负数的个数就行了,最后别忘了减去0这个位置就行了。

实现:
[cpp] view plain copy
  1. class UnionSet  
  2. {  
  3. public:  
  4.        UnionSet(int n)  
  5.               :_size(n+1)  
  6.        {  
  7.               _a = new int[n+1];  
  8.               memset(_a,-1,sizeof(int)*(n+1));   //memset只有在0和-1的时候1个字节与4个字节所得到的结果是相同的  
  9.        }  
  10.        ~UnionSet()  
  11.        {  
  12.               delete[] _a;  
  13.        }  
  14.        void Merge(int x1,int x2)  
  15.        {  
  16.               int root1 = GetRoot(x1);  
  17.               int root2 = GetRoot(x2);  
  18.               if (root1 != root2)           //如果这两个元素不再一个集合中  
  19.               {  
  20.                      _a[root1] += _a[root2];    //把root2的值加到根节点处  
  21.                      _a[root2] = root1;         //让root2处存放根节点的下标  
  22.               }  
  23.        }  
  24.        int CountSet()  
  25.        {  
  26.               int count = 0;  
  27.               for (size_t i = 0; i < _size; ++i)  
  28.               {  
  29.                      if (_a[i] < 0)  
  30.                            count++;  
  31.               }  
  32.               return count - 1;  
  33.        }  
  34. protected:  
  35.        int GetRoot(int index)  
  36.        {  
  37.               int root = index;  
  38.               while (_a[root] >= 0)  
  39.               {  
  40.                      root = _a[root];  
  41.               }  
  42.               return root;  
  43.        }  
  44. private:  
  45.        int *_a;  
  46.        size_t _size;  
  47. };  

原创粉丝点击