구현 (implementation) 실전 문제 - Python

예제 4 - 1 상하좌우

여행가 A는 NxN 크기의 정사각형 공간 위에 서 있다. 이 공간은 1x1 크기의 정사각형으로 나누어져 있다. 가장 왼쪽 위 좌표는 (1, 1)이며, 가장 오른쪽 아래 좌표는 (N, N)에 해당한다.

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

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

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

* 이때 여행가 A가 NxN 크기의 정사각형 공간을 벗어나는 움직임은 무시된다. 예를 들어 (1,1)의 위치에서 L 혹은 U를 만나면 무시된다.

다음은 N=5인 지도와 계획서이다.

시작	R	R	R
			D
			D

계획서 : R -> R -> R -> U -> D -> D

이 경우 6개의 명령에 따라서 여행 가가 움직이게 되는 순서대로 (1, 2), (1, 3), (1, 4), (1, 4), (2, 4), (3, 4)이므로, 최종적으로 여행가 A가 도착하게 되는 곳의 좌표는 (3, 4)이다.

다시 말해 3행 4열의 위치에 해당하므로 (3, 4)라고 적는다. 계획서가 주어졌을 때 여행가 A가 최종적으로 도착할 지점의 좌표를 출력하는 프로그램을 작성하시오.

입력 조건 

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

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

 

출력조건

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

 

입력 예시

5 R R R U D D

 

출력 예시

3 4

---

이러한 문제는 일련의 명령에 따라서 개체를 차례대로 이동시킨다는 점에서 시뮬레이션 유형으로 분류된다.

 

1. 처음 내가 생각한 풀이

# 4-1 상하좌우
N = int(input())
data = input().split()
row = 1  # 행
col = 1  # 열
for i in data:
    if i == 'L':
        if col == 1:
            pass
        else:
            col -= 1
    elif i == 'R':
        if col == N:
            pass
        else:
            col += 1
    elif i == 'U':
        if row == 1:
            pass
        else:
            row -= 1
    elif i == 'D':
        if row == N:
            pass
        else:
            row += 1
print(row, col)

나는 많은 생각 하지않고 조건문을 활용하여 풀었다.

 

2. 교재 답지

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)

 

두개의 코드의 아이디어는 모두 같다.

이동하면서 N x N 행렬이 넘어가면 무시 해주고 그렇지 않으면 이동방향으로 1만큼 옮겨주면 된다.

 

문제, 정답 출처 :

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