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

👆문제풀러 가기👆

문제 설명

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

 

입출력 예

park	routes	result
["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]

---

풀이 [ 메모리: 16.7mb, 최대시간: 2.72ms ] 

func solution(_ park:[String], _ routes:[String]) -> [Int] {
    var s = [0, 0]
    var x = [[Int]]()
    // 동서남북 최대,최소값
    let nswe = (north: 0, south: park.count-1, west: 0, east: park[0].count-1)
    
    // 공원 시작위치, 장애물 위치 파악
    checkCoordinates(of: park, &s, &x)
    
    // 로봇강아지 이동
    for i in 0..<routes.count {
        let route = routes[i].split(separator: " ").map { String($0) }
        let direction = route[0]
        guard let distance = Int(route[1]) else { break }
        
        var tempS = s
        
        // 방향별 1칸씩 이동해보기
        for _ in 0..<distance {
            // 이동할 수 있는 거리인지 확인
            switch direction {
            case "N" where (s[0] - distance) >= nswe.north: tempS[0] -= 1
            case "S" where (s[0] + distance) <= nswe.south: tempS[0] += 1
            case "E" where (s[1] + distance) <= nswe.east: tempS[1] += 1
            case "W" where (s[1] - distance) >= nswe.west: tempS[1] -= 1
            default: break
            }
            // 이동 중 장애물을 만나면 초기화
            if x.contains(tempS) {
                tempS = s
                break
            }
        }
        // 무사히 이동하면 값 저장
        s = tempS
    }
    
    return s
}

func checkCoordinates(of park:[String], _ s: inout [Int], _ x: inout [[Int]]){
    for h in 0..<park.count {
        let parkLine = park[h].map { String($0) }
        
        for w in 0..<parkLine.count {
            if parkLine[w] == "S" {
                s = [h, w]
            }
            if parkLine[w] == "X" {
                x.append([h, w])
            }
        }
    }
}

문제에서 고려해야 할 사항을 다 알려주고 있기 때문에 어렵진 않은데,, 시간이 좀 걸릴 수 있는 문제라고 생각한다.

함수를 더 나누고 싶었는데, 파라미터가 너무 많아질 것 같아 하나만 따로 뺐다.

 

1. 시작좌표(s)에서 로봇강아지 이동하기 전, 시작 좌표(s)와 장애물 좌표(x)를 확인
2. routes에서 띄어쓰기 기준 방향(direction)과 거리(distance) 확인
3. 공원을 벗어나지 않는 범위 내에서, 한 칸씩 이동하며 장애물에 닿는지 확인. (범위를 벗어나거나 장애물에 닿으면 이동하지 않음)
4. 이동 좌표(tempS -> s) 저장