[BOJ | Python] 1261번: 알고스팟

https://www.acmicpc.net/problem/1261

 

문제

알고스팟 운영진이 모두 미로에 갇혔다. 미로는 N*M 크기이며, 총 1*1크기의 방으로 이루어져 있다. 미로는 빈 방 또는 벽으로 이루어져 있고, 빈 방은 자유롭게 다닐 수 있지만, 벽은 부수지 않으면 이동할 수 없다.

알고스팟 운영진은 여러명이지만, 항상 모두 같은 방에 있어야 한다. 즉, 여러 명이 다른 방에 있을 수는 없다. 어떤 방에서 이동할 수 있는 방은 상하좌우로 인접한 빈 방이다. 즉, 현재 운영진이 (x, y)에 있을 때, 이동할 수 있는 방은 (x+1, y), (x, y+1), (x-1, y), (x, y-1) 이다. 단, 미로의 밖으로 이동 할 수는 없다.

벽은 평소에는 이동할 수 없지만, 알고스팟의 무기 AOJ를 이용해 벽을 부수어 버릴 수 있다. 벽을 부수면, 빈 방과 동일한 방으로 변한다.

만약 이 문제가 알고스팟에 있다면, 운영진들은 궁극의 무기 sudo를 이용해 벽을 한 번에 다 없애버릴 수 있지만, 안타깝게도 이 문제는 Baekjoon Online Judge에 수록되어 있기 때문에, sudo를 사용할 수 없다.

현재 (1, 1)에 있는 알고스팟 운영진이 (N, M)으로 이동하려면 벽을 최소 몇 개 부수어야 하는지 구하는 프로그램을 작성하시오.

 

입력

첫째 줄에 미로의 크기를 나타내는 가로 크기 M, 세로 크기 N (1 ≤ N, M ≤ 100)이 주어진다. 다음 N개의 줄에는 미로의 상태를 나타내는 숫자 0과 1이 주어진다. 0은 빈 방을 의미하고, 1은 벽을 의미한다.

(1, 1)과 (N, M)은 항상 뚫려있다.

 

출력

첫째 줄에 알고스팟 운영진이 (N, M)으로 이동하기 위해 벽을 최소 몇 개 부수어야 하는지 출력한다.

 

풀이

다익스트라... 왜이리어려운거야!

import heapq

# M은 열의 개수, N은 행의 개수
M, N = map(int, input().split())

# 그래프 정보 입력 (N x M 크기)
graph = [list(map(int, input().split())) for _ in range(N)]

# 최단 거리 테이블을 무한대 값으로 초기화
distance = [[float('inf')] * M for _ in range(N)]

# 방향 벡터 (상, 우, 하, 좌)
delta_row = [-1, 0, 1, 0]
delta_col = [0, 1, 0, -1]


def dijkstra():
    # 시작점 (0, 0)에서 초기 비용 0으로 시작
    queue = []
    heapq.heappush(queue, (0, 0, 0))  # (비용, 행, 열)
    distance[0][0] = 0  # 시작 위치의 거리는 0으로 설정

    while queue:
        # 우선순위 큐에서 현재 비용과 위치를 꺼냄
        current_cost, current_row, current_col = heapq.heappop(queue)

        # 이미 처리된 위치라면 무시
        if current_cost > distance[current_row][current_col]:
            continue

        # 상하좌우 인접한 위치 확인
        for i in range(4):
            next_row = current_row + delta_row[i]
            next_col = current_col + delta_col[i]

            # 위치가 그래프 내에 있는지 확인
            if 0 <= next_row < N and 0 <= next_col < M:
                # 다음 위치까지의 비용 계산
                new_cost = current_cost + graph[next_row][next_col]

                # 더 짧은 경로가 발견된 경우 갱신하고 큐에 삽입
                if new_cost < distance[next_row][next_col]:
                    distance[next_row][next_col] = new_cost
                    heapq.heappush(queue, (new_cost, next_row, next_col))


# 다익스트라 알고리즘 수행
dijkstra()

# 목적지 위치의 최단 거리 출력
print(distance[N - 1][M - 1])