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

선주·2022년 1월 13일
0

Python PS

목록 보기
17/65
post-thumbnail

📌 문제

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)의 위치로 이동할 수 있다. 칸을 셀 때에는 시작 위치와 도착 위치도 포함한다.

입력

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

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

예제 입력 1

4 6
101111
101010
101011
111011

예제 출력 1

15

예제 입력 2

4 6
110110
110110
111111
111101

예제 출력 2

9

예제 입력 3

2 25
1011101110111011101110111
1110111011101110111011101

예제 출력 3

38

예제 입력 4

7 7
1011111
1110001
1000001
1000001
1000001
1000001
1111111

예제 출력 4

13


📌 풀이

💬 Code

import sys
from collections import deque
input = sys.stdin.readline


def bfs(x, y):
    q = deque([(x-1, y-1)])
    while q:
        x, y = q.popleft()

        for i in range(4):
            nx = x + dx[i]
            ny = y + dy[i]

            if 0 <= nx < n and 0 <= ny < m and maze[nx][ny] == 1:
                q.append((nx, ny))
                maze[nx][ny] += maze[x][y]

    return maze[n-1][m-1]


if __name__ == '__main__':
    n, m = map(int, input().split())
    maze = []

    for _ in range(n):
        maze.append(list(map(int, input().rstrip())))
	
    # 방향벡터 (상, 하, 좌, 우)
    dx = [-1, 1, 0, 0]
    dy = [0, 0, -1, 1]

    print(bfs(1, 1))

💡 Solution

이코테에서 예제로 나왔던 문제라 쉽게 풀었다.

  1. 좌표가 주어지면 해당 좌표의 상하좌우 좌표로 이동할 수 있는지 각각 체크하기 위해 좌표마다 for문을 4번 돌린다.

  2. nx, ny는 이동좌표를 나타낸다. 이동좌표가 (-1, -1)처럼 좌표를 벗어나지 않았고 해당 좌표에 놓인 값이 1이라면 그 곳으로 이동할 수 있는 것이므로 이동 후 큐에 (nx, ny)쌍을 넣어준다.

  3. 지나야 하는 최소 칸수를 구해야 하므로 이동할 때마다 이동칸수를 누적시키자. maze[x][y]에서 maze[nx][ny]로의 이동칸수는 1칸인데, maze[nx][ny]의 값이 이미 1이기 때문에 maze[nx][ny] += maze[x][y] 해주면 maze[nx][ny]에 여태까지 지나온 칸수가 입력된다.

  4. 목표 좌표의 값을 출력해주면 끝!

profile
기록하는 개발자 👀

0개의 댓글