https://www.acmicpc.net/problem/1753
1753번: 최단경로
첫째 줄에 정점의 개수 V와 간선의 개수 E가 주어진다. (1≤V≤20,000, 1≤E≤300,000) 모든 정점에는 1부터 V까지 번호가 매겨져 있다고 가정한다. 둘째 줄에는 시작 정점의 번호 K(1≤K≤V)가 주어진다.
www.acmicpc.net
import sys
import heapq
inf = sys.maxsize
v, e = map(int, sys.stdin.readline().split())
g = [[] for _ in range(v + 1)]
k = int(sys.stdin.readline())
for i in range(e):
a, b, c = map(int, sys.stdin.readline().split())
g[a].append([c, b])
result = [inf for _ in range(v + 1)]
result[k] = 0
q = []
heapq.heappush(q, [0, k])
while q:
dis, end = heapq.heappop(q)
for d, x in g[end]:
d += dis
if d < result[x]:
result[x] = d
heapq.heappush(q, [d, x])
for i in range(1, v + 1):
print(result[i] if result[i] != inf else "INF")
다익스트라 알고리즘에 대하여 공부하고, 백준에서 문제를 풀었습니다. 다익스트라를 좀더 공부할수 있는 계기가 되었지만 생각보다 알고리즘을 알고있어도 푸는데 어려움이 많았던것 같습니다
'2021하계모각코' 카테고리의 다른 글
모각코 4회차 결과 (0) | 2021.08.04 |
---|---|
모각코 4회차 목표 (0) | 2021.08.04 |
모각코 3회차 목표 (0) | 2021.07.28 |
모각코 2회차 결과 (0) | 2021.07.21 |
모각코 2회차 목표 (0) | 2021.07.21 |