백준/ Gold 2 문제 , 백준 파이썬 9370 , 미확인 도착지 [다익스트라 알고리즘]

백준/ Gold 2 문제 , 백준 파이썬 9370 , 미확인 도착지 [다익스트라 알고리즘]

 

풀이 시간

 

 

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

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

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

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

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

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

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


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


<덧붙일 말>
처음에 문제 이해를 잘못해서 로직을 다시 짜느라 시간이 좀 더 걸렸다.
처음에 생각한건 GH나 HG를 무조건 지난다음에 각 목적지의 최단거리를 비교해서 가장 작은 값들을 서로 비교해줬는데, 이렇게 하는게 아니고 각 목적지의 최단거리 (HG랑 GH를 지나는지를 생각 안하고 구한 최단거리)랑 GH나 HG를 지났을때의 최단거리가 서로 같은지를 비교해주는 것이다. 서로 같다면 교차로를 지난 것이니 옳은 목적지 후보이다.

 

 

문제 출처

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

9370번: 미확인 도착지

 

 

 

 

 

 

정답

 

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

def dijkstra(start,end):
    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 cost < dist[next]:
                dist[next]=cost
                heapq.heappush(q,[cost,next])
    return dist[end]

for i in range(int(input())):
    N,M,T=map(int,input().split())
    S,G,H=map(int,input().split())
    graph=[[] for _ in range(N+1)]
    for j in range(M):
        a,b,c=map(int,input().split())
        graph[a].append((c,b))
        graph[b].append((c,a))
    target=[ [int(input()),inf] for _ in range(T)]
    for n in range(T):
        finish=target[n][0]
        target[n][1]=dijkstra(S,finish)
    
    target_G=dijkstra(S,G)
    target_H=dijkstra(S,H)
    target_GH=dijkstra(G,H)
    target_HG=dijkstra(H,G)
    answer=[]
    for k in range(T):
        if target[k][1]==inf:
            continue
        finish=target[k][0]
        HG=dijkstra(G,finish)
        GH=dijkstra(H,finish)
        check=False
        if target_G+target_GH+GH==target[k][1]:
            check=True
        if target_H+target_HG+HG==target[k][1]:
            check=True
        if check==True:
            answer.append(target[k][0])
    
    answer.sort()
    print(*answer)