문제 - 상하좌우
-
여행가 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