[프로그래머스] 공원 산책 - 파이썬

링크: https://school.programmers.co.kr/learn/courses/30/lessons/172928

문제 설명

지나다니는 길을 '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

---

입출력 예

park	routes	result
["SOO","OOO","OOO"]	["E 2","S 2","W 1"]	[2,1]
["SOO","OXX","OOO"]	["E 2","S 2","W 1"]	[0,1]
["OSO","OOO","OXO","OOO"]	["E 2","S 3","W 1"]	[0,0]

---

입출력 예 설명

입출력 예 #1

입력된 명령대로 동쪽으로 2칸, 남쪽으로 2칸, 서쪽으로 1칸 이동하면 [0,0] -> [0,2] -> [2,2] -> [2,1]이 됩니다.

입출력 예 #2

입력된 명령대로라면 동쪽으로 2칸, 남쪽으로 2칸, 서쪽으로 1칸 이동해야하지만 남쪽으로 2칸 이동할 때 장애물이 있는 칸을 지나기 때문에 해당 명령을 제외한 명령들만 따릅니다. 결과적으로는 [0,0] -> [0,2] -> [0,1]이 됩니다.

입출력 예 #3

처음 입력된 명령은 공원을 나가게 되고 두 번째로 입력된 명령 또한 장애물을 지나가게 되므로 두 입력은 제외한 세 번째 명령만 따르므로 결과는 다음과 같습니다. [0,1] -> [0,0]

---

문제풀이

문제 이해하기:

공원을 나타내는 문자열 배열 park, 로봇 강아지가 수행할 명령이 담긴 문자열 배열 routes가 매개변수로 주어진다. 로봇 강아지는 명령을 수행하기 전 다음 두 가지를 먼저 확인함.

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

만약 위 두 가지중 어느 하나라도 해당되면, 해당 명령을 아예 무시함.

목표: 로봇 강아지가 모든 명령을 수행한 후 놓인 위치를 [세로 방향 좌표, 가로 방향 좌표] 순으로 배열에 담아 return

내 풀이:

동서남북에 따라서 x, y 좌표를 + 할지 - 할지 선언해야하고, routes 를 돌면서 다음 조건들에 해당하는지 살펴야 한다.

• 주어진 방향으로 이동할 때 공원을 벗어나는지 확인.
  • 최대 높이, 최대 넓이, 그리고 0 보다 작아지는지 확인해야하고
• 주어진 방향으로 이동 중 장애물을 만나는지 확인.
  • 이동했을 때 해당 좌표의 park 가 ‘X’ 인지 확인

위의 조건들에 해당하면 움직일 수 없고, x와 y의 좌표는 마지막 상태로 돌아가야 함.

따라서 해야할 작업을 순서대로 정리해보면

1. 동서남북 x, y 좌표를 + 할지 - 할지 선언
2. park 를 돌면서 ‘S’ 를 찾아 시작 좌표 x, y 찾기
3. routes 를 돌면서 가야하는 x, y 좌표와 몇 번 가야하는지 나타내는 n 을 split 해서 저장
4. 이후 잠재적인 이동 경로를 나타내는 임시 좌표 nx, ny 를 x, y 로 선언, 움직일 수 있는지를 나타내는 변수 can_move 를 True 로 설정
5. n 만큼 for 문을 돌면서 움직일 수 있는지 없는지 확인하며 x, y 를 업데이트 한다.

최종코드

def solution(park, routes):
    answer = []
    op = {"N": [-1, 0], "S": [1, 0], "E": [0, 1], "W": [0, -1]}
    
    # 시작점 찾기
    max_width = len(park[0])
    max_height = len(park)
    for i in range(max_height):
        for j in range(max_width):
            if park[i][j] == 'S':
                x, y = i, j

    # 명령을 돌면서 
    for i in routes:
        dx, dy = op[i.split()[0]]  # 가야하는 x 좌표(수평)와 y 좌표(수직)를 각각 저장
        n = int(i.split()[1])  # 몇 번 가야하는지 저장
        nx, ny = x, y  # 만약 명령을 수행하지 못하는 경우 다시 이전의 상태로 돌아가기 위한 좌표
        can_move = True
        
        for _ in range(n):
            nx += dx
            ny += dy
            # 만약 아래의 조건에 해당하면 다시 이전의 상태로 복귀
            if (nx < 0 or nx >= max_height) or (ny < 0 or ny >= max_width) or (park[nx][ny] == 'X'):
                can_move = False
                break
            
        # 위의 조건들에 해당하지 않으면 x, y 업데이트
        if can_move:
            x, y = nx, ny
    
    return [x, y]