[4코1파] #96. LV 1. 공원 산책

gunny·2023년 4월 9일

4코1파 코딩테스트 코딩테스트 python 프로그래머스 공원 산책 프로그래머스 공원 산책 python

코딩테스트

목록 보기

96/538

[4코1파] 4명의 안드로이드 개발자와 1명의 파이썬 개발자의 코딩 테스트 서막 : 4코1파

Rule :

하루에 1문제씩 풀기.
한 문제당 30분씩은 고민하기.
왜 그렇게 풀었는지 공유하기.
하루라도 놓친다면 벌금은 1,000원

START :

[3코1파] 2023.01.04~ (96일차)
[4코1파] 2023.01.13~ (86일차)

Today :

2023.04.09 [96일차]

프로그래머스 LV 1
공원 산책
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 함수를 완성해주세요.

제한사항

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]

문제 풀이 방법

저런식으로 1단계에서 문자열과 숫자열 방향을 조작해서 핸들링 해야 하는건 enumerate 를 사용해서 인덱스와 값을 return 해 놓는게 신상에 좋다. 자칫하다간 코드가 너무 길어서져서 난해하기 때문

뭐 항상 동,서,남,북으로 이동해야하니까 position에 해당하는 좌표와 그에 상응하는 위치를 딕셔너리로 만들고,
첫 좌표 pos 지정

주어진 공원에 해당하는 행과, 그 행에 따른 elements들을
enumerate로 핸들링했고, "S" 인 시작점을 찾는다.

시작점을 찾은 후에는, routes에 에 해당하는 postion 위치와 거리를 가지고 오는데, 위에서 구했던 시작 점을 기반으로 딕셔너리에서 찾아서 위치를 갱신한다.

물론,, 항상 있는 조건 움직일 수 있는 거리의 제한점 체크
0보다 길이가 작으면 안되고, 공원보다 작으면 안되는,,,
아울러 'X'이면 안되는.. 생각보다 구현 복잡한데..

내 코드

def solution(park, routes):
    positions = {'E': (0,1), 'W': (0,-1), 'N' : (-1,0), 'S': (1,0)}
    pos = [0,0]

    for i, row in enumerate(park):
        for j, r in enumerate(row):
            if r == 'S':
                pos = [i, j]

    for position, distance in map(lambda x: x.split(), routes):
        pre_pos = pos.copy()
        for d in range(int(distance)):
            pos[0] += positions[position][0]
            pos[1] += positions[position][1]

            if not (0<=pos[0]<len(park) and 0<=pos[1]<len(park[0]) and
                   park[pos[0]][pos[1]]!='X'):
                pos = pre_pos
                break
            
    return pos

증빙