프로그래머스 - '공원산책' [코딩테스트 연습]

# 공원산책 # 172928 # lv1 # Python

 

# 문제

---

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

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

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

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

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

 

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

 

# 제한사항

---

• 3 ≤ park의 길이 ≤ 50
  • 3 ≤ park[i]의 길이 ≤ 50
    • park[i]는 다음 문자들로 이루어져 있으며 시작지점은 하나만 주어집니다.
    • S : 시작 지점
    • O : 이동 가능한 통로
    • X : 장애물
  • park는 직사각형 모양입니다.
• 1 ≤ routes의 길이 ≤ 50
  • routes의 각 원소는 로봇 강아지가 수행할 명령어를 나타냅니다.
  • 로봇 강아지는 routes의 첫 번째 원소부터 순서대로 명령을 수행합니다.
  • routes의 원소는 "op n"과 같은 구조로 이루어져 있으며, op는 이동할 방향, n은 이동할 칸의 수를 의미합니다.
    • op는 다음 네 가지중 하나로 이루어져 있습니다.
    • N : 북쪽으로 주어진 칸만큼 이동합니다.
    • S : 남쪽으로 주어진 칸만큼 이동합니다.
    • W : 서쪽으로 주어진 칸만큼 이동합니다.
    • E : 동쪽으로 주어진 칸만큼 이동합니다.
• 1 ≤ n ≤ 9

 

# 풀이

---

def solution(park, routes):
    # <아이디어>
    # 1. 'S' : 시작점 찾기
    # 2. 찾은 시작점으로부터 이동하는 방향, 횟수를 파악
    # 3. 동, 서, 남, 북에 따라 x, y 좌표값을 다르게 계산하여 할당.
    #  > 만약 'X': 장애물이 있으면 해당 명령은 건너뜀.
    #  > 만약 최종 움직인 횟수가 공원의 넓이(x, y)보다 커지면 해당 명령 건너뜀
    
    park_y = len(park) # y 좌표값
    park_x = len(park[0]) # x 좌표값
    
    # 시작점 찾기
    for i in range(len(park)):
        for j in range(len(park[i])):
            if park[i][j] == 'S':
                y, x = i, j     
    
    # 이동방향에 따른 좌표이동 값
    move_y = {'E': 0, 'W': 0, 'S': 1, 'N': -1}
    move_x = {'E': 1, 'W': -1, 'S': 0, 'N': 0}
    
    for route in routes:
        d, n = route.split(" ") # 이동방향, 이동횟수
        n = int(n)
        
        # 새로 이동하고 난 뒤의 좌표값
        new_move_y = y + move_y[d] * n
        new_move_x = x + move_x[d] * n
        
        # 만약 새로 이동한 좌표가 '0보다 작거나', 'park의 크기 이상'이라면 아래 명령 건너뜀. 
        if new_move_y < 0 or new_move_x < 0 or new_move_y >= park_y or new_move_x >= park_x:
            continue
        
        # n번 만큼 x, y 좌표값을 이동. 
        for i in range(1, n + 1):
            new_move_y = y + move_y[d] * i
            new_move_x = x + move_x[d] * i
            # 만약 새로 움직인 좌표에 'X' 장애물이 있다면, for문 탈출.
            if park[new_move_y][new_move_x] == 'X':
                break
        
        # 새로 이동한 좌표가 0보다 크고, 'park'의 크기보다 작으면 
        else:
            y, x = new_move_y, new_move_x  
        
    return [y, x]