[Python] 14503. 로봇 청소기

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

14503번: 로봇 청소기

---

문제 조건에 첫행,끝행,첫열,끝열은 모두 벽으로 막혀있다고 했으므로

next_r과 next_c가 이동할 수 있는지 확인할 필요 없다.

문제를 잘 읽어보면 알고리즘 구현의 힌트를 찾을 수 있다.

di = ((-1,0),(0,1),(1,0),(0,-1))

N, M = map(int, input().split())
r, c, d = map(int, input().split())
MAP = [list(map(int, input().split())) for _ in range(N)]
check = [[False]*M for _ in range(N)]
cnt = 0

while True:
    check[r][c] = True
    next_d = (d-1)%4
    next_r = r + di[next_d][0]
    next_c = c + di[next_d][1]

    if MAP[next_r][next_c] == 0 and not check[next_r][next_c]:
        d = next_d
        r = next_r
        c = next_c
        cnt = 0
    elif MAP[next_r][next_c] == 1 or check[next_r][next_c] and cnt != 4:
        d = next_d
        cnt += 1

    if cnt == 4:
        next_r = r - di[d][0]
        next_c = c - di[d][1]
        if MAP[next_r][next_c] == 0:
            r = next_r
            c = next_c
            cnt = 0
        else:
            break

answer = 0
for r in range(N):
    for c in range(M):
        if check[r][c]:
            answer += 1

print(answer)

 

문제를 푸는데 도움이 된 테스트 케이스

# Input 4 4 1 1 0 1 1 1 1 1 0 0 1 1 1 0 1 1 1 1 1 # Output 3