본문 바로가기
Algorithm/이것이 코딩 테스트다

[ 파이썬(python) ] 이것이 코딩 테스트다 - 미로 탈출(BFS)

by YWTechIT 2021. 4. 15.
728x90

📌 이것이 코딩테스트다 - 미로 탈출

영상 링크(51:30)


💡 나의 풀이

BFS를 이용하는 흐름을 알아야 풀 수 있었는데, 그 흐름을 잘 몰라서 못 풀었다.


이전까지는 입력값을 그래프로 변환해서 풀었는데, 입력값이 이중 리스트일때는 어떻게 줘야 할지 감이 잘 잡히지 않았다.

또, BFS문제를 풀어보니까 값이 어떻게 바뀌는지 흐름을 잡는것이 중요하다고 느꼈다.

from collections import deque

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

# 상하좌우 선언
dx = [-1, 1, 0, 0]
dy = [0, 0, -1, 1]

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 nx < 0 or ny < 0 or nx >= n or ny >= m:
                continue

            # 새로운 그래프의 좌표가 괴물이 있는 부분으로 가는 경우
            if graph[nx][ny] == 0:
                continue

            # 새로운 그래프의 좌표가 괴물이 없는 부분으로 가는 경우, 이전 값에 +1을 해줘서 최단 경로를 계산한다.
            if graph[nx][ny] == 1:
                graph[nx][ny] = graph[x][y] + 1
                queue.append((nx, ny))

    # index는 0부터 세므로 출구(N, M)의 좌표는?
    return graph[n-1][m-1]

print(bfs(0, 0))
반응형

댓글