n개의 노드가 있을 때 각 노드들이 연결되어 있는 지 여부를 출력 및 합연산을 하는 유니온 파인드 문제. 기존 유니온 파인드로 풀 수 있는 문제들을 플로이드 와샬, DFSBFS 등의 방법으로 풀었으나 유니온 파인드 방법도 알아두면 좋을 것 같아 공부해보았다. 유니온 파인드는 주어진 노드 또는 집합을 합하는 Union과 노드의 루트 노드가 무엇인 지 반환하는 Find로 나뉜다. 자세한 사항은 주석을 참고. 해당 문제는 간단한 유니온 파인드의 구현으로 풀었다.
\nimport sys\ninput = sys.stdin.readline\n\ndef find(node):\n # 현재 노드가 루트 노드일 시\n if node == parent[node]:\n # 루트 노드 값을 반환\n return node\n\n # 루트 노드가 아닐 시 재귀적으로 탐색\n p = find(parent[node])\n # 루트 노드를 현재 node 값에 저장\n parent[node] = p\n return p # 해당 루트 노드 값을 반환\n\n # 루트 노드가 아닐 시 탐색한 후 할당, 반환하는 과정을 경로 압축 최적화라고 한다\n # return find(parent[u])와 같이 작성시 경로 압축 최적화가 안되었다고 할 수 있는데\n # 이는 단순히 배열에 할당하지 않기 때문인가 ?\n # 아무튼 사용 이유\u001d는 한 쪽으로 치우쳐진 트리일 때 걸리는 시간이라고 한다.\n # 문제를 더 풀어가며 이해해야할 부분 같다.\n\n\ndef union(a, b):\n # a, b의 루트 노드를 찾는다.\n parent_a = find(a)\n parent_b = find(b)\n\n # 루트노드가 갖지 않을 시 b의 루트 노드에 a 루트 노드 값을 저장\n # a의 루트 노드에 b 루트 노드 값을 저장해도 된다.\n if parent_a != parent_b:\n parent[parent_b] = parent_a\n\n\nn, m = map(int, input().split())\n# 각 노드의 루트 노드를 저장할 배열\n# 처음에는 자기 자신을 루트 노드로 생성\nparent = [i for i in range(n+1)]\n\nfor _ in range(m):\n c, a, b = map(int, input().split())\n if c == 0:\n union(a, b)\n else:\n parent_a = find(a)\n parent_b = find(b)\n print(\"YES\" if parent_a == parent_b else \"NO\")