并查集

1.将两个集合合并

2.询问两个元素是否在一个集合当中

 

基本原理,每个集合用一个树表示,树根的编号就是集合的编号,每个节点存储它的父节点,p[x]表示x的父节点

问题1.如何判断树根:if(p[x]==x)

问题2:如何求x的集合编号 while(p[x]!=x) x=p[x];

问题3:如何合并两个集合:px是x的集合编号,py是y的集合编号。p[x]=y

#include <iostream>
#include <stdio.h>
using namespace std;

const int N=10010;
int m,n;
int p[N];
int find(int x)
{
if(p[x]!=x) p[x]=find(p[x]);
return p[x];
}
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++) p[i]=i;
while(m--)
{
char op;
cin>>op;
int a,b;
cin>>a>>b;
if(op=='M') p[find(a)]=find(b);
else
{
if(find(a)==find(b)) puts("Yes");
else puts("NO");
}
}
return 0;
}