[swift] 프로그래머스 - 마법의 엘리베이터

프로그래머스

---

성공 풀이

import Foundation

func solution(_ storey:Int) -> Int {
    var result = 0
    var intArr = String(storey).map {Int(String($0))!}
    
    for i in (0..<intArr.count).reversed() {
        let n = intArr[i]
        if i == 0 { // 현재층이 제일 높은 층 일 때
            if n > 5 { 
                result += 11-n
            } else {
                result += n
            }
        } else {
            if n == 5 { // 현재 층이 5층일 때
                if intArr[i-1] > 4 { // 앞층이 4층 보다 크면
                    result += 10-n
                    intArr[i-1] += 1
                } else { // 앞층이 4층 이하라면
                    result += n
                }
            } else if n > 5 { // 현재 층이 5층 이상일 때
                result += 10-n
                intArr[i-1] += 1
            } else { // 현재 층이 4층 이하일 때
                result += n
            }
        }
    }
    return result
}

풀이 키워드

구현, 시뮬레이션

---