[이것이 취업을 위한 코딩테스트다] 구현 - 상하좌우

 

문제 

 

여행가 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)

 

 

출력 

 

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

 

 

나의 코드

# size와 여행 계획서 변수 설정
size = int(input())
way = list(map(str,input().split()))

# 현재 위치를 배열로 지정 
now = [1, 1]

for i in way:
  # 오른쪽으로 이동하는 경우 now[1]값이 최대 size만큼만 이동 가능
  if i == "R":
    if now[1] < size:
      now[1] += 1
  # 왼쪽으로 이동하는 경우 now[1]값이 최소 1까지만 이동 가능 
  elif i == "L":
    if now[1] > 1:
      now[1] -= 1
  # 위로 이동하는 경우 now[0]값이 최소 1까지만 이동 가능
  elif i == "U":
    if now[0] > 1:
      now[0] -= 1

  # 아래로 이동하는 경우 최대 now[0]값이 최대 size만큼만 이동 가능
  else:
    if now[0] < size:
      now[0] += 1
      
print(now[0],now[1])

 

 

해설 코드에 대한 분석 

# 크기 입력 
n = int(input())
# 현재 위치 입력
x,y = 1,1
# 여행 계획 입력 
plans = input().split()

# 이동조건 
dx = [0,0,-1,1]
dy = [-1,1,0,0]
move_types = ["L","R","U","D"]

# 여행 계획을 하나씩 훑어봄
for plan in plans:
  # move_types의 길이만큼 반복하며, move_types의 원소와 여행계획 비교
  for i in range(len(move_types)):
    if plan == move_types[i]:
      nx = x + dx[i]
      ny = y + dy[i]

  # nx,ny의 경우 모두 최대는 n, 최소는 1이기 때문에 예외사항들을 적어줌
  if nx < 1 or ny < 1 or nx > n or ny > n:
    continue

  x,y = nx,ny

print(x,y)

여기서 기억해야 할 점은 move_types 라는 배열을 만들고 이들의 x,y 의 움직임을 각각 따라 dx 와 dy에 분리하여 적어, 반복문을 돌 때 move_types와 같은 인덱스를 가진 dx, dy 배열 속 원소들을 현재 위치에서 더하고 빼게 만들었다는 점이다. 나는 이들의 움직임을 각각 조건문에 적어서 해결하였는데, 이렇게 반복문을 활용해 move_types 와 plans 배열을 비교할 수 있다는 점이 인상깊었다. 그리고 마지막에 예외 사항들은 continue로 빼주어서 조건에 부합하는 경우만 반복문을 돌 수 있도록 만들어준 점이 효율적인 것 같다.