이코테-미로탈출-python

Ne5s·2022년 9월 5일
0

알고리즘 문제풀기

목록 보기
27/31
post-thumbnail
post-custom-banner

📖 문제

A는 N x M 크기의 직사각형 형태의 미로에 갇혀 있다. 미로에는 여러 마리의 괴물이 있어 이를 피해 탈출해야 한다. 현재 위치는 (1, 1)이고 미로의 출구는 (N,M)의 위치에 존재하며 한 번에 한 칸씩 이동할 수 있다. 괴물이 있는 부분은 0으로, 괴물이 없는 부분은 1로 표시되어 있다. 미로는 반드시 탈출할 수 있는 형태로 제시된다. 탈출하기 위해 움직여야 하는 최소 칸의 개수를 구하라. 칸을 셀 때는 시작 칸과 마지막 칸을 모두 포함해서 계산한다.

입력조건

첫째 줄에 두 정수 N, M(4 <= N, M <= 200)이 주어진다. 다음 N개의 줄에는 각각 M개의 정수(0혹은 1)로 미로의 정보가 주어진다. 각각의 수들은 공백 없이붙어서 입력으로 제시된다. 또한 시작 칸과 마지막 칸은 항상 1이다.

출력 조건

첫째 줄에 최소 이동 칸의 개수를 출력한다.

입력 예시

5 6
101010
111111
000001
111111
111111

출력 예시

10

🎈 풀이

  • python에서는 큐를 사용하기 위해 deque를 사용한다.
  • map함수를 이용해서 '111111'의 문자열을 정수형으로 바꾸면서 하나씩 분리해낸 뒤 graph에 넣는다.
  • 상,하,좌,우 이동을 나타내기 위한 dx,dy를 만들고
  • bfs 함수 내에서 초기 값을 큐에 넣고 큐가 빌 때까지 while문을 돌도록 하였다.
  • 큐에서 방문할 노드 하나를 pop하고 상하좌우를 봤을 때 처음 방문하는 노드를 찾아 현재 노드의 최단거리에 1을 더해서 이동할 노드 index의 값으로 넣어주고, 방문해야할 목록인 큐에 넣어주었다.
  • 방문할 노드가 없을 때까지 반복을 한 뒤, 도착지 index의 값을 return 해주었다.

💻 코드

  • bfs를 처음 풀어봐서 반 정도는 생각해냈는데, 나는 가면 안되는 길을 처리해줘야 한다고 생각을 했었는데, bfs 개념을 생각해보니 고려할 필요가 없었다. 단순히 도착지의 값을 출력하면 된다는 것..!
    아쉽게 고민을 하다가 해설에서 힌트를 조금 얻고 풀었다.
  • 주석은 코드 이해를 돕기 위해 답안에 있는 것을 옮겨 적었다.
from collections import deque

# N, M을 공백으로 구분하여 입력받기
n, m = map(int, input().split())
# 2차원 리스트의 맵 정보 입력받기
graph = []
for i in range(n):
    graph.append(list(map(int, input())))

# 이동할 네 방향 정의(상, 하, 좌, 우)
dx = [-1, 1, 0, 0]
dy = [0, 0, -1, 1]

# BFS 소스코드 구현
def bfs(x,y):
    # 큐(Queue) 구현을 위해 deque 라이브러리 사용
    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
            # 해당 노드를 처음 방문하는 경우에만 최단 거리 기록
            if graph[nx][ny] == 1:
                graph[nx][ny] = graph[x][y] + 1
                queue.append((nx, ny))
    # 가장 오른쪽 아래까지의 최단 거리 반환
    return graph[n-1][m-1]

# BFS를 수행한 결과 출력
print(bfs(0,0))

💡 문제 출처

이것이 취업을 위한 코딩테스트다 with 파이썬
이코테 저자(나동빈님) 깃허브(참고코드)

profile
초보개발자
post-custom-banner

0개의 댓글