프로그래머스: 공원 산책

[level 1] 공원 산책 - 172928

문제 링크

성능 요약

메모리: 10.4 MB, 시간: 0.05 ms

구분

코딩테스트 연습 > 연습문제

채점결과

정확성: 100.0
합계: 100.0 / 100.0

제출 일자

2024년 1월 2일 2:59: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

---

입출력 예

park	routes	result
["SOO","OOO","OOO"]	["E 2","S 2","W 1"]	[2,1]
["SOO","OXX","OOO"]	["E 2","S 2","W 1"]	[0,1]
["OSO","OOO","OXO","OOO"]	["E 2","S 3","W 1"]	[0,0]

---

입출력 예 설명

입출력 예 #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):
    start_x = None
    start_y = None
    for idx1, row in enumerate(park):
        if 'S' in row:
            start_y = idx1
            break

    if start_y is not None:
        for idx2, col in enumerate(park[start_y]):
            if 'S' == col:
                start_x = idx2
                break

    for cnt in routes:
        dir = cnt[0]
        move = int(cnt[-1])
        prev_x, prev_y = start_x, start_y
        for _ in range(move):
            if dir == 'E' and check(start_x + 1, start_y, len(park[0]), park):
                start_x += 1

            elif dir == 'W' and check(start_x - 1, start_y, len(park[0]), park):
                start_x -= 1

            elif dir == 'S' and check(start_x, start_y + 1, len(park), park):
                start_y += 1

            elif dir == 'N' and check(start_x, start_y - 1, len(park), park):
                start_y -= 1

            else:            
                start_x, start_y = prev_x, prev_y
                break

    return [start_y, start_x]

def check(x, y, n, park):
    return 0 <= x and x < n and 0 <= y and y < n and park[y][x] != 'X'

느낀점

격자 범위를 체크하는 부분까지는 머릿속에서 바로바로 떠올라서 작성을 완성했다. 하지만 이후에 이동할 거리에 장애물이 있는지 확인하는 로직을 구현하는 부분에서 한 번에 확인해야 한다는 생각 때문에 쉽게 풀이가 떠오르지 않아서 여러 방법을 고민해 봤다. 그 뒤로 생각난것은 주어진 공원의 크기가 가로와 세로가 최대 50이고 명령어의 길이가 50이기 때문에 이동해야 하는 길이가 전부 9에 명령 50 크기 50, 50으로 전부 다 해도 수가 작기 때문에 한 칸씩 이동하며 장애물을 확인하고, 만약 장애물이 발견되면 여태 이동한 거리는 이전 거리로 다시 바꾼 뒤에 해당 이동을 종료하는 방법을 사용했다.

최악의 경우에도 백만정도의 크기를 가지기 때문에 수월하게 통과한것 같다.

추후에 level2를 풀때는 bfs, dfs 등을 더 공부해서 이러한 방법도 적용해서 문제를 풀어내면 더 좋을것 같다.

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges