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

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

14503번: 로봇 청소기

 

# n x m
n, m = map(int, input().split())

#좌표값, 방향
x, y, d = map(int, input().split())

#청소했는지 안했는지 체크
cheak = [[0]*m for _ in range(n)]

#MAP
arr = []
for i in range(n):
    arr.append(list(map(int, input().split())))

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

cheak[x][y] = 1
count = 1
turn = 0

#왼쪽
def left():
    global d
    d -= 1
    if d == -1:
        d = 3

while True:
    left()
    nx = x + dx[d]
    ny = y + dy[d]
    #청소하지 않은 공간이 존재한다면
    if cheak[nx][ny] == 0 and arr[nx][ny] == 0:
        cheak[nx][ny] = 1
        x = nx
        y = ny
        count += 1
        turn = 0
        continue
    else:
        turn += 1

    #네 방향 모두 청소 or 벽
    if turn == 4:
        nx = x - dx[d]
        ny = y - dy[d]
        if arr[nx][ny] == 0:
            x = nx
            y = ny
        else:
            break
        turn = 0

print(count)

틀린게 없는데 자꾸 답이 이상하게 나와서 오래 고민했는데 

else문이 이상한곳에 있었다..

 

쨋든 이대로 코드만 짜면 되는데 아직은 시간이 좀 걸림 

문제 많이 풀어봐야할듯

1. 현재 위치를 청소한다.
2. 현재 위치에서 현재 방향을 기준으로 왼쪽 방향부터 차례대로 인접한 칸을 탐색한다.
   1. 왼쪽 방향에 아직 청소하지 않은 공간이 존재한다면, 그 방향으로 회전한 다음 한 칸을 전진하고 1번부터 진행한다.
   2. 왼쪽 방향에 청소할 공간이 없다면, 그 방향으로 회전하고 2번으로 돌아간다.
   3. 네 방향 모두 청소가 이미 되어있거나 벽인 경우에는, 바라보는 방향을 유지한 채로 한 칸 후진을 하고 2번으로 돌아간다.
   4. 네 방향 모두 청소가 이미 되어있거나 벽이면서, 뒤쪽 방향이 벽이라 후진도 할 수 없는 경우에는 작동을 멈춘다.