공원 산책 Swift

문제 설명

지나다니는 길을 '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. [방향: 1 or -1] Dictionary 에 저장한다.

2. 공원을 2차원배열로 변경한다.

3. 명령의 움직임 만큼 증가 혹은 감소했을때 맵을 벗어나는지 1차확인한다.

4. 벗어나지 않는다면 이동한 배열의 범위 추출한다.

5. 추출한 배열에서 장애물 "X" 가 있는지 확인.

6. 위에 조건을 모두 통과하면 시작점에서 이동

 

처음에 어떻게 접근해야할지 고민이 많이 되서, 아예 천천히 꼼꼼히 따져보자고 여러가지 조건문을 걸었다.

결과추출이 [y,x]로 고정 되어있어서, 2차원 배열에서 인덱스 접근이 헷갈렸고, route[0][1...2] 이런식으로 추출은 됬는데,

route[1...2][0] 추출은 안되서 for문으로 추출했다. 

 

import Foundation

func solution(_ park:[String], _ routes:[String]) -> [Int] {
    var route = [[String]]()
    var start = [0,0]
    var dict = ["E": 1, "S": 1, "W": -1, "N": -1]
    
    for i in 0..<park.count {            // 공원을 2차원배열로 변경 및 시작점 찾기
        let arr = Array(park[i]).map{String($0)}
        if arr.contains("S") {
            start[0] = i
            for j in 0..<arr.count {
                if arr[j] == "S" {
                    start[1] = j
                }
            }
        }
        route.append(arr)
    }
    
    for i in 0..<routes.count {         //명령들 순회
        var arr = routes[i].split(separator: " ").map{ String($0)}
        var move = dict[arr[0]]! * Int(arr[1])!
        let rowIdx = route[0].count - 1
        let colIdx = route.count - 1
        var check = [String]()
        
        if arr[0] == "E" || arr[0] == "W" {
            var moved = start[1] + move
            if moved < 0 || moved > rowIdx {    // 움직였을때 맵 밖으로 나갔는지 확인
                continue
            }
            if arr[0] == "E" {                  // 움직인 범위의 배열 추출
                for j in start[1]...moved {
                check.append(route[start[0]][j])
                }
            } else {
                for j in moved...start[1] {
                check.append(route[start[0]][j])
                }
            }
            if check.contains("X") {            // 장애물 있는지 확인
                continue
            } else {
                start = [start[0], moved]
            }
        }
        
        if arr[0] == "S" || arr[0] == "N" {
            var moved = start[0] + move
            if moved < 0 || moved > colIdx {    // 움직였을때 맵 밖으로 나갔는지 확인
                continue
            }
            if arr[0] == "S" {                  // 움직인 범위의 배열 추출
                for j in start[0]...moved {
                check.append(route[j][start[1]])
                }
            } else {
                for j in moved...start[0] {
                check.append(route[j][start[1]])
                }
            }
            if check.contains("X") {            // 장애물 있는지 확인
                continue
            } else {
                start = [moved, start[1]]
            }
        }
        
    }
    
    return start
}