3-dhlee777

[dhlee777]

🔗 문제 링크

https://www.acmicpc.net/problem/1717

✔️ 소요된 시간

3시간

✨ 수도 코드

유니온 파인드(uion-find)란 대표적인 그래프 알고리즘으로 특정 두 노드가 같은집합에 속해있는지 판별해주는 알고리즘이다.
크게 특정 두 집합을 합치는 union 연산과 루트노드를 찾는 find 연산으로 이루어진다.
우선 parent 배열은 어떤노드의 부모노드를 저장하는 배열로 최종적으로는 어떤노드의 루트노드가 저장된다.

#include<stdio.h>
#include<iostream>
using namespace std;
int parent[1000001];          //어떤수의 부모노드를 저장하는 배열 ,최종적으로는 그 수의 루트노드가 저장된다.
int find_parent(int a) {        //어떤 노드의 루트노드를 찾는 함수
	if (a == parent[a])  return a;
	else  return parent[a]=find_parent(parent[a]);
}
void add (int a,int b) {     
	a = find_parent(a);      //a의 루트 노드를 찾고
	b = find_parent(b);		//b의 루트노드를 찾아
	if (a == b) return;     //두 수의 루트노드가 같으면 이미 같은집합에 있으므로 함수 종료
	if (a < b) parent[b] = a;   //더 작은 수가 루트가 된다.
	else parent[a] = b;
}
void compare(int a, int b) {
	a = find_parent(a);
	b = find_parent(b);
	if (a == b) cout<<"yes\n";  //두수의 루트가 같으면(같은집합) yes 아니면 no 출력 
	else  cout<<"no\n";
}

int main(void) {
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);

	int num, calculate_num,a,b,c;
	cin >> num >> calculate_num;
	for (int j = 1; j <= num; j++) {
		parent[j] = j; //자신의 루트는 자신이므로 초기화해준다.
	}
	for (int i = 0; i < calculate_num; i++) {
		cin >> a >> b >> c;
		if (!a)  add(b, c);  //0일경우 합집합 연산이므로 두 집합을 합쳐준다.
		else  compare(b, c);  //1일경우 두 수가 같은집합에 있는지 비교
		
	}
	
	return 0;
}

find_parent(int a) 함수는 a의 루트노드를 찾는 함수로 재귀호출을 통해 a의 루트노드를 찾아낸다.
add(int a, int b) 함수는 a집합과 b집합을 합치는 함수로 두 수의 루트노드를 각각 찾아 연결해준다.
compare(int a,int b)함수는 두 수가 같은집합에 속해있는지 각 노드의 루트노드를 통해 확인하는 함수이다.

1. 최대노드값 num,와 연산의 개수 calculate_num을 각각 입력받는다.
2. 자신의 루트는 자신이므로 parent[j] = j을 통해 초기화해준다.
3. 이제 연산들을 진행하는데 만약 0 b c 의 형태일 경우 합집합연산이므로 add()호출
4. 1 b c 형태일 경우 두 수가 같은집합에 있는지 비교해야하므로 compare()호출

📚 새롭게 알게된 내용

제일 처음에 find_parent() 함수의 리턴 부분에서return find_parent(parent[a]);를 했더니 시간초과가 계속 났다.
그의후 return parent[a]=find_parent(parent[a]); 로 parent[a]에 결과를 저장하는 코드를 추가 했더니 시간초과문제가
해결되었다. 아마 이과정을 통해 parent 배열의 값을 부모노드 값에서 루트노드의 값으로 변화시켜주는 과정에서 시간초과가 해결된 것 같다.

[yuyu0830]

일종의 트리 만들기군요! 문제만 읽고 일일이 노드 만들어서 해결하나? 싶었는데 1차원 배열에 부모 값만 저장하면 문제가 없네요! 코드가 읽기 쉬워서 좋았습니다 :)

[tgyuuAn]

코드 블럭에 ```cpp 처럼 언어 명을 명시해주면 코드 블럭 하이라이팅 됩니다!

이번엔 제가 고쳐놨어요!

[dhlee777]

헉 감사합니다 다음부터는 코드블럭 하이라이팅 될수있게 저렇게 하겠습니다..!

[seongwon030]

유니온 파인드는 처음 보네요! 새로운 알고리즘 잘 보고 갑니다.

[tgyuuAn]

유니온 파인드는 처음 보네요! 새로운 알고리즘 잘 보고 갑니다.

😠😠😠

[InSange]

Union find는 연결되어 있는 상단의 공통 부분을 찾아 비교한다는 점에서 처음 봤었을 때 되게 헷갈려 했던 적이 기억이 나더군요..
대부분의 문제는 visited로 방문했는지 안했는지 비교만 해주면 되던 문제들이 주로 나오다가루트를 비교해서 공통점을 찾는 부분이 Union find를 풀게되면서 신선했었던 것 같습니다.
바이러스 해당 문제에서 저는 visited로 풀어줬지만 Union find로도 풀 수 있어서 재밌었습니다. 문제 추천드립니다!
이러한 탐색문제들 중 heap 관련 문제들과 문자열 탐색에서 트라이 문제들이 있는데 한 번 알아보시면 추후에 난이도 있는 문제들을 풀이하시는데 도움이 될 거라 생각합니다!

[dhlee777]

헉..저도 바이러스문제를 푼적이 있었는데 이걸 유니온파인드 로도 접근할수 있었군요..? 트라이라는 자료구조도 있는지 처음 알았네요.. 관련 자료구조에대해 더 공부해보겠습니다. 감사합니다 !