[백준 2211] 네트워크 복구

Junyoung Park·2022년 3월 13일
0

코딩테스트

목록 보기
257/631
post-thumbnail

1. 문제 설명

네트워크 복구

2. 문제 분석

다익스트라 알고리즘을 통해 시작 노드에서 각 노드까지 최단 경로를 구하는 데 사용하는 모든 간선의 집합을 구하자.

3. 나의 풀이

import sys
import heapq

n, m = map(int, sys.stdin.readline().rstrip().split())
nodes = [[] for _ in range(n+1)]
for _ in range(m):
    a, b, c = map(int, sys.stdin.readline().rstrip().split())
    nodes[a].append([b, c])
    nodes[b].append([a, c])
INF = sys.maxsize
def Dijkstra(start):
    distances = [INF for _ in range(n+1)]
    path = [0 for _ in range(n+1)]
    distances[start] = 0

    pq = []
    heapq.heappush(pq, [0, start])

    while pq:
        cur_cost, cur_node = heapq.heappop(pq)

        if distances[cur_node] < cur_cost: continue

        for next_node, next_cost in nodes[cur_node]:
            if distances[next_node] > next_cost+cur_cost:
                distances[next_node] = next_cost+cur_cost
                path[next_node] = cur_node
                heapq.heappush(pq, [cur_cost+next_cost, next_node])
    return path

path = Dijkstra(1)

result = set()

for i in range(2, n+1):
    cursor = i

    while cursor != 1:
        result.add(tuple((cursor, path[cursor])))
        cursor = path[cursor]

print(len(result))

for edge in result:
    a, b = edge
    print(a, b)
profile
JUST DO IT

0개의 댓글