백준_BFS_미로 탐색_2178_파이썬

석준·2022년 7월 14일
0

백준_문제풀이

목록 보기
5/30
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개의 정수로 미로가 주어진다. 각각의 수들은 붙어서 입력으로 주어진다.

✅ 알고리즘 분류

  • 너비 우선 탐색

💡풀이

쉽게 풀 수 있는 BFS문제다
배열을 4방향 탐색을 하며 '1'이면서 방문하지 않은곳은 좌표와 현재까지 거리+1을 큐에 담아
너비 우선 탐색을 하고 종점에 도착하면 거리의 최솟값을 갱신하는 방식

from collections import deque

n, m = map(int, input().split())
arr = [list(input()) for _ in range(n)]

direct = [(1, 0), (-1, 0), (0, 1), (0, -1)]
res = n*m+1

visit = [[0]*m for _ in range(n)]	# 방문 체크할 배열
visit[0][0] = 1
Q = deque([(0, 0, 1)])

while Q:
    nr, nc, cnt = Q.popleft()		# 행, 열, 현재까지 거리

    if nr == n-1 and nc == m-1:

        res = min(res, cnt)
        continue

    for d in direct:
        r, c = nr+d[0], nc+d[1]
        if 0 <= r < n and 0 <= c < m and not visit[r][c] and arr[r][c] == '1':
            visit[r][c] = 1
            Q.append((r, c, cnt+1))

print(res)
profile
파이썬 서버 개발자 지망생

0개의 댓글