[프로그래머스] 리코쳇 로봇 파이썬

문제 설명

리코쳇 로봇이라는 보드게임이 있습니다.

이 보드게임은 격자모양 게임판 위에서 말을 움직이는 게임으로, 시작 위치에서 목표 위치까지 최소 몇 번만에 도달할 수 있는지 말하는 게임입니다.

이 게임에서 말의 움직임은 상, 하, 좌, 우 4방향 중 하나를 선택해서 게임판 위의 장애물이나 맨 끝에 부딪힐 때까지 미끄러져 이동하는 것을 한 번의 이동으로 칩니다.

다음은 보드게임판을 나타낸 예시입니다.

...D..R
.D.G...
....D.D
D....D.
..D....

여기서 "."은 빈 공간을, "R"은 로봇의 처음 위치를, "D"는 장애물의 위치를, "G"는 목표지점을 나타냅니다.
위 예시에서는 "R" 위치에서 아래, 왼쪽, 위, 왼쪽, 아래, 오른쪽, 위 순서로 움직이면 7번 만에 "G" 위치에 멈춰 설 수 있으며, 이것이 최소 움직임 중 하나입니다.

게임판의 상태를 나타내는 문자열 배열 board가 주어졌을 때, 말이 목표위치에 도달하는데 최소 몇 번 이동해야 하는지 return 하는 solution함수를 완성하세요. 만약 목표위치에 도달할 수 없다면 -1을 return 해주세요.

---

제한 사항

• 3 ≤

  board

의 길이 ≤ 100

• 3 ≤ board의 원소의 길이 ≤ 100
• board의 원소의 길이는 모두 동일합니다.
• 문자열은 ".", "D", "R", "G"로만 구성되어 있으며 각각 빈 공간, 장애물, 로봇의 처음 위치, 목표 지점을 나타냅니다.
• "R"과 "G"는 한 번씩 등장합니다.

---

입출력 예

board	result
["...D..R", ".D.G...", "....D.D", "D....D.", "..D...."]	7
[".D.R", "....", ".G..", "...D"]	-1

---

입출력 예 설명

입출력 예 #1

• 문제 설명의 예시와 같습니다.

입출력 예 #2

.D.R
....
.G..
...D

• "R" 위치에 있는 말을 어떻게 움직여도 "G" 에 도달시킬 수 없습니다.
• 따라서 -1을 return 합니다.

참고 답안

• 처음에는 문제를 이해하기 어려웠는데 위 사진처럼 움직이는 것이었다.
• 그래서 작성하지 못했던 부분이 while문이다.
  • while문 안에서는 장애물을 만나거나 마지막 지점이 될 때 까지 계속 같은 지점으로 움직이도록 한 것이다.
  • 이 부분만 해결한다면 나머지는 일반 bfs문이랑 동일하다.

from collections import deque
import sys

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

def bfs(y,x,board):
    n, m = len(board), len(board[0])
    visited = [[False] * m for _ in range(n)]
    visited[y][x] = True
    q = deque([(y,x, 0)])
    while q:
        y,x,cnt = q.popleft()
        if board[y][x] == "G":
            return cnt
        for k in range(4):
            ny = y
            nx = x
            while 0 <= ny + dy[k] < n and 0 <= nx + dx[k] < m and board[ny + dy[k]][nx + dx[k]] != 'D':
                ny += dy[k]
                nx += dx[k]
            if not visited[ny][nx]:
                visited[ny][nx] = True
                q.append((ny, nx, cnt +1))
    return -1


def solution(board):
    new_board = []
    for i, b in enumerate(board):
        b = list(b)
        if "R" in b:
            start = (i, b.index('R'))
        new_board.append(b)
    answer = bfs(*start, new_board)
    return answer