implementation(simulation): 상하좌우

문제정의

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

계획서에는 하나의 줄에 띄어쓰기를 기준으로 L, R, U, D 중 하나의 문자가 반복적으로 적혀있습니다.
각 문자의 의미는 다음과 같습니다.

L: 왼쪽으로 한 칸 이동
R: 오른쪽으로 한 칸 이동
U: 위로 한 칸 이동
D: 아래로 한 칸 이동

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

입력조건

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

출력조건

  • 첫째 줄에 게임의 룰에 맞게 선택한 카드에 적힌 숫자를 출력합니다.

입/출력예시

#입력 
5
R R R U D D

출력 
3 4

내가짠코드

n = int(input())
plans = input().split()

n = 5
plans = ['R','R','R','U','D','D']

curx, cury = 1, 1

for plan in plans:
    if plan == "L" and curx > 0 and curx <= n:
        curx -= 1
    if plan == "R" and curx >= 0 and curx < n:
        curx += 1
    if plan == "U" and cury > 0 and cury <= n:
        cury -= 1
    if plan == "D" and cury >= 0 and cury < n:
        cury += 1
print(curx, cury)

개선 코드

n = int(input())
plans = input().split()

n = 5
plans = ['R','R','R','U','D','D']

curx, cury = 1, 1

# 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 = curx + dx[i]
            ny = cury + dy[i]
    if nx < 1 or ny < 1 or nx > n or ny > n:
      	continue
  	curx, cury = nx, ny
    
print(curx, cury)

좋은 웹페이지 즐겨찾기