Lv.1 공원 산책

문제 설명

지나다니는 길을 'O', 장애물을 'X'로 나타낸 직사각형 격자 모양의 공원에서 로봇 강아지가 산책을 하려합니다. 산책은 로봇 강아지에 미리 입력된 명령에 따라 진행하며, 명령은 다음과 같은 형식으로 주어집니다.

["방향 거리", "방향 거리" … ]
예를 들어 "E 5"는 로봇 강아지가 현재 위치에서 동쪽으로 5칸 이동했다는 의미입니다. 로봇 강아지는 명령을 수행하기 전에 다음 두 가지를 먼저 확인합니다.

주어진 방향으로 이동할 때 공원을 벗어나는지 확인합니다.
주어진 방향으로 이동 중 장애물을 만나는지 확인합니다.
위 두 가지중 어느 하나라도 해당된다면, 로봇 강아지는 해당 명령을 무시하고 다음 명령을 수행합니다.
공원의 가로 길이가 W, 세로 길이가 H라고 할 때, 공원의 좌측 상단의 좌표는 (0, 0), 우측 하단의 좌표는 (H - 1, W - 1) 입니다.

공원을 나타내는 문자열 배열 park, 로봇 강아지가 수행할 명령이 담긴 문자열 배열 routes가 매개변수로 주어질 때, 로봇 강아지가 모든 명령을 수행 후 놓인 위치를 [세로 방향 좌표, 가로 방향 좌표] 순으로 배열에 담아 return 하도록 solution 함수를 완성해주세요.

입출력

코드 풀이

def solution(park, routes):
    result = [] #최종 위치를 저장
    
    #park 배열에서 시작 위치 'S'를 찾기
    for i in range(len(park)):
        for j in range(len(park[0])):
            if park[i][j] == 'S':
                cur_x, cur_y = j, i
                continue
                
    # 주어진 좌표 (x, y)가 유효한지 검사
    def check(x, y):
        if (0 > x or x >= len(park[0])) or (0 > y or y >= len(park)) or park[y][x] == 'X':   #좌표가 공원 내에 있고 장애물('X')이 아니면 T/F
            return False
        return True
        
    #방향과 이동 거리를 분리하고, 이동 거리를 정수로 변환
    for route in routes:
        direct, move = route.split(' ')
        move = int(move)
        if direct == 'E':
            n_x = cur_x + move
            for t_x in range(cur_x, n_x + 1):
                if check(t_x, cur_y) == False:
                    break
            else:
                cur_x = n_x
        elif direct == 'W':
            n_x = cur_x - move
            for t_x in range(n_x, cur_x + 1):
                if check(t_x, cur_y) == False:
                    break
            else:
                cur_x = n_x
        elif direct == 'N':
            n_y = cur_y - move
            for t_y in range(n_y, cur_y + 1):
                if check(cur_x, t_y) == False:
                    break
            else:
                cur_y = n_y
        elif direct == 'S':
            n_y = cur_y + move
            for t_y in range(cur_y, n_y + 1):
                if check(cur_x, t_y) == False:
                    break
            else:
                cur_y = n_y

    return [cur_y, cur_x]

출처: 프로그래머스 코딩테스트 연습