특정 문제를 메모리 효율적이고 빠른 성능으로 해결하려면 알고리즘을 공부해야 합니다.
다음 내용은 "이것이 취업을 위한 코딩 테스트다(나동빈)"를 보고 정리한 내용입니다.
[문제]
N x M 크기의 직사각형 형태의 미로에 여러 마리의 괴물이 있어 이를 피해 탈출해야 한다. 현재 위치는 (1, 1)이고 미로의 출구는 (N,M)의 위치에 존재하며 한 번에 한 칸씩 이동할 수 있다. 괴물이 있는 부분은 0으로, 괴물이 없는 부분은 1로 표시되어 있다. 미로는 반드시 탈출할 수 있는 형태로 제시된다. 탈출하기 위해 움직여야 하는 최소 칸의 개수를 구하라. 칸을 셀 때는 시작 칸과 마지막 칸을 모두 포함해서 계산한다.
[입력 조건]
첫째 줄에 두 정수 N, M(4 <= N, M <= 200)이 주어진다. 다음 N개의 줄에는 각각 M개의 정수(0혹은 1)로 미로의 정보가 주어진다. 각각의 수들은 공백 없이붙어서 입력으로 제시된다. 또한 시작 칸과 마지막 칸은 항상 1이다.
[출력 조건]
첫째 줄에 최소 이동 칸의 개수를 출력한다.
[내 풀이]
from collections import deque
N, M = map(int, input().split())
maze = []
for i in range(N):
maze.append(list(map(int,input())))
dx = [1, -1, 0, 0]
dy = [0, 0, -1, 1]
#이건 재귀가 아니다 명심!
def bfs(x, y):
queue = deque()
queue.append((x,y))
while queue:
x,y = queue.popleft()
for i in range(4):
nx = x + dx[i]
ny = y + dy[i]
if nx < 0 or ny < 0 or nx >= N or ny >= M:
continue
if maze[nx][ny] == 0:
continue
if maze[nx][ny] == 1:
maze[nx][ny] = maze[x][y] + 1
queue.append((nx,ny))
return maze[N - 1][M - 1]
print(bfs(0,0))