(프로그래머스) lv.1 공원 산책

 

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

프로그래머스

문제풀이

def solution(park, routes):
    answer = []
    n, m = len(park), len(park[0])

    d_dict = {
        "N": (-1, 0),
        "S": (1, 0),
        "W": (0, -1),
        "E": (0, 1),
    }

    def search(y, x):
        for route in routes:
            key, value = route.split()  # 방향, 거리 분리

            dy, dx = d_dict[key]  # 명령에 해당 하는 이동 방향 선언
            ty, tx = y, x  # 현재 좌표 초기화

            for _ in range(int(value)):
                # 다음 이동 경로 탐색
                ny, nx = ty + dy, tx + dx

                if 0 <= ny < n and 0 <= nx < m and park[ny][nx] not in "X":
                    # 맵을 벗어 나지 않고 이동 경로에 X가 없을 경우 현재 좌표 갱신
                    ty, tx = ny, nx
                else:
                    # 맵을 벗어 나거나 경로에 X가 있을 경우 현재 좌표 초기화
                    ty, tx = y, x
                    break
            # 최종 결과 저장
            y, x = ty, tx
        return [y, x]

    # 행, 열 기준 시작지점 탐색
    for i in range(n):
        for j in range(m):
            if park[i][j] == "S":
                answer = search(i, j)
                break

    return answer