[Python] 상하좌우

문제

• (1, 1) 에서 출발하여 계획서에 따라 이동한 후, 도착할 지점의 좌표를 출력하는 프로그램을 작성하시오
• L : 왼쪽으로 한 칸 이동
• R : 오른쪽으로 한 칸 이동
• U : 위로 한 칸 이동
• D : 아래로 한 칸 이동

입력

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

출력

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

문제 풀이

• 좌우상하를 이동할 수 있는 좌표 리스트를 만들어 이동
• 입력 받은 지도만큼 for 문을 돌면서 이동할 위치가 리스트에 저장된 값과 같으면 이동
• 범위를 벗어나는 경우 (지도 밖 좌표) 무시
• 좌표 출력

코드

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

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

for plan in plans:
    for i in range(len(move)):
        if plan == move[i]:
            nx = x + dx[i]
            ny = y + dy[i]

    if nx < 1 or ny < 1 or nx > n or ny > n:
        continue
    x = nx
    y = ny
    
print(y, x)