[백준/Python] 13567 로봇

 

📌 문제 탐색하기

정사각형 영역 S의 한변의 길이 M (1 ≤ M ≤ 1,000)

정사각형 영역 S 위, 왼쪽 하단 (0,0)에 로봇이 동쪽 방향으로 위치, 오른쪽 위 꼭짓점 (M,M)

명령어 n개(1 ≤ n ≤ 1,000)
└ TURN dir (dir = 0이면 현재 위치에서 왼쪽으로 90도 회전, dir = 1이면 현재 위치에서 오른쪽으로 90도 회전)
└ MOVE d (d는 1,000 이하의 양의 정수)

명령의 수행 후 로봇이 S의 바깥으로 나가게 되면 해당 명령어 열은 유효하지 않음
명령어 열이 유효하지 않다면 -1 출력
명령어 열이 유효하다면 명령어 수행 후 로봇의 최종 위치 x좌표 y좌표 출력

 

아래의 그림처럼 우리가 일반적으로 생각하는 이차원 리스트의 방향과 반대로 좌표가 설정되었다는 점을 제외하면 큰 특이사항이 없는 구현 문제이다.

---

📌 코드 설계하기

1. 정사각형 S 한변의 길이(M)와 명령어 개수(n) 입력받기
2. 방향 리스트(directions) 초기화하기 : [(1, 0), (0, 1), (-1, 0), (0,-1)]
3. 로봇의 위치와 방향 초기화하기 : x = y = 0, dir_idx = 0
4. n개의 명령어 입력받기
5-1. 명령어가 TURN 0 이라면 : (dir_idx+1)%4, TURN 1 이라면 : (dir_idx-1)%4
5-2. 명령어가 MOVE d라면 : nx, ny = x+d*directions[dir_idx][0], y+d*directions[dir_idx][1]
           nx, ny가 정사각형 영역 안에 있는지 확인 후, 영역을 벗어나면 -1 출력 후 프로그램 종료
6. 모든 명령 수행 후 최종 로봇의 위치 출력

---

📌 정답 코드

import sys
input = sys.stdin.readline

M, n = map(int, input().split())

directions = [(1, 0), (0, 1), (-1, 0), (0, -1)]
x = y = dir_idx = 0

orders = [input().split() for _ in range(n)]

for order, order_num in orders:
  order_num = int(order_num)
  if order == "TURN":
    dir_idx = (dir_idx-1)%4 if order_num else (dir_idx+1)%4
  else:
    dx, dy = order_num * directions[dir_idx][0], order_num * directions[dir_idx][1]
    nx, ny = x + dx, y + dy
    if not (0 <= nx <= M and 0 <= ny <= M):
      print(-1)
      break
    x, y = nx, ny
else:
  print(x, y)