用途
合并两个集合
询问两个元素是否在一个集合中
原理
1.用树来维护每一个集合
示例
一共有n个数,编号是1~n,最开始每个数各自在一个集合中。
现在要进行m个操作,操作共有两种:
- “M a b”,将编号为a和b的两个数所在的集合合并,如果两个数已经在同一个集合中,则忽略这个操作;
- “Q a b”,询问编号为a和b的两个数是否在同一个集合中;
输入格式
第一行输入整数n和m。
接下来m行,每行包含一个操作指令,指令为“M a b”或“Q a b”中的一种。
输出格式
对于每个询问指令”Q a b”,都要输出一个结果,如果a和b在同一集合内,则输出“Yes”,否则输出“No”。
每个结果占一行。
数据范围
1≤n,m≤1051≤n,m≤105
输入样例:
4 5
M 1 2
M 3 4
Q 1 2
Q 1 3
Q 3 4
输出样例:
Yes
No
Yes
#include <iostream>
using namespace std;
const int N = 100010;
int p[N];//N代表节点编号,p[N]里存的是父亲的节点编号
//返回祖宗节点加路径压缩
int find(int x)
{
if (p[x] != x) p[x] = find(p[x]);//如果节点编号和他爹的编号不一样,就找他爹的爹,直到找到他祖宗
return p[x];
}
int main()
{
int n, m;
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; i ++ ) p[i] = i;
while (m -- )
{
char op[2];
int a, b;
scanf("%s%d%d", op, &a, &b);
if (*op == 'M') p[find(a)] = find(b);//a的祖宗节点的父亲等于b的祖宗节点
else
{
if (find(a) == find(b)) puts("Yes");
else puts("No");
}
}
return 0;
}
示例2
给定一个包含n个点(编号为1~n)的无向图,初始时图中没有边。
现在要进行m个操作,操作共有三种:
- “C a b”,在点a和点b之间连一条边,a和b可能相等;
- “Q1 a b”,询问点a和点b是否在同一个连通块中,a和b可能相等;
- “Q2 a”,询问点a所在连通块中点的数量;
输入格式
第一行输入整数n和m。
接下来m行,每行包含一个操作指令,指令为“C a b”,“Q1 a b”或“Q2 a”中的一种。
输出格式
对于每个询问指令”Q1 a b”,如果a和b在同一个连通块中,则输出“Yes”,否则输出“No”。
对于每个询问指令“Q2 a”,输出一个整数表示点a所在连通块中点的数量
每个结果占一行。
数据范围
1≤n,m≤1051≤n,m≤105
输入样例:
5 5
C 1 2
Q1 1 2
Q2 1
C 2 5
Q2 5
输出样例:
Yes
2
3
#include <iostream>
using namespace std;
const int N = 100010;
int n, m;
int p[N], size[N];//size表示每一个集合中点的数量,只看根节点的size
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;
size[i] = 1;//最初状态集合里只有一个点
}
while (m -- )
{
string op;
int a, b;
cin >> op;
if (op == "C")//两个点间添加一条边
{
cin >> a >> b;
a = find(a), b = find(b);
if (a != b)//不在一个集合时,合并集合,在一个集合就不用操作了
{
p[a] = b;
size[b] += size[a];//合并后size相加
}
}
else if (op == "Q1")
{
cin >> a >> b;
if (find(a) == find(b)) puts("Yes");
else puts("No");
}
else
{
cin >> a;
cout << size[find(a)] << endl;
}
}
return 0;
}