并查集(集合合并,路径压缩优化)
作者:互联网
并查集(路径压缩优化)
摘自acwing算法模板
并查集
并查集的作用:
1.两个集合合并
2.询问两个集合是否在同一个集合中
怎么实现路径压缩?
如果x不是祖宗结点,就让父亲结点 = 祖宗结点 , 最后返回父亲结点
怎么实现集合合并
让a祖宗的结点的父亲等于b结点的结点
代码
#include<iostream>
using namespace std ;
const int N = 100010 ;
int 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);
else
{
if(find(a) == find(b)) puts("Yes");
else puts("No");
}
}
return 0;
}
标签:结点,路径,int,压缩,查集,集合 来源: https://www.cnblogs.com/rdisheng/p/16536342.html