[백준] 2178 - 미로 탐색 (Python)

주재민·2024년 4월 2일

코딩테스트

목록 보기
10/10
post-thumbnail

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

문제

N×M크기의 배열로 표현되는 미로가 있다.

미로에서 1은 이동할 수 있는 칸을 나타내고, 0은 이동할 수 없는 칸을 나타낸다. 이러한 미로가 주어졌을 때, (1, 1)에서 출발하여 (N, M)의 위치로 이동할 때 지나야 하는 최소의 칸 수를 구하는 프로그램을 작성하시오. 한 칸에서 다른 칸으로 이동할 때, 서로 인접한 칸으로만 이동할 수 있다.

위의 예에서는 15칸을 지나야 (N, M)의 위치로 이동할 수 있다. 칸을 셀 때에는 시작 위치와 도착 위치도 포함한다.

입력

첫째 줄에 두 정수 N, M(2 ≤ N, M ≤ 100)이 주어진다. 다음 N개의 줄에는 M개의 정수로 미로가 주어진다. 각각의 수들은 붙어서 입력으로 주어진다.

출력

첫째 줄에 지나야 하는 최소의 칸 수를 출력한다. 항상 도착위치로 이동할 수 있는 경우만 입력으로 주어진다.

풀이

많이 볼 수 있는 BFS 문제이다. 먼저 (1,1)에서 출발해 (N, M)까지 가야하므로 헷갈리지 않게 위아래 양옆으로 0을 집어넣은 미로형태로 만들었다. 이후로는 상하좌우로 움직이며 0인지 아닌지와 방문한 곳인지의 여부를 확인하면서 점점 누적해간다. 이동하다가 좌표가 (N, M)에 도착하면 멈추고 해당 칸에서 움직인 칸수를 출력한다.

코드

import sys
from collections import deque

N, M = map(int, sys.stdin.readline().split())
graph = [[0] * (M+2)]
visited = [[False for _ in range(M+2)] for _ in range(N+2)]
direction = [(1,0), (-1,0), (0,1), (0,-1)]
que = deque()

for _ in range(N):
    line = [0]
    line.extend(list(map(int, input())))
    line.append(0)
    graph.append(line)
    
graph.extend([[0] * (M+2)])

visited[1][1] = True
que.append((1,1))

while que:
    x, y = que.popleft()
    
    if x == N and y == M:
        print(graph[x][y])
        break
        
    for i in range(4):
        nx, ny = x + direction[i][0], y + direction[i][1]
        
        if graph[nx][ny] != 0 and visited[nx][ny] == False:
            visited[nx][ny] = True
            graph[nx][ny] += graph[x][y]
            que.append((nx,ny))

결과


처음 입력받아 그래프를 생성할 때 sys.stdin.readline()을 이용해 입력을 받으려 했으나 개행문자(\n)도 같이 입력을 받아 해당 방법으로는 안된다는 것을 이번에 알았다.

0개의 댓글