[백준] 4386번 별자리 만들기

💡문제접근

• 선을 하나 이을 때마다 두 별 사이의 거리만큼의 비용이 발생한다.
• 문자열 포맷팅을 절대/상대 오차 10^(-2)까지 허용하도록 출력

💡코드(메모리 : 33376KB, 시간 : 52ms)

import math
import sys
input = sys.stdin.readline

def find_parent(parent, x):
    if parent[x] != x:
        parent[x] = find_parent(parent, parent[x])
    return parent[x]

def union_parent(parent, a, b):
    a = find_parent(parent, a)
    b = find_parent(parent, b)
    if a < b:
        parent[b] = a
    else:
        parent[a] = b

n = int(input())
parent = [0] * (n+1)
stars = []
edges = []
result = 0

for i in range(1, n+1):
    parent[i] = i

for _ in range(n):
    x, y = map(float, input().strip().split())
    stars.append([x, y])

for i in range(n-1):
    for j in range(i+1, n):
        distance = math.sqrt((stars[i][0] - stars[j][0]) ** 2 + (stars[i][1] - stars[j][1]) ** 2)
        edges.append([distance, i, j])
edges.sort()

for edge in edges:
    cost, a, b = edge
    if find_parent(parent, a) != find_parent(parent, b):
        union_parent(parent, a, b)
        result += cost
print("{:.2f}".format(result))

💡소요시간 : 41m