46-9kyo-hwang

[9kyo-hwang]

🔗 문제 링크

14503 로봇 청소기

✔️ 소요된 시간

30분

✨ 수도 코드

1. 문제

$1$ x $1$ 크기의 정사각형 칸으로 이루어진 $N$ x $M$ 크기의 방이 있다. 각 칸은 빈 칸(0) 이거나 벽(1)이다.
동, 서, 남, 북 방향으로 움직일 수 있는 로봇 청소기가 $(r, c)$ 좌표에서 출발해 다음과 같이 동작한다.

1. 현재 칸이 아직 청소되지 않은 경우, 현재 칸을 청소한다.
2. 현재 칸의 주변 $4$칸 중 청소되지 않은 빈 칸이 없는 경우,
   1. 바라보는 방향을 유지한 채로 한 칸 후진할 수 있다면 한 칸 후진하고 1번으로 돌아간다.
   2. 바라보는 방향의 뒤쪽 칸이 벽이라 후진할 수 없다면 작동을 멈춘다.
3. 현재 칸의 주변 $4$칸 중 청소되지 않은 빈 칸이 있는 경우,
   1. 반시계 방향으로 $90^\circ$ 회전한다.
   2. 바라보는 방향을 기준으로 앞쪽 칸이 청소되지 않은 빈 칸인 경우 한 칸 전진한다.
   3. 1번으로 돌아간다.

로봇 청소기가 작동을 시작한 후 작동을 멈출 때까지 청소하는 칸의 개수를 출력한다.

2. 풀이

심플한 구현/시뮬레이션 문제이다. 문제에서 설명한 대로 차근차근 하나씩 구현해보자.

입력

N, M = map(int, input().split())
r, c, d = map(int, input().split())
room = [list(map(int, input().split())) for _ in range(N)]
offset = ((-1, 0), (0, 1), (1, 0), (0, -1))

주어지는 데이터를 저장한다. 로봇 청소기는 4가지 방향을 가지고 있는데 각각 북 - 0, 동 - 1, 남 - 2, 서 - 3 이라 했으므로 그에 맞게 방향 오프셋 정보를 함께 저장한다.

cnt = 0
while True:
    ...

print(cnt)

로봇 청소기가 동작을 멈출 때까지 무한 반복문을 돌린다. 동작을 멈추면 청소한 방 개수 cnt를 출력한다.

동작 1

if room[r][c] == 0:
    room[r][c] = 2
    cnt += 1

만약 현재 칸이 청소되지 않은 빈 칸(0)이라면 청소(2)한다. 그리고 청소한 칸 수(cnt)를 1 증가시킨다.

동작 2

if not is_there_a_blank_space():
    dr, dc = offset[(d + 2) % 4]
    nr, nc = r + dr, c + dc
    
    if not out_of_bound(nr, nc) and room[nr][nc] != 1:
        r, c = nr, nc
    else:
        break

주변 4방향에 청소되지 않은 빈 칸이 없다면(is_there_a_blank_space() == False)

• 현재 바라보는 방향의 뒤쪽((d + 2) % 4) 칸을 검사했을 때, 갈 수 있는 칸이라면 이동한다.
• 아니라면 로봇청소기 동작을 중단한다.

is_there_a_blank_space() 함수는 다음과 같다.

def is_there_a_blank_space() -> bool:
    for dr, dc in offset:
        nr, nc = r + dr, c + dc
        
        if not out_of_bound(nr, nc) and room[nr][nc] == 0:
            return True
            
    return False

방향 오프셋 값을 하나씩 꺼내와 현재 좌표(r, c)에 더해 인접 칸 좌표(nr, nc)를 얻어온다.

• 만약 인접 칸이 청소되지 않은 빈 칸(0)이라면 True를 반환한다.
• 4방향 모두 청소되지 않은 빈 칸이 아니라면 False를 반환한다.

동작 3

else:
    d = (d - 1) % 4
    dr, dc = offset[d]
    nr, nc = r + dr, c + dc
    
    if not out_of_bound(nr, nc) and room[nr][nc] == 0:
        r, c = nr, nc

is_there_a_blank_space() 함수가 True인 경우 진입한다.

• 반시계 방향으로 90도 회전((d - 1) % 4)한다.
• 만약 회전한 방향의 인접 칸이 청소되지 않은 빈 칸(0)이라면 이동한다.

📚 새롭게 알게된 내용

엄 solved.ac에서 골랜디 할려고 검색해서 안 푼 거 하나 골랐는데 생각보다 너무 쉬운 걸 푼 것 같네요.
다음엔 좀 더 어려운 거 가져오겠습니다.

[mjj111]

고생하셨습니다! 여러 문제들도 있었는데 로봇청소기가 이슈에 있어서 로봇 청소기만 일단 리뷰했습니다.
cpp은 익숙하지 않은데 로직을 너무 설명을 잘하셔서 읽어졌습니다 ㅋㅋㅋㅋ
덕분에 많이 배웁니다 🙇‍♂️

