백준/ Gold 2 문제 , 백준 파이썬 2211 , 네트워크 복구 [다익스트라 알고리즘]

백준/ Gold 2 문제 , 백준 파이썬 2211 , 네트워크 복구 [다익스트라 알고리즘]

 

풀이 시간

 

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

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

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

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

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

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

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


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


<덧붙일 말>
어렵겠다 싶었는데 나름 한번에 통과했음. result 라는 배열을 따로 만들어서 각 노드에 연결되는 이전 노드 번호들을 저장해주었고 2번 인덱스 부터 확인해가면서 값이 담겨있다면 해당 자리 번호와 연결된 이전 노드번호를 같이 출력시켜주면 된다.

 

 

문제 출처

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

2211번: 네트워크 복구

 

 

 

 

 

정답

 

 

from cmath import inf
import heapq
import sys
input=sys.stdin.readline

N,M=map(int,input().split())
graph=[[] for _ in range(N+1)]
for i in range(M):
    a,b,c=map(int,input().split())
    graph[a].append((c,b,a))
    graph[b].append((c,a,b))
    
q=[(0,1,0)]
dist=[inf]*(N+1)
dist[1]=0
result=[0]*(N+1)

while q:
    Z,X,Y=heapq.heappop(q)
    if dist[X]<Z:
        continue
    for cost,next,now in graph[X]:
        cost+=Z
        if cost<dist[next]:
            dist[next]=cost
            result[next]=now
            heapq.heappush(q,[cost,next,now])
            
cnt=0
for i in range(2,N+1):
    if result[i]!=0:
        cnt+=1

print(cnt)
for j in range(2,N+1):
    if result[j]!=0:
        print(j,result[j])