Baekjoon #2887

Baekjoon #2887

1개의 풀이 · Python

문제 원문 보기 ↗

SOLUTION INFO

Python · main.py

main.py
import sys
import heapq

def input():
    return sys.stdin.readline().rstrip()

def find_parent(p):
    if p_parents[p] != p:
        p_parents[p] = find_parent(p_parents[p])
    return p_parents[p]

def union_planet(p1, p2):
    p1_parent = find_parent(p1)
    p2_parent = find_parent(p2)
    if p1_parent == p2_parent:
        return False
    if p1_parent < p2_parent:
        p_parents[p2_parent] = p1_parent
    else:
        p_parents[p1_parent] = p2_parent
    return True

N = int(input())
planets = []
p_parents = [i for i in range(N)]
edges = []

for i in range(N):
    p_info = list(map(int, input().split()))
    planets.append(p_info+[i])

for sort_n in range(3):
    planets.sort(key=lambda x: x[sort_n])
    for i in range(N - 1):
        p1x, p1 = planets[i][sort_n], planets[i][3]
        p2x, p2 = planets[i + 1][sort_n], planets[i + 1][3]
        heapq.heappush(edges, [ abs(p1x - p2x), p1, p2 ])

answer = 0
cnt = 0
while edges:
    cost, p1, p2 = heapq.heappop(edges)
    if union_planet(p1, p2):
        answer += cost
        cnt += 1
        if cnt == N - 1:
            break
print(answer)

SOLUTION DESCRIPTION

풀이 설명

등록된 풀이 설명이 없습니다.

GITHUB COMMUNITY

커뮤니티 평가

GitHub 계정으로 남긴 최신 평가 1건만 반영하며, 데이터는 매일 저장소에 동기화됩니다.

최근 동기화

체감 난이도

아직 평가 없음

첫 난이도 평가를 남겨주세요.

문제 추천

아직 평가 없음

이 문제가 도움이 되었는지 알려주세요.

이 문제 평가하기

GitHub 로그인 후 난이도를 제출하고, 원하는 경우 추천 여부도 함께 남길 수 있습니다.

추천 여부 (선택)

평가하려면 GitHub로 로그인해 주세요.

평가는 자동으로 저장소 Discussion에 기록되므로 Discussion 화면을 직접 열 필요가 없습니다. 같은 문제를 다시 평가하면 기존 평가가 갱신됩니다.

DISCUSSION

댓글

GitHub 로그인 후 작성할 수 있으며 모든 댓글은 이 저장소의 Discussions에 보관됩니다.