[백준-14503] 로봇 청소기 / Python

14503번: 로봇 청소기

import sys

input = sys.stdin.readline

n, m = map(int, input().split())
r, c, d = map(int, input().split())
graph = [list(map(int, input().split())) for _ in range(n)]
visited = [[0] * m for _ in range(n)]
visited[r][c] = 1

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


def turnLeft():
    global d
    d -= 1
    if d == -1:
        d = 3


turnTime = 0
answer = 1
while True:
    turnLeft()
    nx = r + dx[d]
    ny = c + dy[d]
    if visited[nx][ny] == 0 and graph[nx][ny] == 0:
        visited[nx][ny] = 1
        r, c = nx, ny
        answer += 1
        turnTime = 0
        continue
    else:
        turnTime += 1
    if turnTime == 4:
        nx = r - dx[d]
        ny = c - dy[d]
        if graph[nx][ny] == 0:
            r, c = nx, ny
        else:
            break
        turnTime = 0

print(answer)

 

해설

시뮬레이션 문제. 로봇 청소기는 0인 곳을 돌아다니며 청소를 한다. 여기서 청소를 한 곳의 수를 출력하는 문제다.

turnLeft는 왼쪽으로 회전을 한 방향을 정하는 함수다. 방문하지 않았고, 벽이 아닌 곳에 방문할 경우 answer를 하나씩 늘려준다. 회전해서 바라보는 방향이 방문을 했거나 벽이라면 회전을 계속한다. 여기서, 회전을 4번 모두 한 경우에는 마지막으로 바라보고 있는 방향의 뒷 쪽으로 이동을 해준다. 이 때, 벽이 아닌 곳으로 후진을 해야하고, 후진할 곳이 벽으로 막혀있다면 반복문을 멈춘다.