글 읽기 - 1504 파이썬 key error?? 어떻게 해야 key error가 나는걸까요

nayounsang1 1년 전 0

90%대에서 틀리는거보면 일반적인 경우는 맞는단건데

어떤 데이터여야지 key error가 나올까요?

import heapq
from collections import defaultdict
#inf = 10**10
def stra(g,start):
    dist = {n:10**10 for n in g}
    dist[start] = 0
    q = []
    heapq.heappush(q,(dist[start],start))
    while q:
        d,node = heapq.heappop(q)
        if dist[node] < d:
            continue
        for n,newd in g[node].items():
            tmp = newd + d
            if tmp < dist[n]:
                dist[n] = tmp
                heapq.heappush(q,(tmp,n))
    return dist
g = defaultdict(dict)
n,e = map(int,input().split(' '))
for i in range(e):
    s,e,c = map(int,input().split(' '))
    g[s][e] = c
    g[e][s] = c
e1,e2 = map(int,input().split(' '))
one = stra(g,1)
last = stra(g,n)
const = stra(g,e1)
dist = min(
    one[e1] + const[e2] + last[e2],
    one[e2] + const[e2] + last[e1],
    one[e1] + 2 * const[e2] + last[e1],
    one[e2] + 2 * const[e2] + last[e2]
)
dist = dist if dist < 10**10 else -1
print(dist)

nayounsang1 1년 전 0

참고로 defaultdict가 아닌

댓글 코드처럼 작성했더니 맞다 나오네요

테스트 데이터중에 아예 붕뜬 노드가 있나봐요

g = {}
for i in range(n):
    g[i+1] = {}

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

문제

문제

출처

ICPC

1504 파이썬 key error?? 어떻게 해야 key error가 나는걸까요

nayounsang1 1년 전 0

nayounsang1 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말