단의 개발 블로그

공원 산책 본문

Algorithm/PYTHON

공원 산책

danso 2024. 7. 24. 15:49

💡 문제

지나다니는 길을 '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

입출력 예 #1

입력된 명령대로 동쪽으로 2칸, 남쪽으로 2칸, 서쪽으로 1칸 이동하면 [0,0] -> [0,2] -> [2,2] -> [2,1]이 됩니다.

입출력 예 #2

입력된 명령대로라면 동쪽으로 2칸, 남쪽으로 2칸, 서쪽으로 1칸 이동해야하지만 남쪽으로 2칸 이동할 때 장애물이 있는 칸을 지나기 때문에 해당 명령을 제외한 명령들만 따릅니다. 결과적으로는 [0,0] -> [0,2] -> [0,1]이 됩니다.

입출력 예 #3

처음 입력된 명령은 공원을 나가게 되고 번째로 입력된 명령 또한 장애물을 지나가게 되므로 입력은 제외한 번째 명령만 따르므로 결과는 다음과 같습니다. [0,1] -> [0,0]

 풀이

def solution(park, routes):
    # 로봇 강아지가 움직일 필드
    field = [[0 for j in range(0, len(list(park[0])))] for i in range(0, len(park))]
    
    # 2차원 배열 행 값
    row = 0
    
    # 로봇 강아지가 움직이는 필드를 2차원 배열로 생성
    for i in range(0, len(park)):
        convert_list = list(park[i])
        for j in range(0, len(convert_list)):
            field[row][j] = convert_list[j]
            # 로봇 강아지가 시작하는 지점 변수에 저장
            if(convert_list[j] == 'S'):
                location = [row, j]
        row = row + 1
        
    # 로봇 강아지 움직임 시작
    for i in routes:
        # 방향과 움직이는 칸 수를 변수에 저장
        direct, move = i.split(" ")
        
        # 제약 조건을 만날 경우 움직이기 전 위치로 되돌리기 위한 임시 변수
        tempx = location[0]
        tempy = location[1]
        
        # 방향에 따른 움직임 시작
        for j in range(0, int(move)):
            if direct == 'N':
                location[0] = location[0] - 1
            if direct == 'S':
                location[0] = location[0] + 1
            if direct == 'W':
                location[1] = location[1] - 1
            if direct == 'E':
                location[1] = location[1] + 1
                
            # 로봇 강아지가 움직일 때 체크해야 될 조건
            # 필드 밖으로 나갈 경우
            if location[0] < 0 or location[1] < 0 or location[0] >= len(field) or location[1] >= len(field[0]):
                location[0] = tempx
                location[1] = tempy
                break;
            # 필드 밖으로 나가지 않았지만, 이동했을 때 X를 만날 경우
            if field[location[0]][location[1]] == 'X':
                location[0] = tempx
                location[1] = tempy
                break;
        
    return location

 

 

'Algorithm > PYTHON' 카테고리의 다른 글

덧칠하기  (0) 2024.07.31
바탕화면 정리  (0) 2024.07.30
추억 점수  (5) 2024.07.23
달리기 경주  (0) 2024.06.03
약수의 합  (0) 2024.05.08