프로그래머스 - 주차 요금 계산(Swift)

프로그래머스 - 주차 요금 계산(Swift)

 

 

 문제 설명

 

https://school.programmers.co.kr/learn/courses/30/lessons/92341

프로그래머스

 

 나의 풀이

 

문제를 풀면서 알게된 점은 딕셔너리를 for문 돌면 튜플형태로 key와 value가 생긴다는 것이다.

이 점을 이용해서 문제를 풀어내면 된다.

 

import Foundation

func solution(_ fees:[Int], _ records:[String]) -> [Int] {
    
    func calculate(_ minute: Int) -> Int {
        if minute <= fees[0] {
            return fees[1]
        } else {
            let price = fees[1] + Int(ceil(Double(minute - fees[0]) / Double(fees[2]))) * fees[3]
            return price
        }
    }
    
    var resultDic: [String: Int] = [:]
    var timeDic: [String: String] = [:]
    
    for record in records {
        let res = record.components(separatedBy: " ")
        let time = res[0], carNum = res[1], signal = res[2]
        if timeDic[carNum] != nil && signal == "OUT" {
            let inTimeArr = timeDic[carNum]!.components(separatedBy: ":")
            let inHour = Int(inTimeArr[0])!, inMinute = Int(inTimeArr[1])!
            let outTimeArr = time.components(separatedBy: ":")
            let outHour = Int(outTimeArr[0])!, outMinute = Int(outTimeArr[1])!
            let accTime = (outHour - inHour) * 60 + (outMinute - inMinute)
            if resultDic[carNum] != nil {
                resultDic[carNum]! += accTime 
            } else {
                resultDic[carNum] = accTime
            }
            timeDic[carNum] = nil
        } else {
            timeDic[carNum] = time
        }
    }
    print(timeDic)
    for x in timeDic {
        print(x)
        let inTime = x.1.components(separatedBy: ":") 
        let inHour = Int(inTime[0])!, inMinute = Int(inTime[1])!
        let outHour = 23, outMinute = 59
        let accTime = (outHour - inHour) * 60 + (outMinute - inMinute)
        if resultDic[x.0] != nil {
            resultDic[x.0]! += accTime    
        } else {
            resultDic[x.0] = accTime
        }
        timeDic[x.0] = nil 
    }
    let arr = resultDic.sorted(by: <).map{ $0.value }
    
    return arr.map{ calculate($0) }
}