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

Programmers

공원 산책

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

프로그래머스

문제

코드

• 시작점(S)를 찾기
• 이동 방향을 한칸씩 이동하도록 만들어서 중간에 장애물이 있는 경우를 고려

def solution(park, routes):

    x, y = 0, 0

    for i in range(len(park)):
        for j in range(len(park[i])):
            if park[i][j] == 'S':
                x = j
                y = i
                break

    for route in routes:
        new_x, new_y = x, y

        for i in range(int(route[2])):
            if route[0] == 'E' and new_x < len(park[0])-1 and park[new_y][new_x+1] != 'X':
                new_x += 1
                if i == int(route[2]) - 1:
                    x = new_x
            if route[0] == 'W' and new_x > 0 and park[new_y][new_x-1] != 'X':
                new_x -= 1
                if i == int(route[2]) - 1:
                    x = new_x
            if route[0] == 'N' and new_y > 0 and park[new_y-1][new_x] != 'X':
                new_y -= 1
                if i == int(route[2]) - 1:
                    y = new_y
            if route[0] == 'S' and new_y < len(park)-1 and park[new_y+1][new_x] != 'X':
                new_y += 1
                if i == int(route[2]) - 1:
                    y = new_y

    return [y, x]