백준/ Gold 4 문제 , 백준 파이썬 1504 , 특정한 최단 경로 [다익스트라 알고리즘]

백준/ Gold 4 문제 , 백준 파이썬 1504 , 특정한 최단 경로 [다익스트라 알고리즘]

 

 

풀이 시간

 

 

 

Check Point !   ( 해당사항 ✓체크 )

1. 막힘 없이 수월하게 풀린 문제인가?

2. 1시간이내로 풀렸던 문제인가?

3. 1시간 이상 or 며칠을 두고 풀어봤더니 풀린 문제인가?

4. 시간을 써도 도무지 풀 수 없는 문제인가?

5. 솔루션을 찾아봤는가?

-------------------------------------------------------------------------------------------
난이도 체감
1. 최상
2. 상
3. 중
4. 하


<이해도>
1. 완벽히 이해
2. 다소 헷갈리는 부분들이 있음
3. 이해 못함


<덧붙일 말>

 

 

문제 출처

https://www.acmicpc.net/problem/1504

1504번: 특정한 최단 경로

 

 

 

풀이

 

두가지 경우를 생각해서 풀어야한다.

 

1. 시작 지점  =>  u 지나기  =>  v 지나기  => 끝지점

 

2. 시작 지점  =>  v 지나기  =>  u 지나기  => 끝지점

 

 

이렇게 두 케이스를 체크해줘야한다.

 

만약 결과값이 inf 가 나온다면 -1을 출력하자. 

 

 

 

 

 

 

 

 

 

 

정답

 

 

 

from cmath import inf
import heapq
import sys
input=sys.stdin.readline

N,M=map(int,input().split())
graph=[[] for _ in range(N+1)]
for i in range(M):
    a,b,c=map(int,input().split())
    graph[a].append((c,b))
    graph[b].append((c,a))
    
u,v=map(int,input().split())
check=[1,u,v]
result_1=0
result_2=0
for i in range(3):
    q=[(0,check[i])]
    dist=[inf]*(N+1)
    dist[check[i]]=0
    while q:
        Z,X=heapq.heappop(q)
        if dist[X]<Z:
            continue
        for cost,next in graph[X]:
            cost=cost+Z
            if dist[next]>cost:
                dist[next]=cost
                heapq.heappush(q,[cost,next])
    if i==0:
        result_1+=dist[u]
        result_2+=dist[v]
    elif i==1:
        result_1+=dist[v]
        result_2+=dist[N]
    elif i==2:
        result_1+=dist[N]
        result_2+=dist[u]

answer=min(result_1,result_2)
if answer==inf:
    print(-1)
else:
    print(answer)