[프로그래머스] LV.1 공원 산책(python)

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

문제 해결

def solution(park, routes):
    start = [0,0]
    ob = []
    for i in range(len(park)):
        for j in range(len(park[i])):
            if park[i][j] == 'S':
                start[0] = i
                start[1] = j
            if park[i][j] == 'X':
                tmp = []
                tmp.append(i)
                tmp.append(j)
                ob.append(tmp)
    for i in routes:
        if i[0] == 'E':
            n = start[1]
            if n + int(i[2]) >= len(park[0]):
                continue
            else:
                for j in range(int(i[2])):
                    start[1] += 1
                    if start in ob:
                        start[1] = n
                        break
        if i[0] == 'W':
            n = start[1]
            if n - int(i[2]) < 0:
                continue
            else:
                for j in range(int(i[2])):
                    start[1] -= 1
                    if start in ob:
                        start[1] = n
                        break
        if i[0] == 'S':
            n = start[0]
            if n + int(i[2]) >= len(park):
                continue
            else:
                for j in range(int(i[2])):
                    start[0] += 1
                    if start in ob:
                        start[0] = n
                        break
        if i[0] == 'N':
            n = start[0]
            if n - int(i[2]) < 0:
                continue
            else:
                for j in range(int(i[2])):
                    start[0] -= 1
                    if start in ob:
                        start[0] = n
                        break
    return start

 

문제 풀이 설명

start 변수는 시작지점을 의미하며, 최종 결과로 return하는 값 또한 시작지점에서 변동된 값을 그대로 return한다.

ob는 장애물의 위치를 기록해놓는 리스트이며, 이동 중에 장애물을 만났는지 여부를 확인하기 위한 용도로 사용했다.

첫번째 for문으로 시작지점을 기록하고, 장애물의 위치 또한 ob 리스트에 append한다.

 

두번째 for문으로 routes 매개변수를 돌며, 어느 방향으로 몇번 이동하는지 파악한다.

방향의 종류는 총 N, S, W, E 4가지이며, 각각 방향마다 start의 0번째 인덱스 또는 1번째 인덱스가 더하거나 빼진다.

이때 해당방향으로 움직였을 때 공원의 가로길이와 세로길이를 벗어나는지 먼저 파악한 후, 벗어난다면 continue로 해당 명령을 무시하고 다음 명령으로 넘어간다.

 

만약 벗어나지 않는다면 각 방향에 맞게 위치를 더하거나 빼는데, 이때 for문으로 하나씩 더하거나 빼면서 ob(장애물 리스트)에 포함이 되어있는지 하나하나 확인한다. 만약 포함되어있다면 이동중에 장애물을 만나는 경로이므로 역시 해당 값을 초기화하고 break를 사용해 다음 명령으로 넘어간다.

단순구현문제여서 재밌게 풀 수 있었던 것 같다.