상하좌우 - 구현 알고리즘

이것이 취업을 위한 코딩테스트다 110쪽

📖문제

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

L : 왼쪽으로 한 칸 이동
R : 오른쪽으로 한 칸 이동
U : 위로 한 칸 이동
D : 아래로 한 칸 이동
이때 여행가 A가 N x N 크기의 정사각형 공간을 벗어나는 움직임은 무시된다.
계획서가 주어졌을 때 여행가 A가 최종적으로 도착할 지점의 좌표를 출력하는 프로그램을 작성하시오

입력 조건

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

출력 조건

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

나의 풀이

N = int(input())
temp_list = list(input().split())
temp_dict = {'L': [0, -1], 'R': [0, 1], 'U': [-1, 0], 'D': [1, 0]}
result = [1, 1]
for i in temp_list:
    if N >= result[0] + temp_dict[i][0] >= 1 and N >= result[1] + temp_dict[i][1] >= 1:
        result[0] += temp_dict[i][0]
        result[1] += temp_dict[i][1]
print(result[0], result[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)

배운 점

• 나의 코드에서 리스트를 만들어서 더하면 된다고 생각했는데 이건 파이썬의 기본 문법을 생각하지 못한 나의 실수였다.

[0, 2] + [3, 4] ======> [0, 2, 3, 4]

[3, 6]이 될 줄 알았는데 그건 절대 아니었다.

• 이번 문제는 시뮬레이션 문제이다. 주어진 시나리오 대로 잘 구현을 해내는지에 대한 역량을 확인하는 문제다.