로봇 청소기 (feat. 백준, 14503번)

로봇 청소기
https://www.acmicpc.net/problem/14503

14503번: 로봇 청소기

반응형

# woohyeon.kim
N, M = map(int, input().split())
x, y, d = map(int, input().split())
arr = []
for _ in range(N):
  arr.append(list(map(int, input().split())))

# 북 동 남 서
dx = [-1, 0, 1, 0]
dy = [0, 1, 0, -1]

def spin():
  global d
  if d == 0:
    d = 3
  elif d == 1:
    d = 0
  elif d == 2:
    d = 1
  else:
    d = 2

count = 0
while True:
  if arr[x][y] == 0:
    arr[x][y] = 2
    count += 1

  check = False
  for _ in range(len(dx)):
    spin()
    nx, ny = x + dx[d], y + dy[d]
    if 0 <= nx < N and 0 <= ny < M:
      if arr[nx][ny] == 0:
        x, y, check = nx, ny, True
        break;

  if not check:
    nx, ny = x - dx[d], y - dy[d]
    if 0 <= nx < N and 0 <= ny < M:
      if arr[nx][ny] == 1:
        break
      else:
        x, y = nx, ny
    else:
      break;

print(count)
# *&)*@*

동서남북 방향이 트릭이었습니다...