并查集的使用

目录

一.介绍

二.并查集的实现

三路径压缩

 四.相关题型

4.1省份数量

---

---

一.介绍

什么是并查集？

将n个不同的元素划分成一些不相交的集合。开始时，每个元素自成一个 单元素集合，然后按一定的规律将归于同一组元素的集合合并。在这个过程中要反复用到查询某一 个元素归属于哪个集合的运算。适合于描述这类问题的抽象数据类型称为并查集(union-find set)。

例如：

 一开始共有0-9，共10个数，将它们的下表初始化为-1，表示这10个数各不相关，都属于不同的集合。现在，将它们开始合并，假设是这样合并：

 

 将要合并的两个元素进行如下操作：

1-将要合并的两个元素的值进行相加，将相加的值赋予其中一个元素。

2-将那个未被赋值的元素的值改为那个之前被赋值的元素。

例如：元素0要和元素6进行合并，开始时它们的值都是-1，合并后元素0的值为-2（代表那个集合有两个元素），元素6的值为0，（元素的值不为0，说明它属于某个集合），这里说明元素6属于元素0那个集合。

那么怎么去查找某个元素属于哪个集合，并且知道那个集合的元素个数是多少呢？

例如：

 你要查找元素9,9对应的值是1，不是负数，说明元素9属于元素1那个集合，那么再去查找元素1。当查找元素1时，发现它的值是0，也不是负数，说明元素1属于元素0那个集合，那么再去查找元素0。查找元素0时，发现它的值为-7，是负数，说明9属于元素0的集合，集合一共有7个元素。

合并也很简单，分别找到两个元素的根（元素值为负数的），将它们合并即可。

由此可知并查级可以解决的问题：

1. 查找元素属于哪个集合 沿着数组表示树形关系以上一直找到根(即：树中中元素为负数的位置)

2. 查看两个元素是否属于同一个集合 沿着数组表示的树形关系往上一直找到树的根，如果根相同表明在同一个集合，否则不在

3. 将两个集合归并成一个集合 将两个集合中的元素合并 将一个集合名称改成另一个集合的名称

4. 集合的个数 遍历数组，数组中元素为负数的个数即为集合的个数

二.并查集的实现

class UnionFindSet
{
public:
	UnionFindSet(int n)
	{
		_set.resize(n, -1);
	}
	int find(int x)  //查找某个元素属于哪个集合
	{
		while(_set[x] >= 0)
		{
			int tmp = _set[x];
			x = tmp;
		}
		return x;
	}
	void Union(int x, int y)
	{
		int x = find(x);
		int y = find(y);
		if (x != y)
		{
			int sum = _set[x] + _set[y];
			_set[x] = sum;
			_set[y] = x;
		}
	}
	int SetCount()
	{
		size_t count = 0;
		for (size_t i = 0; i < _set.size(); ++i)
		{
			if (_set[i] < 0)
				count++;
		}
		return count;
	}

private:
	vector<int> _set;
};

三路径压缩

查找元素属于哪个集合 沿着数组表示树形关系以上一直找到根（就是值为负数），如果每次都要查找最低层的那些元素，岂不是每次都要去根吗?所以可以每次去更新元素的值，对路劲进行压缩。例如：先找到根，在将那条路径上的依次元素链接到根上

 压缩后：

 代码实现：

   int find(int x)  //查找某个元素属于哪个集合
	{
		int p = x;
		while(_set[x] >= 0)
		{
			int tmp = _set[x];
			x = tmp;
		}

		while (_set[p] >= 0)//路径压缩
		{
			int tmp = _set[p];
			_set[p] = x;
			p = tmp;
		}
		return x;
	}

 四.相关题型

4.1省份数量

题目描述：

有 n 个城市，其中一些彼此相连，另一些没有相连。如果城市 a 与城市 b 直接相连，且城市 b 与城市 c 直接相连，那么城市 a 与城市 c 间接相连。

省份 是一组直接或间接相连的城市，组内不含其他没有相连的城市。

给你一个 n x n 的矩阵 isConnected ，其中 isConnected[i][j] = 1 表示第 i 个城市和第 j 个城市直接相连，而 isConnected[i][j] = 0 表示二者不直接相连。

返回矩阵中 省份 的数量。

 

 代码示例：

int findCircleNum(vector<vector<int>>& isConnected) {
        UnionFindSet U(isConnected.size());
        int n=isConnected.size();
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<=i;j++)
            {
                if(isConnected[i][j]==1)
                {
                    U.Union(i,j);
                }
            }
        }
        return U.SetCount();

    }