并查集

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


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

实现:
class UnionSet{public:       UnionSet(int n)              :_size(n+1)       {              _a = new int[n+1];              memset(_a,-1,sizeof(int)*(n+1));   //memset只有在0和-1的时候1个字节与4个字节所得到的结果是相同的       }       ~UnionSet()       {              delete[] _a;       }       void Merge(int x1,int x2)       {              int root1 = GetRoot(x1);              int root2 = GetRoot(x2);              if (root1 != root2)           //如果这两个元素不再一个集合中              {                     _a[root1] += _a[root2];    //把root2的值加到根节点处                     _a[root2] = root1;         //让root2处存放根节点的下标              }       }       int CountSet()       {              int count = 0;              for (size_t i = 0; i < _size; ++i)              {                     if (_a[i] < 0)                           count++;              }              return count - 1;       }protected:       int GetRoot(int index)       {              int root = index;              while (_a[root] >= 0)              {                     root = _a[root];              }              return root;       }private:       int *_a;       size_t _size;};void TestUnionSet(){       const int n = 5;       const int m = 6;       int r[m][2] = { { 1, 2 }, { 2, 3 }, { 4, 5 }, { 1, 3 }, { 5, 4 }, {3,5} };       UnionSet set(n);       for (int i = 0; i < m;i++)       {              set.Merge(r[i][0],r[i][1]);       }       cout << set.CountSet() << endl;}


1 0
原创粉丝点击