백준/ Gold 4 문제 , 백준 파이썬 23793 , 두 단계 최단 경로 1 [다익스트라 알고리즘]

백준/ Gold 4 문제 , 백준 파이썬 23793 , 두 단계 최단 경로 1 [다익스트라 알고리즘]

 

 

풀이 시간

 

 

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

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

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

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

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

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

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


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


<덧붙일 말>
양방향이 아님에 주의

 

 

문제 출처

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

23793번: 두 단계 최단 경로 1

 

 

풀이

 

하나는 Y 를 무조건 거쳐야 하므로

 

X => Y 로가는 다익스트라랑

 

Y => Z 로 가는 다익스트라를 구해 더해준다.

 

 

두번째는 Y를 지나면 안되므로 Y값을 같이 넣어서 다익스트라를 돌려 Y값이외의 노드들로만 이동시킨다.

 

 

 

 

 

 

정답

 

 

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

def dijkstra(start,end,center):
    q=[(0,start)]
    dist=[inf]*(N+1)
    dist[start]=0

    while q:
        Z,X=heapq.heappop(q)
        if dist[X]<Z:
            continue
        for cost,next in graph[X]:
            cost+=Z
            if next!=center:
                if cost<dist[next]:
                    dist[next]=cost
                    heapq.heappush(q,[cost,next])
    return dist[end]

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))
    
start,center,end=map(int,input().split())
A=dijkstra(start,center,0)
B=dijkstra(center,end,0)
XZ=dijkstra(start,end,center)
XYZ=A+B
if XYZ==inf:
    XYZ=-1
if XZ==inf:
    XZ=-1
print(XYZ,XZ)