学习并查集算法：洛谷P1551

学习并查集算法：洛谷P1551

思路：使用并查集并优化并查集算法复杂度
参考：链接: link.

#include <iostream>
#include <vector>
#include <string>
#include <stdlib.h>
#include <math.h>
#include <map>
using namespace std;
string s;
vector<int> fa,rankk;
void init(int x)//初始化 
{
	fa.push_back(0);
	rankk.push_back(0);
	for(int i=1;i<=x;i++)
	{
		fa.push_back(i);  //将每个数的父节点设为自己 
		rankk.push_back(1); //每个数的深度为1 
	}
}
int find(int x)  //递归找根节点 
{
	if(fa[x]==x)return x;  //当一个数的父节点就是自己时，则找到根节点 
	else
	{
		fa[x] = find(fa[x]);  //将中间节点的父节点更新为它的根节点方便下次查找 
		return fa[x]; 
	}
}
void hebing(int x,int y)//合并两个树 
{
	int i = find(x),j = find(y);//两个数的根节点 
	if(rankk[i]<=rankk[j]) //将深度低的树合并到深度高的树中合并 ，降低合并后树的深度 
	{
		fa[i] = j;     //
	}
	else
	{
		fa[j] = i; 
	}
	if(rankk[i]==rankk[j]&&i!=j)//两个树的深度相同但是根节点不同，则需要将深度加1 
	{
		rankk[j]++;          //由于深度相等时是将i的根节点变为j,所以应该将j的深度加1 
	}
}

int main()
{
	int n,m,p,i,j,x1,x2;
	cin >>n>>m>>p;
	init(n);
	for(i=1;i<=m;i++)
	{
		cin>>x1>>x2;
		hebing(x1,x2); 
	}
	for(i=1;i<=p;i++)
	{
		cin>>x1>>x2;
		if(find(x1)==find(x2))cout << "Yes"<<endl;
		else cout << "No" <<endl;
	} 
	return 0;
}