[구현] 이코테 - 상하좌우(py)

문제

예제 4 - 1

여행가 A는 N × N 크기의 정사각형 공간 위에 서 있다. 이 공간은 1 × 1 크기의 정사각형으로 나누어져 있다. 가장 왼쪽 위 좌표는 (1, 1)이며, 가장 오른쪽 아래 좌표는 (N, N)에 해당한다. 여행가 A는 상, 하, 좌, 우 방향으로 이동할 수 있으며, 시작 좌표는 항상 (1, 1)이다. 우리 앞에는 여행가 A가 이동할 계획이 적힌 계획서가 놓여 있다. 계획서에는 하나의 줄에 띄어쓰기를 기준으로 L, R, U, D 중 하나의 문자가 반복적으로 적혀있다.

 

각 문자의 의미는 다음과 같다.

 

L: 왼쪽으로 한 칸 이동
R: 오른쪽으로 한 칸 이동
U: 위로 한 칸 이동
D: 아래로 한 칸 이동

 

이때 여행가 A가 N × N 크기의 정사각형 공간을 벗어나는 움직임은 무시된다. 예를 들어 (1, 1)의 위치에서 L 혹은 U를 만나면 무시된다.
다음은 N = 5인 지도와 계획이다.

 

입력 조건

• 첫째 줄에 공간의 크기를 나타내는 N이 주어진다. (1 <= N <= 100)
• 둘째 줄에 여행가 A가 이동할 계획서 내용이 주어진다. (1 <= 이동 횟수 <= 100)

출력 조건

• 첫째 줄에 여행가 A가 최종적으로 도착할 지점의 좌표 (X, Y)를 공백으로 구분하여 출력한다.

 

입력 예시	출력 예시
5	3 4
R R R U D D

 

---

풀이

- 내 풀이

n = int(input())

x = [1, 1]

plans = input().split()

for p in plans:
  if p == "R":
    if x[1] + 1 <= n:
      x[1] += 1
  elif p == "L":
    if x[1] - 1 > 0:
      x[1] -= 1
  elif p == "U":
    if x[0] - 1 > 0:
      x[0] -= 1
  elif p == "D":
    if x[0] + 1 <= n:
      x[0] += 1

print(x[0], x[1])

 

- 책 풀이

n = int(input())
x, y = 1, 1
plans = input().split()

dx = [0, 0, -1, 1]
dy = [-1, 1, 0, 0]
move_types = ['L', 'R', 'U', 'D']

for plan in plans:
    for i in range(len(move_types)):
        if plan == move_types[i]:
            nx = x + dx[i]
            ny = y + dy[i]
    if nx < 1 or ny < 1 or nx > n or ny > n:
        continue
    x, y = nx, ny

print(x, y)

 

아이디어

• 연산 횟수는 이동 횟수에 비례
• 일련의 명령에 따라 개체를 차례대로 이동시킨다는 점에서 시뮬레이션(Simulation) 유형으로 분류되며 구현이 중요한 대표적인 문제 유형
• 상하좌우 => 이동 방향 알고 있기!

 

 

---

 

출처: 나동빈, 「이것이 취업을 위한 코딩테스트다 with 파이썬」, 한빛미디어, p.110