<청춘> 격정적으로 사는 것

밤을 새고 공부한 다음 날 새벽에 느꼈던 생생한 환희와 야생적인 즐거움을 잊을 수 없다

파이썬 Python

[구현] 구현 예제 : 상하좌우 / 코딩테스트 / 파이썬

수학도 2021. 8. 3. 16:16

코딩 테스트에서 구현(Implementation)이란?

머릿속에 있는 알고리즘을 소스코드로 바꾸는 과정을 의미한다.

 

완전 탐색

모든 경우의 수를 저저 없이 다 계산하는 해결 방법을 의미한다.

 

시뮬레이션

문제에서 제시한 알고리즘을 한 단계씩 차례대로 직접 수행해야 하는 문제 유형을 의미한다.

 


예제<4-1> 상하좌우 : 시뮬레이션

 

여행가 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 크기의 정사각형 공간을 벗어나는 움직임은 무시된다.

예를 들어 (1, 1)의 위치에서 L 혹은 U를 만나면 무시된다.

 

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

(1, 1) 출발 (1, 2) (1, 3) (1, 4) (1, 5)
(2, 1) (2, 2) (2, 3) (2, 4) (2, 5)
(3, 1) (3, 2) (3, 3) (3, 4) 도착 (3, 5)
(4, 1) (4, 2) (4, 3) (4, 4) (4, 5)
(5, 1) (5, 2) (5, 3) (5, 4) (5, 5)

계획서 : R → R → R → U → D → D

 

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

 

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

 

입력 조건

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

 

출력 조건

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

 

입력 예시

5
R R R U D D

 

출력 예시

3 4

소스코드

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

dx = [0, 0, -1, 1]
dy = [-1, 1, 0, 0]
direction = ['L', 'R', 'U', 'D']

for plan in plans:
    # 이동 후 좌표 구하기
    for i in range(len(direction)):
        if plan == direction[i]:
            nowx = x + dx[i]
            nowy = y + dy[i]
    # 이동 후 좌표가 지도 범위를 벗어나면 무시
    if nowx < 1 or nowy <1 or nowx > N or nowy > N:
        continue
    # 벗어나지 않으면 이동
    x, y = nowx, nowy

print(x, y)