[알고리즘/백준] 2178번 : 미로 탐색(python)

유현민·2022년 3월 10일
0

알고리즘

목록 보기
44/253

처음에는 count 변수를 두고 풀었는데 그렇게 풀면 주변에 1이 있으면 계속 더해져서 답이 안나왔다. 확실한 길로 갈때만 1을 더해주면 된다.

from collections import deque

N, M = map(int, input().split())
a = [list(map(str, input())) for _ in range(N)]
visit = [[0]*M for _ in range(N)]
dx = [1, -1, 0, 0]
dy = [0, 0, 1, -1]
q = deque()
q.append((0,0))
visit[0][0] = 1

while q:
    x, y = q.popleft()
    if x == N-1 and y == M-1:
        print(visit[x][y])
        break
    for i in range(4):
        nx = dx[i] + x
        ny = dy[i] + y
        if 0 <= nx < N and 0 <= ny < M:
            if visit[nx][ny] == 0 and a[nx][ny] == '1':
                visit[nx][ny] = visit[x][y] + 1
                q.append((nx, ny))
profile
smilegate megaport infra

0개의 댓글