并查集-----小米面试题之朋友圈问题

先来看一下题目：

已知有n个人和m个好友关系（存在数字r），如果两个人是直接或者间接的好友（好友的好友的好友……），则认为他们属于同一个朋友圈。请写程序求出这n个人里一共有多少个朋友圈？
例如：n=5，m=3，r={{1,2}，{2,3}，{4,5}}，表示有5个人，1和2是好友，2和3是好友，4和5是好友，则1、2、3属于一个朋友圈，4、5属于一个朋友圈。最终结果就是2个朋友圈

为了追求高效且准确，我们用到了并查集来完成这道题

一、并查集？

并查集就是将n个不同的元素划分成为一组不相交的集合。开始时，每个元素自成一个单元素集合，然后按照一定的规律将归于同一组元素的集合合并。

它是一种树形的数据结构，每个集合以一棵树表示，树的每一个结点代表集合的一个单元素。所有各个集合的全集合构成一个森林，并用树与森林的父指针表示来实现。其下标代表元素名。第i个数组元素代表集合元素i的树结点。树的根结点的下标表示集合名称，根结点的父为-1，表示集合中元素个数

二、图示表示一下
以上面的题的为基础为例：
n=5，m=3，r={{1,2}，{2,3}，{4,5}}，表示有5个人，1和2是好友，2和3是好友，4和5是好友，则1、2、3属于一个朋友圈，4、5属于一个朋友圈。最终结果就是2个朋友圈

集合为：r={1,2,3,4,5}
开始时，每个元素就是一个集合

按照上面所给的要求将集合中的元素合并成为三个子集合
r1 = {1，2}
r2 = {2，3}
r3 = {4，5}

用树状来表示所形成的朋友圈就是：

与此同时，我们将好友关系用数组的方式来记录，因为2是1的朋友，所有就将2的下标叠加到1的下面去，2下标下的数据也存放在的1的下面。3是2的朋友，2是1的朋友，所以就把3的下标叠加到1的下面去，3下标下的数据也存放在的1的下面。4和5的朋友圈如同上述


在同一棵树上的所有节点所表示的集合元素在同一个子集合中。对于任意给定的集合元素x，只要借助这个映射就能找到存放x的树结点，沿此存放x结点的父指针向上一直走到树的根结点，就可以得到x所在集合的名字。

总结：

* （1）只要某一个下标里的数据为负数，则它就是其所在朋友圈的根。负数的大小表示着朋友圈的人数（eg：下标1中的数据为-3，则其所在的朋友圈人数为3）
* （2）数组中有几个负数就表示有几个朋友圈
* （3）可以根据下标中所对应的数字找到根节点，相同根节点的就在同一个朋友圈内

下面来看看代码实现：

#include <iostream>
using namespace std;
#include <vector>
class UnionFriend
{
public:
      // 构造函数（用-1初始化）
      UnionFriend(int size)
            :_set(size, -1)
      {
            // 其他方式
            // _set.resize(size, -1)
            // _set.assign(size, -1)
      }
      // 求交集
      void Union(int x1, int x2)
      {
            // 先找x1、x2的根
            int root1 = FindRoot(x1);
            int root2 = FindRoot(x2);
            //比较x1和x2的根是否相等
            if (root1 != root2)
            {
                  _set[root1] += _set[root2];
                  _set[root2] = root1;
            }
      }
      // 求合并后集合的个数
      size_t count()
      {
            size_t count = 0;
            //统计负数的个数
            for (size_t i = 0; i < _set.size(); i++)
            {
                  if (_set[i] < 0)
                  {
                        count++;
                  }
            }
            return count;
      }
private:
      //查找根节点
      int FindRoot(int index)
      {
            while (_set[index] >= 0)
            {
                  index = _set[index];
            }
            return index;
      }
private:
      //一段连续的空间
      std::vector<int> _set;
};
int friends(const int m, const int n, int r[][2])
{
      //让下标从1开始，符合题目
      UnionFriend u(m + 1);
      for (int i = 0; i < n; i++)
      {
            u.Union(r[i][0], r[i][1]);
      }
      return u.count() - 1;
}
void Test()
{
      int r[][2] = { {1,2},{2,3},{4,5} };
      cout << "朋友圈个数：" << friends(5, 3, r) << endl;
}
int main()
{
      Test();
      system("pause");
      return 0;
}