[Python][이코테] 상하좌우

조건

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

가장 왼쪽 위 좌표는 (1, 1)이며, 가장 오른쪽 아래 좌표는 (N, N)에 해당한다.

 

각 L R U D (좌 우 상 하)를 입력받으며 도착 지점을 찾는다.

(1, 1) 지점에서 L U 를 입력받으면 공간을 벗어나므로 무시된다.

 

입력 조건

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

출력 조건

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

 

 

입력 예시

5
R R R U D D

 

 

출력 예시

3 4

 

 

---

 

 

입력 받은 문자열을 (1, 1) 지점을 기준으로 더하고 빼며

 

현 위치 기준 L U 이 1 좌표일 때 이동 불가

R D 가 N 좌표일때 이동 불가능하게 예외처리를 해주면 된다고 생각했다.

 

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

for plan in plans:
    if plan == "L" and y>1:
        y -= 1
    elif plan == "R" and y<n:
        y += 1
    elif plan == "U" and x>1:
        x -= 1
    elif plan == "D" and x<n:
        x += 1

print(x, y)

단순하게 입력받은 문자열을 문자열 갯수만큼 반복문을 돌려주며

 

switch문을 사용한것처럼 예외처리를 해 주었다.

 

이번 문제는 쉬웠다.

 

물론 저게 정답인지 나는 모른다.. 어떤 히든 케이스가 있을지

---

저자의 정답

 

# N 입력받기
n = int(input())
x, y = 1, 1
plans = input().split()

# L, R, U, D에 따른 이동 방향
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)