[mjj111]

아래에서 사용하는 메서드와 같이 boolean의 경우 is로 네이밍하시면 더 읽기 좋을 것 같아요 👍

 bool OutOfBound(int InR, int InC) ->  bool isOutOfBound(int InR, int InC) 

[9kyo-hwang]

저도 예전엔 is를 붙였었는데, C에서 배열 인덱스 벗어나면 컴파일러에서 out of bound 에러를 띄어주는 거 보고 "굳이 is 안붙여도 될 듯?" 하면서 안붙이던게 지금까지 버릇이 됐네요 ㅋㅋ
다음부턴 다시 붙여보겠습니다 :)

[mjj111]

해당 라인을 아래와 같이 두고 break 으로 두면 else문을 줄 일 수 있지않을까요..?!

if(isOutOfBound(nr, nc) || Room[nr][nc] == WALL) {
    break; 
}
r = nr, c = nc;

[9kyo-hwang]

맞습니다. 다만 여기 같은 경우엔 문제에서 주어진 조건과 일치하도록 일부러 작성하긴 했습니다 ㅋㅋ

[mjj111]

혹시 중괄호는 읽기 편하게 하시려고, 위와 같이 사용하시고 계신건가요?!

[9kyo-hwang]

요거는 요새 제가 공부하고 있는 Unreal Engine의 C++ 코드 컨벤션과 비슷하게 맞추려고 이렇게 하는 중입니다 :)
언리얼 C++ 컨벤션이 중괄호를 다음 줄로 넘기더라구요.

[xxubin04]

리뷰가 늦었네요....🧎🏻‍♀️
30분만에 뚝딱 풀다니 역시...👍
저는 에러가 나서 하루 지나고 다시 풀었습니다 허허
코드만 스윽 남기고 가겠습니다!!

코드

input = open(0).readline
dir = [0, 1, 2, 3]  # 북, 동, 남, 서
dx = [-1, 0, 1, 0]
dy = [0, 1, 0, -1]
ans = 0

def now_clean(x, y, d):
    global ans
    if room[x][y] == 0:  # 현재 청소 X
        ans += 1
        room[x][y] = 2  # 청소 완료 표시
    around_clean(x, y, d)

def around_clean(x, y, d):
    for _ in range(4):
        d = (d + 3) % 4  # 반시계로 90도 회전
        if room[x + dx[d]][y + dy[d]] == 0:  # 앞의 빈칸이 청소 X
            now_clean(x + dx[d], y + dy[d], d)
            return
    back(x, y, d)

def back(x, y, d):
    bd = (d + 2) % 4  # 후진 방향
    if room[x + dx[bd]][y + dy[bd]] != 1:  # 벽이 아닌 경우 후진
        now_clean(x + dx[bd], y + dy[bd], d)

N, M = map(int, input().split())
initial_x, initial_y, initial_d = map(int, input().split())
room = [list(map(int, input().split())) for _ in range(N)]

now_clean(initial_x, initial_y, initial_d)
print(ans)

[gjsk132]

문제에서 숫자가 뒤죽박죽 섞여있는거 보고 고민했었는데... PR에 문제 정리된 거 보고 이해했네요 😢

문제에 주어진대로 구현하면 되지만...
약간 다른 방법으로 풀이한 방법으로 코드 남깁니다!

2번 과정에서 결국 주변을 돌면서 청소를 했는지 안했는지 확인하기 때문에
3번 과정을 먼저 해주고, 2번을 해주었습니다.

반시계로 4번 돌 때까지 ( 원상태가 될 때까지 ), 청소해야 하는 곳이 없다면
2번을 시행하는 것으로 하였습니다.

input = open("input.txt").readline

N, M = map(int, input().split())
r, c, d = map(int, input().split())
room = [list(map(int, input().split())) for _ in range(N)]
offset = ((-1, 0), (0, 1), (1, 0), (0, -1))


cnt = 0

while True:
    # 1
    if room[r][c] == 0:
        room[r][c] = 2
        cnt += 1
    
    is_clean = False

    # 3 (반시계로 회전)
    for turn in range(1, 5):
        dr, dc = offset[d-turn]
        nr, nc = r + dr, c + dc
        
        if nr < 0 or nr == N or nc < 0 or nc == M or room[nr][nc] != 0:
            continue

        r, c = nr, nc
        d = (d - turn)%4
        is_clean = True
        break

    # 2 (뒤로 한칸)
    if not is_clean:
        dr, dc = offset[d-2]
        nr, nc = r + dr, c + dc

        if nr < 0 or nr == N or nc < 0 or nc == M or room[nr][nc] == 1:
            break
        
        r, c = nr, nc

print(cnt)

파이썬은 마이너스로도 리스트 접근이 가능하고, 나머지를 계산하는 것도 잘 구해주기 때문에 냅다 빼고 다른 처리는 안해줬습니다.