[백준] 14503 로봇 청소기 (python 파이썬)

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

14503번: 로봇 청소기

풀이 과정

로봇 청소기 작동방식을 보고 재귀함수를 사용하여  푼다.

크게 로봇청소기가 왼쪽으로 회전하는 파트(a)와 후진하는 파트(b)로 두개로 나누어서 해결했다.

이 문제는 경계가 모두 벽으로 막혀있어 따로 제약조건은 달지 않았다.

 

def clean(x,y,d,count):
    if area[x][y] == 0:
        area[x][y] = 2

함수는 x,y 좌표 , 바라보는 방향d, 그리고 2a단계가 실행되는 횟수 count로 매개변수를 선언한다.

 

함수를 호출하면 맨 처음 로봇 청소기가 빈공간에 있을때 청소를 시켜준다. 이때 청소했다는 것을

표현하기 위해 빈공간(0)과 벽(1)과 다른 숫자인 2를 입력 해준다.

 

    if count <= 3:
        if d == 0:
            if area[x][y - 1] == 0:
                #print("서쪽으로 가서 청소")
                clean(x,y - 1,3,0)
            else:
                #print("서쪽으로 돌기")
                clean(x,y,3,count + 1)

        elif d == 3:
            if area[x + 1][y] == 0:
                #print("남쪽으로 가서 청소")
                clean(x + 1,y,2,0)
            else:
                #print("남쪽으로 돌기")
                clean(x,y,2,count + 1)

이후 코드 작성순서는 2b파트가 먼저지만 설명을 위해 2a파트를 먼저 풀이한다.

먼저 회전 횟수(count)가 3번 이하이면 바라보는 방향(d)에 따라 if문을 실행시켜주면 된다.

 

왼쪽 공간이 비여있다면 해당 방향으로 가서 청소하고 아니라면 방향으로 돌기만하고 count값만 1 증가시켜준다

    if count == 4:
        #print("4번 실행")
        if d == 0:
            if area[x + 1][y] == 1:
                #print("종료")
                return
            else:
                clean(x + 1,y,0,0)

이후 회전 횟수(count)가 4가 되면 2b파트를 실행한다.

바라보는 방향의 뒤쪽이 벽(1)이라면 로봇 청소기 작동을 멈추고(함수종료), 아니라면 바라보는 방향을 유지하고 뒤로간다.

이때 회전 횟수(count)는 0으로 초기화 시켜준다.

 

result = 0
for i in range(N):
    for j in range(M):
        if area[i][j] == 2:
            result += 1
print(result)

이후 이차원 배열에서 값이 2인것을 카운트해서 출력해준다.

 

파이썬으로 작성한 전체 코드는 다음과 같다. sys를 사용하여 재귀깊이를 증가시켰다.

import sys
sys.setrecursionlimit(10**6)
N, M = map(int,input().split())
r,c,di = map(int,input().split())
# 0 북, 1 동, 2 남, 3서
area = []
for _ in range(N):
    area.append([int(x) for x in input().split()])

def clean(x,y,d,count):
    if area[x][y] == 0:
        area[x][y] = 2

    if count == 4:
        #print("4번 실행")
        if d == 0:
            if area[x + 1][y] == 1:
                #print("종료")
                return
            else:
                clean(x + 1,y,0,0)

        elif d == 3:
            if area[x][y + 1] == 1:
                #print("종료")
                return
            else:
                clean(x,y + 1,3,0)

        elif d == 2:
            if area[x - 1][y] == 1:
                #print("종료")
                return
            else:
                clean(x - 1,y,2,0)

        elif d == 1:
            if area[x][y - 1] == 1:
                #print("종료")
                return
            else:
                clean(x,y - 1,1,0)

    if count <= 3:
        if d == 0:
            if area[x][y - 1] == 0:
                #print("서쪽으로 가서 청소")
                clean(x,y - 1,3,0)
            else:
                #print("서쪽으로 돌기")
                clean(x,y,3,count + 1)

        elif d == 3:
            if area[x + 1][y] == 0:
                #print("남쪽으로 가서 청소")
                clean(x + 1,y,2,0)
            else:
                #print("남쪽으로 돌기")
                clean(x,y,2,count + 1)

        elif d == 2:
            if area[x][y + 1] == 0:
                #print("동쪽으로 가서 청소")
                clean(x,y + 1,1,0)
            else:
                #print("동쪽으로 돌기")
                clean(x,y,1,count + 1)

        elif d == 1:
            if area[x - 1][y] == 0:
                #print("북쪽으로 가서 청소")
                clean(x - 1,y,0,0)
            else:
                #print("북쪽으로 돌기")
                clean(x,y,0,count + 1)

clean(r,c,di,0)
#print(area)
result = 0
for i in range(N):
    for j in range(M):
        if area[i][j] == 2:
            result += 1
print(result)