1753번: 최단경로
첫째 줄에 정점의 개수 V와 간선의 개수 E가 주어진다. (1≤V≤20,000, 1≤E≤300,000) 모든 정점에는 1부터 V까지 번호가 매겨져 있다고 가정한다. 둘째 줄에는 시작 정점의 번호 K(1≤K≤V)가 주어진다.
www.acmicpc.net
아이디어:
다익스트라 알고리즘을 활용하여 푸는 기본적인 문제이다.
다익스트라 알고리즘은 최단경로를 구하는데 사용하는 알고리즘이다.
시작정점으로부터 거리가 가장 작은 정점부터 방문하여 최소경로를 계속 갱신하는 방식이다.
이론은 이해가 가지만 구현하는 게 아직 익숙하지 않아서 어려웠다.
힙을 사용하여 시작정점으로부터의 거리를 저장하는 방식과, 이미 방문한 노드를 거르는 조건에 주의하자.
import sys, heapq
V, E = map(int, sys.stdin.readline().split())
K = int(sys.stdin.readline())
graph = [[] for _ in range(V+1)]
INF = 99999999
ans = [INF]*(V+1)
for _ in range(E):
u, v, w = map(int, sys.stdin.readline().split())
graph[u].append([v,w])
def solve(start):
ans[start] = 0
h = []
heapq.heappush(h, [ans[start], start])
while(h):
now_dist, now_start = heapq.heappop(h)
if ans[now_start]<now_dist:
continue #이미 방문해서 최소경로를 갱신한 노드를 거르기 위해
for v, w in graph[now_start]:
dist = now_dist + w
if (dist<ans[v]):
ans[v] = dist
heapq.heappush(h, [dist, v])
solve(K)
for i in range(1, V+1):
if (ans[i]==INF): print('INF')
else: print(ans[i])
'알고리즘💻 > 우선순위큐&다익스트라' 카테고리의 다른 글
BOJ 11279번: 최대 힙 (0) | 2021.02.15 |
---|---|
BOJ 20046번: Road Reconstruction (0) | 2021.02.13 |
BOJ 1916번: 최소비용 구하기 (0) | 2021.02.13 |
BOJ 11000번: 강의실 배정 (0) | 2021.02.13 |
BOJ 11286번: 절댓값 힙 (0) | 2021.02.13 |