[04구현] 4-1 상하좌우(p110~112)

문제:

여행가 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인 지도와 계획서이다.

 

이 경우 6개 명령에 따라서 여행가가 움직이게 되는 위치는 순서대로 (1, 2), (1, 3), (1, 4), (1, 4), (2, 4), (3, 4) 이므로, 최종적으로 여행가 A가 도착하게 되는 곳의 좌표는 (3, 4)이다. 다시 말해 3행 4열의 위치에 해당하므로 (3, 4)라고 적는다. 계획서가 주어졌을 때 여행가 A가 최종적으로 도착할 지점의 좌표를 출력하는 프로그램을 작성하시오.

 

[입력 조건]

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

[출력 조건]

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

[입력 예시]

5
R R R U D D

 

[출력 예시]

3 4

 

[문제 해설]

• 요구 사항대로 구현시 연산 횟수는 이동 횟수에 비례함. 즉 O(N)의 시간복잡도를 가짐.

[소스 코드]

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=1, y=1.
• plans: 공백으로 구분된 이동 명령들의 리스트.
• move_types=['L','R','U','D']에 대응하는 이동 벡터를 dx, dy로 둔다.
  • L: (0,−1), R: (0,1), U: (−1,0), D: (1,0)
• 각 plan에 대해:
  1. move_types에서 일치하는 인덱스 i를 찾고 nx=x+dx[i], ny=y+dy[i] 계산.
  2. 새 좌표가 격자 밖이면 continue로 건너뛴다.
  3. 아니면 x,y를 갱신.
• 최종 (x, y)를 출력.

[좌표계 해석]

 

• 행을 x, 열을 y로 보고, 위로 가면 x-1, 아래로 가면 x+1.
• 왼쪽은 y-1, 오른쪽은 y+1.
• 유효 범위는 1 ≤ x,y ≤ n