[구현] 이코테 (파이썬) 상하좌우 풀이

[문제]

여행가 A는 N x N 크기의 정사각형 공간 위에 서 있다. 이 공간은 1 x 1 크기의 정사각형으로 나누어져 있다.

 

가장 왼쪽 위 좌표는 (1, 1) 이며, 가장 오른쪽 아래 좌표는 (N, N)에 해당한다. 여행가 A는 상, 하, 좌, 우 방향으로

 

이동할 수 있으며, 시작 좌표는 항상 (1, 1)이다. 우리 앞에는 여행가 A가 이동할 계획이 적힌 계획서가 놓여 있다.

 

계획서에는 하나의 줄에 띄어쓰기를 기준으로 하여 L, R, U, D 중 하나의 문자가 반복적으로 적혀 있다.

 

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

 

(1) L : 왼쪽으로 한 칸 이동
(2) R : 오른쪽으로 한 칸 이동
(3) U : 위로 한 칸 이동
(4) D : 아래로 한 칸 이동

 

이때 여행가 A가 N x N 크기의 정사각형 공간을 벗어나는 움직임은 무시된다.

 

...
... 생략
...

 

계획서가 주어졌을 때 여행가 A가 최종적으로 도착할 지점의 좌표를 출력하는 프로그램을 작성하시오.

 

[입력 조건]

1. 첫째 줄에 공간의 크기를 나타내는 N이 주어진다. (1 <= N <= 100)

 

2. 둘째 줄에 여행가 A가 이동할 계획서 내용이 주어진다. (1 <= 이동 횟수 <= 100)

 

[출력 조건]

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

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

 

[내 풀이]

n = int(input())
direction = input().split()

x, y = 1, 1
move = ['L', 'R', 'U', 'D']
gox = [0, 0, -1, 1]
goy = [-1, 1, 0, 0]
for i in direction :
  for j in range(len(move)) :
    if i == move[j] :
      result_x = x + gox[j]
      result_y = y + goy[j]

  if result_x < 1 or result_y < 1 or result_x > n or result_y > n :
    continue

  x = result_x
  y = result_y

print(x, y)

 

N을 입력받은 후 공백을 통해 여행가 A가 이동할 계획서 내용을 입력받는다. 또한 문제에서 시작좌표는 항상 (1, 1)이기 

 

때문에 변수 x, y에 각 1을 담는다. move 변수는 반복문내에서 계획서에 적혀있는 문자를 차례로 비교하여 수행하기 

 

위한 용도로 사용될 것이기 때문에 선언해준다. gox와 goy는 'L', 'R', 'U', 'D' 의 각 방향별로 이동될 거리이다.

 

반복문 내에서는 계획서에 적혀있는 문자와 방향 변수를 차례로 비교하여 해당 문자가 같다면 해당 위치에서 x와 y값을

 

특정 이동 거리만큼 누적해간다. 

 

또한, 문제에서 주어진 조건은 여행가 A가 N x N 크기의 정사각형 공간을 벗어나는 움직임은 무시된다고 했기 때문에

 

조건문을 통해 공간을 벗어나지 않을 때만 x, y 변수에 각각 현재 위치를 할당한다.

 

최종적으로 x와 y를 출력하여 도착할 지점의 좌표를 출력한다.

 

---

출처

이것이 코딩 테스트다 with 파이썬 - 나동빈 저