GitHub

https://github.com/Choidongjun0830

파이썬 알고리즘 문제 풀이

[프로그래머스] Lv2. 리코쳇 로봇 / BFS

gogi masidda 2024. 9. 11. 01:40
리코쳇 로봇
문제 설명

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

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

이 게임에서 말의 움직임은 상, 하, 좌, 우 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"는 한 번씩 등장합니다.

입출력 예boardresult
["...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 합니다.

내 풀이

from collections import deque
def solution(board):
    queue = deque()
    col = len(board)
    row = len(board[0])
    dist = [[10000 for i in range(row)] for i in range(col)]
    dx = [-1, 1, 0, 0]
    dy = [0, 0, -1, 1]
    for i in range(col):
        for j in range(row):
            if board[i][j] == 'R':
                start = [i, j, 0]
                dist[i][j] = 0
                break
    queue.append(start)
    while queue:
        x, y, cnt = queue.popleft()
        if board[x][y] == 'G':
            return cnt
        for i in range(4):
            nx = x
            ny = y
            #장애물을 안만났으면 계속 미끄러지기
            while 0 <= nx + dx[i] < col and 0 <= ny + dy[i] < row and board[nx + dx[i]][ny + dy[i]] != 'D': 
                    nx += dx[i]
                    ny += dy[i]
            if dist[nx][ny] > cnt + 1:
                dist[nx][ny] = cnt + 1
                queue.append([nx, ny, cnt + 1])
    return -1
  • 첫번째 중첩 for문은 R의 위치를 찾기 위한 것. 시작점을 queue에 append
  • while문: queue에 원소가 있으면
    • x, y, cnt를 queue에서 popleft()로 가져오기
    • 만약 board[x][y]가 'G'이면 cnt를 리턴
    • for문
      • dx, dy를 순회하며 한 방향으로 나가기
      • while문: 한칸 나아간 것이 board에서 벗어나지 않고 D(장애물)이 아니면
        • nx += dx[i], ny += dy[i]를 통해 한 방향으로 쭉 미끄러지기\
      • if문: dist[nx][ny]의 값이 cnt + 1보다 크면. (크면 그 칸까지 가기위한 횟수가 더 많은 횟수가 필요한 것)
  • 'G'를 만나지 않았으면 -1 리턴
728x90