글 읽기 - 뭐가 틀린건가요? (파이썬)

luke0422 3년 전 0

인터넷에 떠돌아다니는거랑 비교했을때 과정도 똑같은데... 어디서 잘못된건질 모르겠습니다

import heapq
import sys

input = sys.stdin.readline

INF=float('inf')
g={}
cost={}
h=[]

v,e=map(int,input().split())
start=int(input())

for i in range(v):
    g[i+1]={}        #그래프 저장할 딕셔너리
    cost[i+1]=INF    #cost 저장할 딕셔너리
cost[start]=0       

for i in range(e):
    a,b,c=map(int,input().split())
    g[a][b]=c        #그래프 저장
        
heapq.heappush(h,(0,start))

while h!=[]:
    x=heapq.heappop(h)
    node=x[1]        #x[1]이 heappop한 노드
    for i in g[node]:#x[1]에서 갈수 있는 모든 노드 탐색
        if cost[node]+g[node][i]<cost[i]:    #만약 start->x[1]->i 로 가는게 start->i보다 짧다면
            cost[i]=cost[node]+g[node][i]    #cost 업데이트
            heapq.heappush(h,(cost[i],i))    

for i in cost:
    if cost[i]==INF:
        print('INF')
    else:
        print(cost[i])

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

뭐가 틀린건가요? (파이썬)

luke0422 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말