알고리즘/풀이

백준 2178번 미로 탐색 파이썬

목표는 커리 2021. 3. 19. 19:16
728x90
SMALL

문제

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

제출답안

import sys
from collections import deque

input = sys.stdin.readline
n, m = map(int, input().split())

miro_list = []
for i in range(n):
    miro_list.append(list(map(int, input().strip())))

dx = [-1, 1, 0, 0]
dy = [0, 0, -1, 1]


def bfs(x, y):
    q = deque()
    q.append((x, y))
    while q:
        x, y = q.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 miro_list[nx][ny] == 0:
                continue
            if miro_list[nx][ny] == 1:
                miro_list[nx][ny] = miro_list[x][y] + 1
                q.append((nx, ny))
    return miro_list[n - 1][m - 1]


print(bfs(0, 0))
  1. deque를 사용하기 위해 import한다.
  2. 사용자에게서 n,m값을 받아온다.
  3. 2중 리스트를 만들어서 해당 인덱스에 값을 저장한다.
  4. 이동한 상,하,좌,우를 정의한다.
  5. deque를 이용하여 인접한 열에 방문했다는 표시를 한다.
  6. queue가 빌때까지 계속 while을 돌린다.
    1. 우선 처음 들어온 값을 꺼내서 상,하,좌,우 방문한다.
    2. 미로를 벗어나거나 벽이면 무시한다.
    3. 처음 방문하는 곳이면 해당 노드에 이전 노드에 +1을 한다.
    4. 그 다음에 해당 인덱스를 다시 queue에 넣고 인접하지 않은 열이 있는지 없는지 처리한 후 빠져나온다.
  7. 가장 오른쪽 아래에 해당하는 인덱스를 주어서 값을 리턴한다.
728x90
LIST