글 읽기 - 파이썬 고수님들 질문드려요!

omy5123 3년 전 0

예제는 맞다고 나오는데요 ,, 혹시 반례나 문제가 있을까요?

import sys
from heapq import heappush, heappop

n, m = map(int, input().split())
def dijkstra(start,s,dp,heap): #처음 다익스트라 알고리즘 
    dp[start] = 0
    heappush(heap, [0, start])
    while heap:
        a, b = heappop(heap)
        for n_n, wei in s[b]:
            n_w = a + wei
            if dp[n_n] > n_w:
                dp[n_n] = n_w
                heappush(heap, [n_w, n_n])
def dijkstra_1(start,s,dp,heap): #최단경로를 빼고 다시 다익스트라
    global mi
    dp[start] = 0
    heappush(heap, [0, start])
    while heap:
        a, b = heappop(heap)
        for n_n, wei in s[b]:
            n_w = a + wei
            if dp[n_n] > n_w:
                if n_w == mi:
                    continue
                dp[n_n] = n_w
                heappush(heap, [n_w, n_n])
while n != 0 and m != 0:
    inf = sys.maxsize
    arr = [[] for i in range(n)]
    dp = [inf for i in range(n)]
    heap = []
    s,d = map(int,input().split())
    for i in range(m):
        u,v,p = map(int,input().split())
        arr[u].append([v,p])
    dijkstra(s,arr,dp,heap)
    mi = dp[d]
    dp = [inf for i in range(n)]
    heap = []
    dijkstra_1(s,arr,dp,heap)
    if dp[d] == inf:
        print(-1)
    else:
        print(dp[d])
    n, m = map(int, input().split())

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

문제

문제

출처

ICPC

파이썬 고수님들 질문드려요!

omy5123 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말