union find 예제코드

parent 찾아가서 그룹여부를 판단하는 코드 

 

using namespace std;
int parent[1000005];
 
int find(int x) {
 
if (parent[x] == x)
return x;
 
 
int n = 0;;
 
n = find(parent[x]);
return parent[x] = n;
}
 
 
void link(int a, int b) {
 
 
int pa = find(a);
int pb = find(b);
 
if (pa == pb) return;
 
parent[pb] = pa;
 
 
 
}