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

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

프로그래머스

📄문제

 

문제 설명

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

• ["방향 거리", "방향 거리" … ]

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

• 주어진 방향으로 이동할 때 공원을 벗어나는지 확인합니다.
• 주어진 방향으로 이동 중 장애물을 만나는지 확인합니다.

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

 

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

 

📝 코드

def solution(park, routes):
    
    n = len(park) # 행
    m = len(park[0])  # 열
    
    # 시작점 찾기
    start = [(i,j) for i in range(n) for j in range(m) if park[i][j] == 'S']
    x, y = int(start[0][0]), int(start[0][1])
    
    # 명령에 따라 이동
    for route in routes:
        
        # 방향, 거리
        direction, d = route.split(' ') 
        dist = int(d)
        
        if direction == 'E' and (y + dist <= m-1) : 
            for i in range(dist):
                if park[x][y+i+1] == 'X':
                    break
            else:
                y += dist
            
            
        elif direction == 'W' and (y - dist >= 0) :
            for i in range(dist):
                if park[x][y-i-1] == 'X':
                    break
            else:
                y -= dist
            
        elif direction == 'S' and (x + dist <= n-1) :
            for i in range(dist):
                if park[x+i+1][y] == 'X':
                    break
            else:
                x += dist
            
            
        elif direction == 'N' and (x - dist >= 0) :
            for i in range(dist):
                if (park[x-i-1][y] == 'X'):
                    break
            else:
                  x -= dist
    
    return [x,y]

 

코드의 주요 로직

1. 시작점 찾기: 주차장 배열에서 'S'로 표시된 시작점의 위치를 찾음
2. 경로에 따라 이동하기: 주어진 경로(routes)를 따라 이동함
3. 이동 제한 조건 처리:
   • 공원을 벗어나는 경우: 이동하려는 방향과 거리가 주차장의 범위를 벗어나지 않도록 함
   • 장애물이 있는 경우: 이동 경로상에 'X'로 표시된 장애물이 있으면 해당 방향으로의 이동을 중단함
4. 최종 도달 위치 반환