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

1. 문제 설명

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

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

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

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

위 두 가지중 어느 하나라도 해당된다면, 로봇 강아지는 해당 명령을 무시하고 다음 명령을 수행합니다. 공원의 가로 길이가 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

 

2. 출력 예시

 

3. 문제 답안

def solution(park, routes):
    answer = []
    start = [] # 시작 좌표
    block = set() # 장애물 좌표
    
    park_w = len(park[0]) # 공원 가로 길이
    park_h = len(park) # 공원 세로 길이
    
    for row, road in enumerate(park):
        for col, val in enumerate(road):
            if val == 'S': # 시작 좌표는 1개
                start.append(row)
                start.append(col)
            elif val == 'X': # 장애물은 0개 이상
                block.add((row, col))
                
    answer = start
    
    # 공원 벗어나는지 확인
    def is_out(row, col):
        if park_w <= col or col < 0:
            return 1
        if park_h <= row or row < 0:
            return 1
        else:
            return 0
    
    # 명령별 순회
    for r in routes:
        direction, distance = r.split()
        distance = int(distance)
        
        check_out = 0 # 공원 벗어남 확인
        check_block = 0 # 장애물 확인
        
        if direction == 'E':
            for i in range(1, distance+1): # 1칸씩 이동
                # 공원 벗어나는지 확인
                check_out = is_out(answer[0], answer[1]+i)
                if check_out == 1:
                    break
                # 장애물 만나는지 확인
                if (answer[0], answer[1]+i) in block:
                    check_block = 1
                    break
            if check_out == 0 and check_block == 0:
                answer[1] += distance  
                    
        elif direction == 'W':
            for i in range(1, distance+1): # 1칸씩 이동
                # 공원 벗어나는지 확인
                check_out = is_out(answer[0], answer[1]-i)
                if check_out == 1:
                    break
                # 장애물 만나는지 확인
                if (answer[0], answer[1]-i) in block:
                    check_block = 1
                    break
            if check_out == 0 and check_block == 0:
                answer[1] -= distance  
                    
        elif direction == 'S':
            for i in range(1, distance+1): # 1칸씩 이동
                # 공원 벗어나는지 확인
                check_out = is_out(answer[0]+i, answer[1])
                if check_out == 1:
                    break
                # 장애물 만나는지 확인
                if (answer[0]+i, answer[1]) in block:
                    check_block = 1
                    break
            if check_out == 0 and check_block == 0:
                answer[0] += distance
                    
        elif direction == 'N':
            for i in range(1, distance+1): # 1칸씩 이동
                # 공원 벗어나는지 확인
                check_out = is_out(answer[0]-i, answer[1])
                if check_out == 1:
                    break
                # 장애물 만나는지 확인
                if (answer[0]-i, answer[1]) in block:
                    check_block = 1
                    break
            if check_out == 0 and check_block == 0:
                answer[0] -= distance            
        
        # print(f'{r} 명령 후 현재 위치', answer)
                    
    return answer