프로그래머스(Lv1) – 공원산책 문제(파이썬)

프로그래머스 문제풀이

문제

문제설명

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

  • [“방향 거리”, “방향 거리” … ]

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

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

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

image

공원을 나타내는 문자열 배열 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

입출력 예

parkroutesresult
[“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]

문제 개념해석

이번 문제를 요약하자면,

  1. 로봇은 시작지점에서 시작한다.
  2. 공원은 [0,0], [H-1,W-1] 크기이다.
  3. 입력된 명령 방향으로 입력된 값만큼 이동한다.
  4. 3번을 진행할때 공원을 벗어나거나 이동경로에 장애물이 있다면 다음명령을 수행한다.
  5. 최종 도착지를 출력한다.

문제 풀이전에 지문과 제한사항을 반드시 읽으시길 바랍니다.

나의 풀이

이번문제는 코드가 좀 길게 나왔네요. 어렵진 않은데 방향별로 헷갈리지만 않고 계산식만 잘해주시면 답을 찾는데 문제없으실 겁니다.

def solution(park, routes):
    answer = []
    point = [] #Start_H,Start_W,H-1,W-1
    for i,v in enumerate(park):                              # park만큼 반복
        for j,k in enumerate(v):                             # park요소의 문자열만큼 반복
            if k == 'S':                                     # 'S'(시작지점)일때
                point = [i,j,len(park)-1,len(v)-1]           # 위치 저장, 공원 끝지점도 같이 저장
    for i in routes:                                         # routes 명령어 요소만큼 반복
        direction = i.split()[0]                             # 요소를 split하여 방향 저장
        move = int(i.split()[1])                             # 요소를 split하여 이동횟수 저장
        if direction == 'N':                                 # 'N'방향 이동
            if point[0] - move >= 0:                         # 0보다 크거나 같은지                
                for j in range(point[0]-move,point[0]+1):    # 현재 이동거리만큼 반복
                    if park[j][point[1]] == 'X':             # 'X'(장애물)이 있는지
                        break                                # 있다면 반복문 종료
                else:                                        # 장애물이 없어 반복문이 끝난다면
                    point[0] = point[0]-move                 # 위치이동
        elif direction == 'S':                               # 'N'방향의 공식을 각 방향에 맞고 수정
            if point[0] + move <= point[2]:
                for j in range(point[0],point[0]+move+1):
                    if park[j][point[1]] == 'X':
                        break
                else:
                    point[0] = point[0]+move
        elif direction == 'W':
            if point[1] - move >= 0:
                for j in range(point[1]-move,point[1]+1):
                    if park[point[0]][j] == 'X':
                        break
                else:
                    point[1] = point[1]-move
        else:
            if point[1] + move <= point[3]:
                for j in range(point[1],point[1]+move+1):
                    if park[point[0]][j] == 'X':
                        break
                else:
                    point[1] = point[1]+move
    answer = [point[0],point[1]]                              # 현재위치 answer에 입력
    return answer

다른 사람의 풀이

다른 풀이들도 비슷해서 가져오지 않았습니다.

관련 링크

문제해결과정에서 기초가 부족하다고 느껴지시면 아래 정리 글을 참고해주세요.
추가적으로 궁금한 사항이나 글에 대한 개선점 알려주시면 피드백 적극 반영하도록 하겠습니다.