CS/알고리즘

백준 2178번 미로 탐색 파이썬 풀이

happy_life 2022. 3. 27. 10:33

백준 2178번 미로 탐색 파이썬 풀이

문제

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

1 0 1 1 1 1
1 0 1 0 1 0
1 0 1 0 1 1
1 1 1 0 1 1

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

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

 


정답 코드

#미로 탐색

from collections import deque

N,M = map(int,input().split())

#초기화
maze = []
#상하좌우
dx = [-1,1,0,0]
dy = [0,0,-1,1]

for i in range(N):
    maze.append(list(map(int,input())))

#미로 모양보기
# for i in maze:
#     print(i)

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(0 <= nx < N and 0 <= ny < M):
                if(maze[nx][ny] == 1):
                    queue.append((nx,ny))
                    maze[nx][ny] = maze[x][y] + 1


BFS(0,0)
print(maze[N-1][M-1])
# for i in maze:
#     print(i)

 

가장 인상깊었던 코드

 if(0 <= nx < N and 0 <= ny < M):
                if(maze[nx][ny] == 1):
                    queue.append((nx,ny))
                    maze[nx][ny] = maze[x][y] + 1

상하좌우를 체크해버리면 어떻게 넓이 기준으로 구분을 해야하나 고민 했는데, 위와같은 코드를 통해 구분하는 아이디어를 보고 굉장히 놀랐다. 알고 나면 쉬운 아이디어지만 모를 때는 떠올리기 정말 어려운 그런 아이디어가 아닌가 싶다.

 

 

배운점 && 공부 방향

 

1. DFS 에는 꽤 익숙해졌으나 BFS 는 아직 미숙하므로 BFS 문제를 더 풀어볼 필요가 있다.

 

2. queue 에 어떤식으로 x,y 좌표를 넣어야하나 고민했는데 튜플 형식으로 넣으면 된다.

queue = deque()
    queue.append((x,y))

튜플을 넣은 queue 의 구조