문제설명
입력조건
출력조건
입력예시
5 6
101010
111111
000001
111111
111111
출력예시
10
내가짠코드
# N, M을 공백 기준으로 구분하여 입력받기
n, m = map(int, input().split())
# 2차원 리스트의 맵 정보 입력받기
graph = []
for i in range(n):
graph.append(list(map(int, input())))
flag = True # left first if True(n>m), else down first(n<m)
if n < m:
flag = False
i, j = 0, 0 # initial position\
cnt = 0
def func(i, j, cnt):
print("HEY", i, j)
if graph[i][j] == 1:
graph[i][j] = True
cnt += 1
if flag == True:
if i < m:
result = func(i + 1, j, cnt)
if result == False:
if j < n:
func(i, j + 1, cnt)
else:
func(i, j + 1, cnt)
else:
print("INIT", i, j, n, m)
if j < n:
result = func(i, j+1, cnt)
print(i, j, result)
if result == False:
if i < m:
func(i+1, j, cnt)
else:
func(i + 1, j, cnt)
return True
else:
return False
func(i, j, cnt)
bfs를 활용해서 다시 짠 코드
from collections import deque
# N, M을 공백 기준으로 구분하여 입력받기
n, m = map(int, input().split())
# 2차원 리스트의 맵 정보 입력받기
graph = []
for i in range(n):
graph.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()
if x == n - 1 and y == m - 1:
return graph[x][y]
for i in range(4):
nx = x + dx[i]
ny = y + dy[i]
if nx < 0 or ny < 0 or nx > n-1 or ny > m-1:
continue
if graph[nx][ny] == 0:
continue
if graph[nx][ny] == 1:
queue.append((nx, ny))
graph[nx][ny] = graph[x][y] + 1
print(bfs(0,0)) # 10
정답코드
from collections import deque
# N, M을 공백 기준으로 구분하여 입력받기
n, m = map(int, input().split())
# 2차원 리스트의 맵 정보 입력 받기
graph = []
for i in range(n):
graph.append(list(map(int, input())))
# 이동할 네 가지 방향 정의 (상, 하, 좌, 우)
dx = [-1, 1, 0, 0]
dy = [0, 0, -1, 1]
def bfs(x, y):
# 큐(Queue) 구현을 위해 deque 라이브러리 사용
queue = deque()
queue.append((x, y))
# 큐가 빌 때까지 반복하기
while queue:
x, y = queue.popleft()
# 현재 위치에서 4가지 방향으로의 위치 확인
for i in range(4):
nx = x + dx[i]
ny = y + dy[i]
# 미로 찾기 공간을 벗어난 경우 무시
if nx < 0 or nx >= n or ny < 0 or ny >= m:
continue
# 벽인 경우 무시
if graph[nx][ny] == 0:
continue
# 해당 노드를 처음 방문하는 경우에만 최단 경로 기록
if graph[nx][ny] == 1:
graph[nx][ny] = graph[x][y] + 1
queue.append((nx, ny))
# 가장 오른쪽 아래의 최단 거리 반환
return graph[n-1][m-1]
# BFS를 수행한 결과 출력
print(bfs(0,0))
해설
이 문제에서는 n-1, m-1에 도달하면 바로 종료를 시키도록 while 루프 안에 조건문을 걸어 줄 수 있다. 최악의 경우 이 조건문을 수행하는 것은 비용이 되지만, 빠르게 목적지에 도달한다면 불필요한 연산을 피할 수도 있다. 최대한 불필요한 연산을 피하기 위해서는 아래 로직을 추가할 수 있다.
n이 m보다 작다면, 위 아래를 먼저 방문하도록 dx = [-1, 1, 0, 0], dy = [0, 0,-1, 1]
n이 m보다 크다면, 좌 우를 먼저 방문하도록 dx = [0, 0, -1, 1], dy = [-1, 1, 0, 0] 할 수 있을 것이다.