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

YoungHyun Kim·2023년 12월 7일
1

매일매일 알고리즘

목록 보기
20/30

문제

지나다니는 길을 '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. 입력으로 받은 park의 내부에 있는 문자열을 배열 형식으로 재정의하기
  2. 시작 지점의 좌표 구하기
  3. 입력으로 받은 만큼 이동하는 알고리즘 작성
  • 고려할 점
    • 특정 방향으로 이동할 시에, 입력받은 거리만큼 이동하는 도중에 공원을 벗어난다면 해당 입력은 무시해야함.
    • 이동할 경로에 X가 있다면, 해당 명령을 무시해야 함.
  1. 문제 해결
import Foundation

func solution(_ park:[String], _ routes:[String]) -> [Int] {
    let height = park.count, length = park[0].count
    var parkArr: [[String]] = []
    var currentX = 0, currentY = 0
    
    for subs in park {
        var substring: [String] = []
        for ob in subs {
            substring.append(String(ob))
        }
        parkArr.append(substring)
    }
    
    for arr in park {
        var found = false
        currentX = 0
        for obj in arr {
            if obj == "S"{
                found = true
                break
            }
            currentX += 1
        }
        if found {
            break
        }
        currentY += 1
    } // 시작 좌표를 찾음
    
    // 입력으로 받은 만큼 이동하는 알고리즘
    for route in routes {
        let op = route.split(separator: " ")[0], n = Int(route.split(separator: " ")[1])
        
        if op == "N" && currentY - n! >= 0 {
            var canMove = true
            for step in 1...n! {
                if parkArr[currentY - step][currentX] == "X" {
                    canMove = false
                }
            }
            if canMove {
                currentY -= n!
            }
        } else if op == "S" && currentY + n! < height {
            var canMove = true
            for step in 1...n! {
                if parkArr[currentY + step][currentX] == "X" {
                    canMove = false
                }
            }
            if canMove {
                currentY += n!
            }
        } else if op == "W" && currentX - n! >= 0 {
            var canMove = true
            for step in 1...n! {
                if parkArr[currentY][currentX - step] == "X" {
                    canMove = false
                }
            }
            if canMove {
                currentX -= n!
            }
        } else if op == "E" && currentX + n! < length {
            var canMove = true
            for step in 1...n! {
                if parkArr[currentY][currentX + step] == "X" {
                    canMove = false
                }
            }
            if canMove {
                currentX += n!
            }
        }
    }
    return [currentY, currentX]
}

회고

  • 고민했던 점
    • 공원을 벗어나는 명령을 필터링하는 조건식을 고안하는데 쓸데없는 고민을 많이 함. 예를 들어, 현재좌표 + 이동할 거리 < 공원의 우측 끝 좌표 등의 간단한 조건식을 고안하는데 잘못된 접근을 해서 약 30분을 고민함.
    • 해당 삽질(?)로 인해서 index 범위 관련한 오류를 많이 겪음
  • 새로 알게된 것
    • '사실 알고리즘 고수들은 조건식을 잘 사용하고, 적절한 식을 고안하는 사람이다'를 알게된 것 같다.
profile
iOS 개발자가 되고 싶어요

0개의 댓글