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 |
Tags
- 힙
- 가상메모리
- 프로그래머스
- dict
- 가상메모리 관리
- 그리디
- level0
- 스택
- BOJ
- MYSQL
- 수학
- 코딩테스트
- level3
- 다익스트라
- programmers
- level2
- dfs
- 브루트포스
- DP
- level1
- python
- 다이나믹 프로그래밍
- 구현
- 파이썬
- N과M
- BFS
- 재귀
- 운영체제
- 딕셔너리
- 백준
Archives
- Today
- Total
동캄의 코딩도장
백준 1504 [특정한 최단 경로] 파이썬 본문
https://www.acmicpc.net/problem/1504
# 백준 1504 특정한 최단 경로
import sys
import heapq
from collections import defaultdict
input = sys.stdin.readline
N, E = map(int, input().split())
graph = defaultdict(list)
for _ in range(E):
a, b, cost = map(int, input().split())
graph[a].append([b, cost])
graph[b].append([a, cost])
def dijkstra(graph, start):
distances = {node: float('inf') for node in range(1, N+1)}
distances[start] = 0
queue = []
heapq.heappush(queue, [distances[start], start])
while queue:
curr_dist, curr_dest = heapq.heappop(queue)
if distances[curr_dest] < curr_dist:
continue
for new_dest, new_dist in graph[curr_dest]:
dist = curr_dist+new_dist
if dist < distances[new_dest]:
distances[new_dest] = dist
heapq.heappush(queue, [dist, new_dest])
return distances
v1, v2 = map(int, input().split())
distance_start = dijkstra(graph, 1)
distance_v1 = dijkstra(graph, v1)
distance_v2 = dijkstra(graph, v2)
answer = float('inf')
v1_v2 = distance_start[v1]+distance_v1[v2]+distance_v2[N]
v2_v1 = distance_start[v2]+distance_v2[v1]+distance_v1[N]
answer = min(answer, v1_v2, v2_v1)
if answer == float('inf'):
print(-1)
else:
print(answer)
다익스트라를 이용한 문제이다.
start, v1 , v2를 기준으로 하는 distance set을 얻고, 이를 이용해
start-v1-v2-N
start-v2-v1-N
중 더 최솟값을 가지는 경로를 선택한다.
'코테 > BOJ' 카테고리의 다른 글
백준 14503 [로봇 청소기] 파이썬 (0) | 2022.05.18 |
---|---|
백준 2206 [벽 부수고 이동하기] 파이썬 (0) | 2022.05.18 |
백준 9935 [문자열 폭발] 파이썬 (0) | 2022.05.17 |
백준 11909 [배열 탈출] 파이썬 (0) | 2022.05.17 |
백준 14284 [간선 이어가기2] 파이썬 (0) | 2022.05.17 |