구현 : 상하좌우 [ 이것이 코딩테스트다 ] [ Python ]

2023. 5. 8. 22:39알고리즘/이것이 코딩테스트다

 

문제

## 상하좌우

여행가 A는 N * X 크기의 정사각형 공간 위에 서 있다. 이 공간은 1 * 1 크기의 정사각형으로 나누어져 있다.
가장 왼쪽 위 좌표는 (1, 1) 이며, 가장 오른쪽 아래 좌표는 (N, N)에 해당한다.
여행가 A는 상, 하, 좌, 우 방향으로 이동할 수 있으며, 시작 좌표는 항상 (1, 1) 이다. 

계획서에는 하나의 줄에 띄어쓰기를 기준으로 하여 L, R, U, D 중 하나의 문자가 반복적으로 적혀있다. 
각 문자의 의미는 다음과 같다.
- L : 왼쪽으로 한 칸 이동
- R : 오른쪽으로 한 칸 이동
- U : 위로 한 칸 이동
- D : 아래로 한 칸 이동

이때 여행가 A 가 N * N 크기의 정사각형 공간을 벗어나는 움직임은 무시한다. 
예를 들어 (1, 1) 의 위치에서 L 혹은 U 를 만나면 무시된다.
다음은 N=5 인 지도와 계획서이다.

계획서와 지도
R -> R -> R -> U -> D -> D

여행가A        
         
      목적지  
         
         



이 경우 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


코드

# 나의 코드

n = input()

move = list(map(str, input().split()))

its_me = [1, 1]

for i in move:
    if i == "R":
        its_me[1] += 1
    elif i == "L":
        its_me[1] -= 1
    elif i == "U":
        its_me[0] -= 1
    elif i == "D":
        its_me[0] += 1
    
    if its_me[0] == 0:
        its_me[0] += 1
    elif its_me[1] == 1:
        its_me[1] += 1
    
print(its_me[0], its_me[1])

 

저자 코드

# N 입력받기
n = int(input())
x, y = 1, 1
plans = input().split()

# L, R, U, D에 따른 이동 방향
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)

회고

저자 코드에 비하면 너무 단순하게 생각해 1차원적으로 문제를 푼것 같다.

그래도 문제를 푼것만으로도 만족한다. 총 3번 회독하며 문제를 풀 생각이니 다음에 풀땐 정답에 근접한 접근을 할 수 있게

연습을 많이 해야겠다.