[알고리즘 문제풀이] 공원 산책

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 함수를 완성해주세요.

제한사항

입출력 예

< 소스코드 >

def solution(park, routes):
    
    row, col = 0, 0 # 현재 위치
    park_row, park_col = len(park), len(park[0]) # park 크기
    move = {"E" : (0, 1), "W" : (0, -1), "S" : (1, 0), "N" : (-1, 0)}
    
    # 시작 위치 찾기
    for i, row in enumerate(park):
        if "S" in row:
            row, col = i, row.find("S")
            break
    
    for route in routes:
        mov_r, mov_c = move[route[0]] # 입력받은 움직임 방향
        new_r, new_c = row, col # 적용 후 위치 저장
        for _ in range(int(route[2])):
            # 한칸씩 움직이며, 장애물이 잇는지 판단
            if (0 <= new_r + mov_r < park_row and 0 <= new_c + mov_c < park_col 
                and park[new_r+mov_r][new_c+mov_c] != "X"):
                new_r, new_c = new_r+mov_r, new_c+mov_c
            else:
                new_r, new_c = row, col
                break
        row, col = new_r, new_c # 위치 업데이트
    
    return [row, col]