n명의 친구와 m개의 친구 관계도, k원이 있을 때 모든 친구들을 살 수 있는 최소 금액을 출력하며 살 수 없을 시 문자열을 출력하는 문제. 유니온 파인드 방법을 이용하여 풀었다. union 연산 시에 입력받는 친구의 가격을 비교하여 값이 작은 친구를 부모로 저장하였다. 그 후 모든 친구들의 부모를 set 자료형에 담은 후 해당 친구들의 값을 더하여 비교 후 출력하여 풀었다.
\nimport sys\ninput = sys.stdin.readline\n\ndef find(node):\n if parents[node] == node:\n return node\n\n p = find(parents[node])\n parents[node] = p\n return p\n\ndef union(a, b):\n pa, pb = find(a), find(b)\n\n if pa != pb:\n if pay[pa] < pay[pb]:\n parents[pb] = pa\n else:\n parents[pa] = pb\n\nn, m, k = map(int, input().split())\npay = [0] + list(map(int, input().split()))\nparents = [i for i in range(n+1)]\n\nfor _ in range(m):\n a, b = map(int, input().split())\n union(a, b)\n\nset_parents = set()\nfor i in range(1, n+1):\n set_parents.add(find(i))\n\nans = sum(map(lambda x: pay[x], set_parents))\nprint(ans if ans <= k else \"Oh no\")