[백준] 알고스팟 / 1261번 / 파이썬 / Python / Dijkstra 알고리즘

 

💡solutions )

💬 단순 다익스트라(Dijkstra) 알고리즘으로 해결한 문제, 해당 문제에서 노드의 비용에 해당하는 것은 벽을 부순 횟수(cnt)

💬 BFS로도 방문 중복으로 하여 해결할 수 있는 문제

 

 

🎫code )

import sys
import heapq


def dijkstra():
    heap = []
    heapq.heappush(heap, [0, 0, 0])  # 시작점
    visit[0][0] = 1

    while heap:
        cnt, x, y = heapq.heappop(heap)  # 벽 부순 횟수가 최소인 경우가 나옴
        for k in range(4):
            nx, ny = x + dx[k], y + dy[k]
            if x == n - 1 and y == m - 1:  # 종료조건
                print(cnt)
                break
            if 0 <= nx < n and 0 <= ny < m and not visit[nx][ny]:
                visit[nx][ny] = 1

                if maze[nx][ny]:  # 벽인 경우
                    heapq.heappush(heap, [cnt + 1, nx, ny])
                else:  # 빈방인 경우
                    heapq.heappush(heap, [cnt, nx, ny])


input = sys.stdin.readline
m, n = map(int, input().rstrip().split(' '))
maze = [list(map(int, input().rstrip())) for _ in range(n)]
visit = [[0] * m for _ in range(n)]
dx, dy = [-1, 1, 0, 0], [0, 0, -1, 1]
dijkstra()

 

📌 description )

문제출처 : www.acmicpc.net/problem/1261

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)으로 이동하기 위해 벽을 최소 몇 개 부수어야 하는지 출력한다.