[백준 10282] 해킹

Junyoung Park·2022년 3월 11일
0

코딩테스트

목록 보기
240/631
post-thumbnail

1. 문제 설명

해킹

2. 문제 분석

다익스트라를 통해 시작 노드에서 건너갈 수 있는 노드가 있다면 그 노드에 대한 최단 거리를 구한다. 이때 영향을 미치지 못하는 노드와의 거리는 처음에 설정한 INF로 세팅되어 있다는 데 주의.

3. 나의 풀이

import sys
import heapq

t = int(sys.stdin.readline().rstrip())
INF = sys.maxsize
for _ in range(t):
    n, d, c = map(int, sys.stdin.readline().rstrip().split())
    nodes = [[] for _ in range(n+1)]

    for _ in range(d):
        a, b, s = map(int, sys.stdin.readline().rstrip().split())
        nodes[b].append([a, s])

    def Dijkstra(start):
        distances = [INF 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
                    pq.append([cur_cost+next_cost, next_node])

        unaffected = 0
        local_max = 0

        for distance in distances[1:]:
            if distance == INF: unaffected += 1
            else:
                local_max = max(local_max, distance)

        print(n-unaffected, local_max)

    Dijkstra(c)
profile
JUST DO IT

0개의 댓글