[프로그래머스] 공원 산책 (Python)

🌱 문제

프로그래머스

 

🌱 풀이

이 문제는 단순 구현 문제로, 문제에서 요구하는 설명을 그대로 코드로 작성하면 해결할 수 있는 문제입니다.

• 우선, S의 위치를 저장합니다. 
• 이동하는 명령들(routes)을 하나씩 돌면서 해당 명령을 수행할 수 있는지 검사합니다.
• 해당 명령을 수행할 수 있는지 검사할 때에는 한 칸씩 이동하며 범위를 벗어나지 않는지, 장애물이 있지 않은지 검사해야 하기때문에, 한칸씩 이동하는 좌표값을 저장할 수 있는 변수(nx, ny)를 사용하였습니다.
• 중간에 break되지 않고 해당 명령을 끝까지 수행할 수 있음이 확인되었으면 S의 위치(cx, cy)를 업데이트합니다.
• 마지막으로 모든 경로를 수행하고 업데이트된 S의 위치(cx, cy)를 리턴합니다.

 

🌱 코드

# !usr/bin/env python
# -*- coding: utf-8 -*-
# programmers 공원 산책

def solution(park, routes):
    answer = []
    h = len(park)
    w = len(park[0])

    # E, W, S, N
    dict = {"E": 0, "W": 1, "S": 2, "N": 3}
    x = [0, 0, 1, -1]
    y = [1, -1, 0, 0]

    # S의 위치 저장
    for i, row in enumerate(park):
        if "S" in row:
            cx, cy = i, row.index("S")

    # 이동하기
    for route in routes:
        direction, count = route.split(" ")
        dx = x[dict[direction]]
        dy = y[dict[direction]]
        for cnt in range(1, int(count) + 1):
            nx, ny = cx + dx * cnt, cy + dy * cnt
            if (nx < 0 or ny < 0 or nx >= h or ny >= w):    # 맵 범위를 벗어나는지 검사
                break
            elif park[nx][ny] == "X":   # 장애물이 있는지 검사
                break
            if cnt == int(count):
                cx += dx * int(count)
                cy += dy * int(count)
    return [cx, cy]