[이코테] 상하좌우

출처: 이것이 취업을 위한 코딩테스트다  with 파이썬 (나동빈)

https://www.youtube.com/watch?v=2zjoKjt97vQ&list=PLRx0vPvlEmdAghTr5mXQxGpHjWqSz0dgC&index=2

 

 

문제

• 여행가 A는 N*N크기의 정사각형 공간 위에 서 있습니다. 이 공간은 1*1 크기의 정사각형으로 나누어져 있습니다. 가장 왼쪽 위 좌표는 (1,1)이며, 가장 오른쪽 아래 좌표는 (N,N)에 해당합니다. 여행가 A는 상,하,좌,우 방향으로 이동할 수 있으며, 시작 좌표는 항상 (1,1)입니다. 우리 앞에는 여행가 A가 이동할 계획이 적힌 계획서가 놓여 있습니다.
• 계획서에는 하나의 줄에 띄어쓰기를 기준으로 하여 L, R, U, D 중 하나의 문자가 반복적으로 적혀 있습니다. 각 문자의 의미는 다음과 같습니다.
  • L: 왼쪽으로 한 칸 이동
  • R: 오른쪽으로 한 칸 이동
  • U: 위로 한 칸 이동
  • D: 아래로 한 칸 이동
• 이때 여행가 A가 N*N크기의 정사각형 공간을 벗어나는 움직임은 무시됩니다.

 

문제 조건

• 풀이시간: 15분
• 시간제한: 2초
• 메모리 제한: 128MB

---

단순 구현 문제로 문제에서 하라는 대로만 하면 된당

 

작성 코드

N = int(input())
data = list(input().split())

x,y = 1,1

for i in data:
  if i=='L':
    nextX=x
    nextY=y-1
  elif i=='R':
    nextX=x
    nextY=y+1
  elif i=='U':
    nextX=x-1
    nextY=y
  elif i=='D':
    nextX=x+1
    nextY=y

  if 1<=nextX<=N and 1<=nextY<=N:
    x=nextX
    y=nextY


print(x,y)

 

L,R,U,D에 따라 다음 좌표를 구하고 그 좌표로 이동 가능하다면(범위 체크) 이동하는 식으로 코드를 작성했다.

어차피 이동 가능한 칸이 인접한 상하좌우 4개밖에 없는 매우 단순한 문제라서 무지성  if-else로 적었는데,

이렇게 하니까 좀 머랄까 없어보이는 것 같긴 하다.

 

강의 보고 수정한 코드

N = int(input())
plans = list(input().split())

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

x,y = 1,1

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

  if 1<=nx<=N and 1<=ny<=N:
      x,y = nx,ny

print(x,y)