并查集学习

在一些应用问题中经常会有把不同个体进行归类的场景，可以先初始化，每个个体自成一个种群，然后把不同的群可以合并，可以查找某个个体是在哪个群中，两个个体是否在同一个群中，查看目前场景有几个种群，这些问题都可以使用并查集来实现。使用数组来表示种群非常方便且好理解，初始时数组中每个个体就是用下标表示，下标处存储的值就是它的父节点，也就是一个种群的，规定存储的值如果为负数，那么该下标个体就代表一个群，且绝对值就是这个群的个体数，所以数组初始化都为-1，就是每个个体都表示一个群，数量就是一个即它自己，然后我们在根据场景合并群。如下图

有十个人 0-9 开始各自为群，我们分成三个群s1{0,6,7,8} s2{1,4,9} s3{2,3,5}

集合的树形表示

可以继续合并群体，就是它一个群体下标处的值，修改成另一个群体的下标，然后把那个群体的数量更新一下。

class UnoinFind
{
public:
	UnoinFind(size_t sz)
		:v(sz,-1)
	{}
	int Findroot(int x)
	{
		int parent = x;
		while (v[parent] >= 0)
		{
			parent = v[parent];
		}
		while (v[x] >= 0)    //路径压缩， 上面已经找到了根，把下面的节点每个都直接挂到根下面   
		{
			int prev = v[x];  //记录它原来的双亲
			v[x] = parent;   //现在双亲直接改为根
			x = prev;     //继续把这条路径上的节点双亲都改为根
		}
		return parent;
	}
	void Unionroot(int x1, int x2)
	{
		int root1 = Findroot(x1);
		int root2 = Findroot(x2);
		if (root1 == root2)
			return;
		else
		{
			if (abs(v[root1]) < abs(v[root2])) //这步优化可以选择，把节点少的合并到节点多的上
				swap(root1, root2);
			v[root1] += v[root2];
			v[root2] = root1;
		}
	}
	bool Inset(int x1, int x2)
	{
		return Findroot(x1) == Findroot(x2);
	}
	size_t count()const
	{
		int n = 0;
		for (int i = 0; i < v.size(); ++i)
		{
			if (v[i] < 0)
				++n;
		}
		return n;
	}
private:
	vector<int>v;
};