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

문제

프로그래머스

풀이

def solution(park, routes):
    dx = {'N':-1 ,'S':1 ,'W':0 ,'E':0 }
    dy = {'N':0 ,'S':0 ,'W':-1 ,'E':1 }
    # 시작좌표 [x,y]
    x = -1
    for p in park:
        x += 1
        if 'S' in p:
            y = p.index('S')
            break
    # 한 칸씩 이동해보면서 끝까지 조건 만족하면, 명령 수행
    for r in routes:
        op, n = r.split()
        for i in range(1,int(n)+1):
            nx = x + i*dx[op]
            ny = y + i*dy[op]
            if 0 <= nx < len(park) and 0 <= ny < len(park[0]) and park[nx][ny] != 'X':
                continue
            else:
                break
        else:
            x, y = nx, ny
    return [x, y]

결과