[BAEKJOON python] 13460_구슬탈출

구슬탈출
구슬 탈출은 직사각형 보드에 빨간 구슬과 파란 구슬을 하나씩 넣은 다음, 빨간 구슬을 구멍을 통해 빼내는 게임

보드의 세로 크기는 N, 가로 크기는 M,  1×1크기의 칸으로 나누어져 있다
가장 바깥 행과 열은 모두 막혀져 있고, 보드에는 구멍이 하나
빨간 구슬과 파란 구슬의 크기는 보드에서 1×1크기의 칸을 가득 채우는 사이즈이고, 각각 하나씩 들어가 있다
빨간 구슬을 구멍을 통해서 빼내는 것이다
이때, 파란 구슬이 구멍에 들어가면 안 된다

구슬을 손으로 건드릴 수는 없고, 중력을 이용해서 이리 저리 굴려야 한다
    왼쪽으로 기울이기, 오른쪽으로 기울이기, 위쪽으로 기울이기, 아래쪽으로 기울이기와 같은 네 가지 동작

각각의 동작에서 공은 동시에 움직인다
빨간 구슬이 구멍에 빠지면 성공이지만, 파란 구슬이 구멍에 빠지면 실패
빨간 구슬과 파란 구슬이 동시에 구멍에 빠져도 실패
빨간 구슬과 파란 구슬은 동시에 같은 칸에 있을 수 없다
빨간 구슬과 파란 구슬의 크기는 한 칸을 모두 차지
기울이는 동작을 그만하는 것은 더 이상 구슬이 움직이지 않을 때 까지

보드의 상태가 주어졌을 때, 최소 몇 번 만에 빨간 구슬을 구멍을 통해 빼낼 수 있는지 구하는 프로그램
10번 이하로 움직여서 빨간 구슬을 구멍을 통해 빼낼 수 없으면 -1을 출력

첫 번째 줄에는 보드의 세로, 가로 크기를 의미하는 두 정수 N, M (3 ≤ N, M ≤ 10)
N개의 줄에 보드의 모양을 나타내는 길이 M의 문자열
'.'은 빈 칸, '#'은 공이 이동할 수 없는 장애물 또는 벽, 'O'는 구멍의 위치, 'R'은 빨간 구슬의 위치, 'B'는 파란 구슬의 위치
입력되는 모든 보드의 가장자리에는 모두 '#'이 있다
구멍의 개수는 한 개 이며, 빨간 구슬과 파란 구슬은 항상 1개

from collections import deque

# 세로, 가로 크기
n, m = map(int, input().split())
graph = []
dx = [-1 ,0, 1, 0]
dy = [0, 1, 0, -1]
rx, ry, bx, by = 0, 0, 0, 0
visited = [[[[False] * m for _ in range(n)] for _ in range(m)] for _ in range(n)]

for i in range(n) :
    data = list(input())
    graph.append(data)
    for d in range(len(data)) :
        if data[d] == "R" :
            rx, ry = i, d
        elif data[d] == "B" :
            bx, by = i, d

def move(x, y, dx, dy) :
    cnt = 0
    # '#'은 공이 이동할 수 없는 장애물 또는 벽, 'O'는 구멍의 위치
    while graph[x+dx][y+dy] != "#" and graph[x][y] != "O" :
        # 움직이는 것을 종료한 후 움직인 후의 위치와 움직인 횟수
        x += dx
        y += dy
        cnt += 1
    return x, y, cnt
        
def bfs(rx, ry, bx, by) :
    queue = deque()
    queue.append((rx, ry, bx, by, 1))
    visited[rx][ry][bx][by] = True
    while queue : 
        r_x, r_y, b_x, b_y, depth = queue.popleft()
        if depth > 10 :
            break
        for i in range(4) :
            r_nx, r_ny, r_cnt = move(r_x, r_y, dx[i], dy[i])
            b_nx, b_ny, b_cnt = move(b_x, b_y, dx[i], dy[i])
            # 기울린 후 파란 구슬의 위치가 구멍이 아니면서 빨간 구슬의 위치가 구멍이라면 기울린 횟수(depth)
            if graph[b_nx][b_ny] != "O" :
                if graph[r_nx][r_ny] == "O" :
                    return depth
                # 빨강, 파랑 구슬 같이 있을 수 없음
                if r_nx == b_nx and r_ny == b_ny :
                    if r_cnt > b_cnt :
                        r_nx -= dx[i]
                        r_ny -= dy[i]
                    else : 
                        b_nx -= dx[i]
                        b_ny -= dy[i]
                if not visited[r_nx][r_ny][b_nx][b_ny] :
                    visited[r_nx][r_ny][b_nx][b_ny] = True
                    queue.append((r_nx, r_ny, b_nx, b_ny, depth + 1))
    return -1

print(bfs(rx, ry, bx, by))

---

예제 입력

7 7
#######
#...RB#
#.#####
#.....#
#####.#
#O....#
#######

예제 출력

5

 

 

 

우왕 4차원리스트 처음 써보늗ㅅ

신기하다

감사합니다.

https://developer-ellen.tistory.com/44