[알고리즘 문제 풀이] 상하좌우

이동욱

2021/08/31

Categories: 개발

문제 - 상하좌우

  • 여행가 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 ~ 100)
  • 둘 째 줄에 여행가 A가 이동할 계획서가 주어진다. (1 ~ 100)

출력

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

  • 정사각형 공간을 벗어나는 움직임은 무시된다.

아이디어

  • 계획서를 읽고 현재 좌표에 더해본다. (L, R, U, D)
  • 정사각형 공간을 벗어났는지 판단한다.
    • IF (1 <= X <= N, 1 <= Y <= N)
  • 이상이 없다면 좌표를 갱신하고 다음 계획서를 끝날때까지 읽는다.

코드

from typing import List
from typing import Tuple


def solution(n: int, maps: List[str]) -> Tuple[int, int]:
    current_x = 1
    current_y = 1

    for map in maps:
        if map == 'L':
            if 1 <= current_x - 1 <= n:
                current_x -= 1
        elif map == 'R':
            if 1 <= current_x + 1 <= n:
                current_x += 1
        elif map == 'U':
            if 1 <= current_y - 1 <= n:
                current_y -= 1
        elif map == 'D':
            if 1 <= current_y + 1 <= n:
                current_y += 1

    return current_y, current_x


def solution2(n: int, plans: List[str]) -> Tuple[int, int]:
    x, y = 1, 1
    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
    return x, y


if __name__ == '__main__':
    n = int(input())
    maps = list(input().split())
    print(solution(n, maps))

개선할 점

  • 좌표 문제를 풀 때 배열에 미리 상, 하, 좌, 우 값을 입력해두는 것이 좀 더 깔끔한것 같다.

  • 맵을 읽을 때 리스트에 가능한 수를 넣어두고 반복문으로 일치하는지 확인하는 패턴도 유용하게 사용할 수 있을 것 같다.

참고 문헌

>> Home