간선 비용을 좌푯값을 통해 직접 계산하는 문제. 입력된 순서를 그대로 노드 번호로 사용하자. 이미 연결된 간선은 비용이 0으로 간주, 우선순위 큐에 넣으면 자동으로 맨 앞에서 출력된다.
import sys
import heapq
n, m = map(int, sys.stdin.readline().rstrip().split())
pq = []
positions = []
parents = [i for i in range(n)]
def get_distance(x1, y1, x2, y2):
return ((x1-x2)**2 + (y1-y2)**2)**0.5
def find(node):
if parents[node] == node: return node
else:
parents[node] = find(parents[node])
return parents[node]
def union(node1, node2):
root1, root2 = find(node1), find(node2)
if root1 == root2: return False
else:
parents[root2] = root1
return True
def Kruskal():
total = 0
edge_cnt = 0
while pq:
cost, node1, node2 = heapq.heappop(pq)
if union(node1, node2):
total += cost
edge_cnt += 1
if edge_cnt == n-1:
return total
return -1
for i in range(n):
x, y = map(int, sys.stdin.readline().rstrip().split())
positions.append([x, y, i])
# 순서까지 기억: 인덱스 포함
for _ in range(m):
node1, node2 = map(int, sys.stdin.readline().rstrip().split())
node1 -= 1
node2 -= 1
heapq.heappush(pq, [0, node1, node2])
# 연결 간선: 비용 0으로 간주
for i in range(n-1):
x1, y1, node1 = positions[i]
for j in range(i+1, n):
x2, y2, node2 = positions[j]
distance = get_distance(x1, y1, x2, y2)
heapq.heappush(pq, [distance, node1, node2])
answer = Kruskal()
print(f"{answer:0.2f}")