<dfs, bfs> 백준 14503번. 로봇 청소기 (골드 5)

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

14503번: 로봇 청소기

import sys
from collections import deque
input = sys.stdin.readline
  
n, m = map(int, input().split()) #세로, 가로
r, c, d = map(int, input().split()) #시작 좌표, 방향
# 0:북, 1:동, 2:남, 3:서
lst = []
visited = [[0] * m for _ in range(n)]
dx = [-1,0,1,0]
dy = [0,1,0,-1] #북, 동, 남, 서
for _ in range(n):
  lst.append(list(map(int, input().split())))

#step 1
visited[r][c] = 1 # 시작 지점 -> 청소 완료
cnt = 1

while 1:
  wall = [1,1,1,1] # 네 뱡향 모두 청소 or 벽인 경우를 위함
  #step 2
  for i in range(4): #모든 방향을 회전하기 위함
    # print(*lst, sep='\n')  
    d = (d+3)%4 # 왼쪽으로 회전
    nx,ny = r + dx[d], c + dy[d] # 그 방향으로 한 칸 전진
    if 0 <= nx < n and 0 <= ny < m and lst[nx][ny] == 0: #step2-2:회전한 방향으로 1칸 전진했을 때, 청소가 필요하다면
      if visited[nx][ny] == 0: #방문하지 않은 장소면
        visited[nx][ny] = 1 # 방문 처리
        r, c = nx, ny
        cnt += 1 # 청소+1
        wall[i] = 0
        break

  #step 2-4    
  if sum(wall)==4 : #네 방향 모두 막혀 있는 경우
    if lst[r-dx[d]][c-dy[d]] == 0: #후진은 가능한 경우
      r,c = r-dx[d], c-dy[d] #후진
    else : # 후진도 못하면
      break #작동멈춤
  
    
 
print(cnt)