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

문제

 

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

---

문제풀이

• 현재 좌표 (now_row,now_col) 와 이동 후 좌표 (new_row_new_col)을 구분해서 다루었다. 
• 현재 좌표 (now_row,now_col) 에서 이동 후 좌표 (new_row_new_col)로 한 칸씩 이동하는 과정에서,
  • 이동을 하지 못할 조건( 장애물 X를 만나거나, 범위를 벗어나는 경우) 에 걸리지 않으면 -> now_row와 now_col을 new_x와 new_y로 갱신
  • 이동을 하지 못할 조건에 해당되면 -> 벗어난다

잘못한 점

신경쓰지 못한 부분이 하나 있었다.

표시한 부분에서 알 수 있듯, 4가지 direction(동,서,남,북)에 따라 for문의 생김새가 달라야 한다

• S(북쪽) : now_row -> new_row까지 1씩 '증가'하는 방향으로 진행
• N(남쪽) : now_row -> new_row까지 1씩 '감소'하는 방향으로 진행
• W(서쪽) : now_col -> new_col까지 1씩 '증가'하는 방향으로 진행
• E(동쪽) : now_col -> new_col까지 1씩 '감소'하는 방향으로 진행

나는 '음의 방향'으로는 조건을 필터링하는걸 생각하지 못하고, 단순히 now<new 일거라고만 생각했다.
 
수정한 코드는 아래와 같다
 

정답

def solution(park, routes): #상,우,하,좌
    H = len(park)
    W = len(park[0])
    d_row = [-1,1]
    d_col = [-1,1]
    for i in range(len(park)):
        for j in range(len(park[i])):
            if park[i][j] == 'S':
                now_row,now_col = i,j 
                
    for rt in routes:
        direction,length= rt.split()
        if direction == 'E': #동 -> j의 이동
            n_row = now_row
            n_col = now_col + (d_col[1]*int(length))
        elif direction =='S': #남 -> i의 이동
            n_row = now_row + (d_row[1]*int(length))
            n_col = now_col
        elif direction == 'W': #서 -> j의 이동
            n_row = now_row
            n_col = now_col + (d_col[0]*int(length))
        elif direction == 'N': #북
            n_row = now_row + (d_row[0]*int(length))
            n_col = now_col
        
        blocked = False
     # 남쪽 또는 북쪽으로 이동하는 경우
        if direction == 'S':
            for i in range(now_row + 1, n_row + 1):
                if i < 0 or i >= H or park[i][now_col] == 'X':
                    blocked = True
                    break
        elif direction == 'N':
            for i in range(now_row - 1, n_row - 1, -1):
                if i < 0 or i >= H or park[i][now_col] == 'X':
                    blocked = True
                    break
        
        # 동쪽 또는 서쪽으로 이동하는 경우
        elif direction == 'E':
            for j in range(now_col + 1, n_col + 1):
                if j < 0 or j >= W or park[now_row][j] == 'X':
                    blocked = True
                    break
        elif direction == 'W':
            for j in range(now_col - 1, n_col - 1, -1):
                if j < 0 or j >= W or park[now_row][j] == 'X':
                    blocked = True
                    break
        
        if blocked:
            continue
        else:
            now_row, now_col = n_row, n_col
    
    result = [now_row, now_col]
    return result

 

첨언

테스트케이스의 생김새만 보고 코드를 유추하는 습관을 지양하자
그렇게 할 수록 사고가 국한되어 버려서 예외를 생각해내기 힘들다
 
문제를 꼼꼼히 읽고, 예외사항까지 생각해내서 처리 후, 테스트케이스를 이용하는 식으로 구현하자