并查集

最新推荐文章于 2026-05-18 20:48:32 发布

原创最新推荐文章于 2026-05-18 20:48:32 发布 · 900 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#并查集

c/c++基础同时被 2 个专栏收录

210 篇文章

订阅专栏

数据结构

59 篇文章

订阅专栏

本文详细介绍了并查集数据结构的实现原理与应用场景，通过一个具体的朋友圈问题实例，展示了如何利用并查集来高效地解决集合合并及查询的问题。

在一个有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;
}