정재영

1일 1문제 푸는 중

문제 설명

지나다니는 길을 ‘O’, 장애물을 ‘X’로 나타낸 직사각형 격자 모양의 공원에서 로봇 강아지가 산책을 하려합니다. 산책은 로봇 강아지에 미리 입력된 명령에 따라 진행하며, 명령은 다음과 같은 형식으로 주어집니다.

  • [“방향 거리”, “방향 거리” … ]

예를 들어 “E 5”는 로봇 강아지가 현재 위치에서 동쪽으로 5칸 이동했다는 의미입니다. 로봇 강아지는 명령을 수행하기 전에 다음 두 가지를 먼저 확인합니다.

  • 주어진 방향으로 이동할 때 공원을 벗어나는지 확인합니다.
  • 주어진 방향으로 이동 중 장애물을 만나는지 확인합니다.

위 두 가지중 어느 하나라도 해당된다면, 로봇 강아지는 해당 명령을 무시하고 다음 명령을 수행합니다.
공원의 가로 길이가 W, 세로 길이가 H라고 할 때, 공원의 좌측 상단의 좌표는 (0, 0), 우측 하단의 좌표는 (H - 1, W - 1) 입니다.

공원을 나타내는 문자열 배열 park, 로봇 강아지가 수행할 명령이 담긴 문자열 배열 routes가 매개변수로 주어질 때, 로봇 강아지가 모든 명령을 수행 후 놓인 위치를 [세로 방향 좌표, 가로 방향 좌표] 순으로 배열에 담아 return 하도록 solution 함수를 완성해주세요.

제한사항

  • 3 ≤ park의 길이 ≤ 50
    • 3 ≤ park[i]의 길이 ≤ 50
      • park[i]는 다음 문자들로 이루어져 있으며 시작지점은 하나만 주어집니다.
        • S : 시작 지점
        • O : 이동 가능한 통로
        • X : 장애물
    • park는 직사각형 모양입니다.
  • 1 ≤ routes의 길이 ≤ 50
    • routes의 각 원소는 로봇 강아지가 수행할 명령어를 나타냅니다.
    • 로봇 강아지는 routes의 첫 번째 원소부터 순서대로 명령을 수행합니다.
    • routes의 원소는 “op n”과 같은 구조로 이루어져 있으며, op는 이동할 방향, n은 이동할 칸의 수를 의미합니다.
      • op는 다음 네 가지중 하나로 이루어져 있습니다.
        • N : 북쪽으로 주어진 칸만큼 이동합니다.
        • S : 남쪽으로 주어진 칸만큼 이동합니다.
        • W : 서쪽으로 주어진 칸만큼 이동합니다.
        • E : 동쪽으로 주어진 칸만큼 이동합니다.
      • 1 ≤ n ≤ 9
  • 문제 보러가기


문제 풀이

(1) 코드 작성

def solution(park, routes):
    # 행,열로 생각했을 때 행을 x, 열을 y로 저장한다.
    x, y = [[j, i] for i, p in enumerate(park) for j, s in enumerate(p) if s == 'S'][0]
    for route in routes:
        direc, steps = route.split()
        new_x, new_y = x, y
        # 방향별로 공원을 벗어나면 안되고 장애물을 만나선 안된다.
        for step in range(int(steps)):
            if direc == 'E' and new_x != len(park[0])-1 and park[new_y][new_x+1] != 'X':
                new_x += 1
                if step == int(steps)-1:
                    x = new_x
            elif direc == 'W' and new_x != 0 and park[new_y][new_x-1] != 'X':
                new_x -= 1
                if step == int(steps)-1:
                    x = new_x
            elif direc == 'S' and new_y != len(park)-1 and park[new_y+1][new_x] != 'X':
                new_y += 1
                if step == int(steps)-1:
                    y = new_y
            elif direc == 'N' and new_y != 0 and park[new_y-1][new_x] != 'X':
                new_y -= 1
                if step == int(steps)-1:
                    y = new_y
    return [y, x]

(2) 코드 리뷰

우선 제출해야 될 답은 행으로 이동한 값을 리스트의 두 번째 값으로, 열을 기준으로 이동한 값을 첫 번째 값으로 출력해야 했다. 코드를 작성할 때 이해력을 높이기 위해 행과 열의 값을 x, y 값으로 지정하고 마지막에 두 값의 위치를 바꾸어 출력하도록 했다. 이동 루트는 시작 위치에서 이동 과정에서 장애물이 없거나 공원을 벗어나지 않는 경우에서만 그 루트대로 움직인다. 그래서 방향별로(동서남북) 조건문을 작성해서 한 칸씩 이동하도록 했고 만약 이동 과정에서 움직이려는 거리만큼 잘 이동했다면 x, y 값에 저장해 두었다. 이번 문제는 여러 방법으로 시도해 보다가(따로 장애물이 있는 위치를 저장한 리스트를 만들기도 함) 위 방법으로 문제를 해결했다. 쉽진 않았다.


👩🏻‍💻개인 공부 기록용 블로그입니다
오류나 틀린 부분이 있을 경우 댓글 혹은 메일로 따끔하게 지적해주시면 감사하겠습니다.

댓글남기기