상하좌우 [구현]

구현 실전 문제

상하좌우

🐾 문제 설명

여행가 A는 N x N 크기의 정사각형 공간 위에 서 있다. 이 공간은 1 x 1 크기의 정사각형으로 나누어져 있다. 가장 왼쪽 위 좌표는 (1, 1)이며, 가장 오른쪽 아래 좌표는 (N, N)에 해당한다. 여행가 A는 상, 하, 좌, 우 방향으로 이동할 수 있으며, 시작 좌표는 항상 (1, 1)이다.
계획서가 주어졌을 때 여행가 A가 최종적으로 도착할 지점의 좌표를 출력하는 프로그램을 작성하시오.
단, 여행가 A는 N x N 크기의 정사각형 공간을 벗어나는 움직임은 무시된다.

🐾 입출력 예시

• 입력
  5
  R R R U D D
• 출력
  3 4

🐾 내 풀이

N = int(input())
action = list(input().split())

UD = 1
LR = 1

for i in action:
    if i == "U" and UD > 1:
        UD -= 1
    elif i == "D" and UD < N:
        UD += 1
    elif i == "L" and LR > 1:
        LR -= 1
    elif i == "R" and LR < N:
        LR += 1

print(UD, LR)

문제에서 제시한 상황에 따른 알고리즘을 하나씩 코드로 구현하여 문제에서 원하는 결과값이 나올 수 있도록 코드를 작성하였다.

🐾 답안 예시

< 이것이 취업을 위한 코딩 테스트다 with 파이썬 > 답안 예시

# 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)

dx, dy, move_types의 배열을 만들어 move_types 값에 따라 index로 각각의 상황에 x, y에 변화를 주는 방식으로 코드를 작성하였는데, 다른 알고리즘 문제를 풀 때에 이러한 방식을 이용해보면 좋을 것 같다.

---

참고자료
이것이 취업을 위한 코딩 테스트다 with 파이썬 - 나동빈