백준/ Gold 4 문제 , 백준 파이썬 16398 , 행성 연결 [최소 스패닝 트리]

백준/ Gold 4 문제 , 백준 파이썬 16398 , 행성 연결 [최소 스패닝 트리]

 

 

풀이 시간

 

Check Point !   ( 해당사항 ✓체크 )

1. 막힘 없이 수월하게 풀린 문제인가?

2. 1시간이내로 풀렸던 문제인가?

3. 1시간 이상 or 며칠을 두고 풀어봤더니 풀린 문제인가?

4. 시간을 써도 도무지 풀 수 없는 문제인가?

5. 솔루션을 찾아봤는가?

-------------------------------------------------------------------------------------------
난이도 체감
1. 최상
2. 상
3. 중
4. 하


<이해도>
1. 완벽히 이해
2. 다소 헷갈리는 부분들이 있음
3. 이해 못함


<덧붙일 말>
행렬의 대각선 기준 위쪽만 확인하면됨.

 

 

문제 출처

https://www.acmicpc.net/problem/16398

16398번: 행성 연결

 

 

풀이

 

 

이전 문제들이랑 다른건 딱히 없고 입력 받는 형식이 행렬로 들어오기 때문에 인덱스 만 주의해서 받아주면 된다. 

 

편의를 위해 여기선 0번 인덱스부터 할당을 해주었다.

 

 

 

정답

 

import heapq

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

graph=[[] for _ in range(N)]
visited=[True]+[False]*(N-1)

for i in range(N):
    for j in range(i+1,N):
        graph[i].append((matrix[i][j],j))
        graph[j].append((matrix[i][j],i))

q=graph[0]
heapq.heapify(q)
result=0

while q:
    Z,X=heapq.heappop(q)
    if not visited[X]:
        result+=Z
        visited[X]=True
        for i in graph[X]:
            if not visited[i[1]]:
                heapq.heappush(q,i)
                
print(result)