[백준] 1753번: 최단경로 - 파이썬

🔈 문제

방향그래프가 주어지면 주어진 시작점에서 다른 모든 정점으로의 최단 경로를 구하는 프로그램을 작성하시오. 단, 모든 간선의 가중치는 10 이하의 자연수이다.

 

📝입력

첫째 줄에 정점의 개수 V와 간선의 개수 E가 주어진다. (1 ≤ V ≤ 20,000, 1 ≤ E ≤ 300,000) 모든 정점에는 1부터 V까지 번호가 매겨져 있다고 가정한다. 둘째 줄에는 시작 정점의 번호 K(1 ≤ K ≤ V)가 주어진다. 셋째 줄부터 E개의 줄에 걸쳐 각 간선을 나타내는 세 개의 정수 (u, v, w)가 순서대로 주어진다. 이는 u에서 v로 가는 가중치 w인 간선이 존재한다는 뜻이다. u와 v는 서로 다르며 w는 10 이하의 자연수이다. 서로 다른 두 정점 사이에 여러 개의 간선이 존재할 수도 있음에 유의한다.

 

📑출력

첫째 줄부터 V개의 줄에 걸쳐, i번째 줄에 i번 정점으로의 최단 경로의 경로값을 출력한다. 시작점 자신은 0으로 출력하고, 경로가 존재하지 않는 경우에는 INF를 출력하면 된다.

 

📚 문제 풀이

간선에 가중치가 있으니 이 문제는 데이크스트라 문제이다!

이 문제는 단순히 데이크스트라 알고리즘을 사용하여 풀면 되었다. 

 

💻 코드

import sys, heapq
input = sys.stdin.readline

V,E = map(int,input().split()) # 노드와 간선
k=int(input()) # 시작 노드

# 그래프 그리기
graph=[[] for _ in range(V+1)]
for _ in range(E):
    e,v,w=map(int,input().split())
    graph[e].append([w,v]) # [가중치, 노드]

def dijkstra(start):
    INF=1e9
    visited=[INF]*(V+1) # 무한으로 초기화
    visited[start]=0 # 시작 노드 가중치는 0
    q=[[0,start]] # [가중치, 노드]
    while q:
        cost,node=heapq.heappop(q)
        if cost > visited[node]: # 현재 이동하고자 하는 경로의 비용이 이전 비용보다 크다면 가지않음
            continue
        for k,u in graph[node]:
            if visited[u] > visited[node]+k: # 이동 비용이 작다면
                visited[u] = visited[node]+k # 그렇게 이동
                heapq.heappush(q,[visited[u], u]) # 우큐에 삽입
    return visited[1:] # 0번째 인덱스 제외

for ans in dijkstra(k):
    print(ans if ans!=1e9 else 'INF')

 

🍭 시행착오

맞왜틀인줄 알았지만 틀틀이였음

틀린 이유는 시작 정점을 무조건 1로 두었었다..ㅎ

반례 몇개 돌려보는데 이상한 값이 나와 코드를 한 번 더 살펴보다가 발견했다.

 

🔗 문제링크 최단경로

1753번: 최단경로