BOJ - 로봇 청소기 14503번 (python)

❓ 문제 - 백준 로봇 청소기 14503번 - python 풀이법

출처 

(https://www.acmicpc.net/problem/14503)

14503번: 로봇 청소기

 

 

 

📝 문제해결법

1. 이 문제는 구현으로 풀었다.

• 로봇 청소기의 회전은 dx, dy로 (북-동-남-서)로 처리하며 change_dir() 함수를 통해 현재 바라보는 방향에서 왼쪽으로 회전했을 때 바라보게 되는 방향을 구한다.
• 처음 문제에 주어진대로 로봇이 처음 있는 곳은 청소가 가능한 곳이니 청소를 하고 answer = 1을 해준다.
• 4방향으로 로봇 청소기 주변을 탐색하면서 만약 회전했을 때 청소가 가능한 지역이면 해당 지역으로 이동하여 청소한다.
• 4방향을 회전하며 다 탐색했지만, 청소가 가능한 지역이 없다면 현재 위치에서 바라보는 방향을 따라 후진한다.
• 후진 했을 때 후진한 지역이 벽이라 이동하지 못한다면 로봇 청소기를 종료시킨다.
• 후진 했을 때 후진한 지역이 이동가능 하면 해당 위치로 이동한 후 while문 돌면서 다시 주변 4방향의 지역이 청소가능한지 탐색한다.

 

2.  문제 풀면서 느낀점

• 구현 문제가 주어지면 최대한 문제에서 주어진 것들을 활용해서 간단하게 구현하려고 노력한다.
• 또한 로봇 청소기가 해당 방향으로 움직였을 때 범위를 벗어나는 ( 0<= nx < n and 0 <= ny < m이 아닌) 경우는 고려할 필요가 없다. 이미 주변이 벽으로 둘러 쌓였기 때문에 범위를 벗어나는 경우는 탐색하지 않는다.

 

 

💻 소스코드

# 로봇 청소기 - BOJ 14503
# 구현
n, m = map(int, input().split())
x, y, dir = map(int, input().split())

# 북 - 동 - 남 - 서
dx = [-1, 0, 1, 0]
dy = [0, 1, 0, -1]

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

graph[x][y] = 2
answer = 1

def change_dir(dir):
    if dir == 0:
        return 3
    else:
        return dir-1

def solve(x, y, dir):
    global answer
    while True:
        check = False
        for i in range(4):
            n_dir = change_dir(dir)
            nx, ny = x + dx[n_dir], y + dy[n_dir]
            dir = n_dir
            if graph[nx][ny] == 0:
                x, y = nx, ny
                graph[nx][ny] = 2
                check = True
                answer += 1
                break

        if not check:
            back_nx, back_ny = x - dx[dir], y - dy[dir]
            if graph[back_nx][back_ny] == 1:
                return answer
            else:
                x, y = back_nx, back_ny

print(solve(x, y, dir))