🟩 [백준] [Python] [Gold5] 14503번_로봇 청소기

문제

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

14503번: 로봇 청소기

풀이

# 백준 14503번 문제 - 로봇 청소기
import sys
from collections import deque
input = sys.stdin.readline

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

# d => 0,3,2,1 순서로 돌아야한다.
dx = [-1,0,1,0]
dy = [0,1,0,-1]


# 처음 시작하는 곳 방문 처리
visited[r][c] = 1
cnt = 1

while 1:
    flag = 0
    # 4방향 확인
    for _ in range(4):
        # 0,3,2,1 순서 만들어주기위한 식
        nx = r + dx[(d+3)%4]
        ny = c + dy[(d+3)%4]
        # 한번 돌았으면 그 방향으로 작업시작
        d = (d+3)%4
        if 0 <= nx < n and 0 <= ny < m and graph[nx][ny] == 0:
            if visited[nx][ny] == 0:
                visited[nx][ny] = 1
                cnt += 1
                r = nx
                c = ny
                #청소 한 방향이라도 했으면 다음으로 넘어감
                flag = 1
                break
    if flag == 0: # 4방향 모두 청소가 되어 있을 때,
        if graph[r-dx[d]][c-dy[d]] == 1: #후진했는데 벽
            print(cnt)
            break
        else:
            r,c = r-dx[d],c-dy[d]

알고리즘 기술 보다는 문제에 나온 내용 그대로 그냥 구현을 하면 되는 문제이다. 

근데 처음 풀어보는 문제이기도 하고 회전하는 방법이 헷갈리고 문제에 대한 이해를 하느라 시간이 조금 소요됐다. 

다시 한번 풀어보도록 해야겠다. 

 

 

출처 : https://resilient-923.tistory.com/164