Notice
Recent Posts
Recent Comments
Link
일 | 월 | 화 | 수 | 목 | 금 | 토 |
---|---|---|---|---|---|---|
1 | 2 | 3 | 4 | 5 | ||
6 | 7 | 8 | 9 | 10 | 11 | 12 |
13 | 14 | 15 | 16 | 17 | 18 | 19 |
20 | 21 | 22 | 23 | 24 | 25 | 26 |
27 | 28 | 29 | 30 | 31 |
Tags
- DFS
- 구현
- 트리
- Python
- 정수론
- 해시 테이블
- 세그먼트 트리
- BFS
- 누적 합
- 플로이드-워셜
- SSAFY
- 문자열
- JavaScript
- 수학
- 정렬
- 브루트포스
- DP
- 에라토스테네스의 체
- 13164
- 맵
- 싸피
- 투 포인터
- boj
- 슬라이딩 윈도우
- 모던 JavaScript 튜토리얼
- 애드 혹
- 그리디
- 2357
- 그래프
- 이분 탐색
Archives
- Today
- Total
흙금이네 블로그
[BOJ] 9370 - 미확인 도착지 (Python) 본문
아이디어
다익스트라를 이용해 목적지 후보에 도착하는 최단 시간과 g와 h 교차로를 거쳐 도착하는 최단 시간을 비교한다.
풀이
import sys
from heapq import heappush, heappop
input = sys.stdin.readline
INF = int(1e9)
def solution():
T = int(input())
for _ in range(T):
n, m, t = map(int, input().split())
s, g, h = map(int, input().split())
graph = [[] for _ in range(n+1)]
for _ in range(m):
a, b, d = map(int, input().split())
graph[a].append((b, d))
graph[b].append((a, d))
X = [int(input()) for _ in range(t)]
D = dict()
res = []
for i in [s, g, h]:
D[i] = [INF]*(n+1)
D[i][i] = 0
heap = [(0, i)]
while heap:
d, idx = heappop(heap)
if D[i][idx] < d:
continue
for j, _d in graph[idx]:
if d+_d < D[i][j]:
D[i][j] = d+_d
heappush(heap, (d+_d, j))
for x in X:
min_d = min(D[s][g]+D[g][h]+D[h][x], D[s][h]+D[h][g]+D[g][x])
if D[s][x] == min_d:
res.append(x)
print(*sorted(res))
solution()
'알고리즘' 카테고리의 다른 글
[BOJ] 17387 - 선분 교차 2 (Python) (0) | 2023.04.08 |
---|---|
[BOJ] 2186 - 문자판 (Python) (0) | 2023.04.07 |
[BOJ] 4991 - 로봇 청소기 (Python) (0) | 2023.04.05 |
[BOJ] 2668 - 숫자고르기 (Python) (0) | 2023.04.03 |
[BOJ] 14938 - 서강그라운드 (Python) (0) | 2023.04.03 |
Comments